test(integration): size restart-mid-burst timeouts from the proxy boot budget (#49376)

This commit is contained in:
yucheng-berri 2026-10-31 10:17:34 -07:00 • committed by GitHub
parent bfef2e1e56
commit c8c0248ac7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 6 additions and 5 deletions

View file

@ -24,6 +24,7 @@ from _openinference_support import (
_rig,
)
from integration._support.client import Gateway
from integration._support.process import graceful_stop_seconds
from integration._support.wire import Reply, Request
_DEFAULT_METADATA: Final = {
@ -411,7 +412,7 @@ def test_arize_otel_v2_c9_upstream_failures_are_recorded(failure_status: int, ga
assert not any(".tool_calls." in key for key in attributes), attributes
@pytest.mark.timeout(180)
@pytest.mark.timeout(4 * graceful_stop_seconds() + 120)
def test_arize_otel_v2_c10_attribute_limit_keeps_tool_prefix(gateway: Gateway, tmp_path: Path) -> None:
marker: Final = "c10-" + uuid.uuid4().hex
calls: Final = _response_tool_calls(marker, ("Paris", "Berlin", "Rome", "Tokyo", "Oslo", "Lima", "Accra", "Delhi"))

View file

@ -56,7 +56,7 @@ pytestmark: Final = pytest.mark.shares_worker_state
_EDGE_DELAY_SECONDS: Final = 5
@pytest.mark.timeout(180)
@pytest.mark.timeout(4 * graceful_stop_seconds() + 120)
def test_K1_policy_edge_restart_mid_burst_keeps_output_observation_fail_open(gateway: Gateway, tmp_path: Path) -> None:
identity: Final = f"logging-scope-k1-{uuid.uuid4().hex}"
marker: Final = uuid.uuid4().hex
@ -215,7 +215,7 @@ def test_K1_policy_edge_restart_mid_burst_keeps_output_observation_fail_open(gat
) == tuple((identity, "logging_only", expected_status) for _ in expected_directions), (index, entries)
@pytest.mark.timeout(180)
@pytest.mark.timeout(4 * graceful_stop_seconds() + 120)
def test_K2_policy_edge_delay_does_not_delay_concurrent_callers(gateway: Gateway, tmp_path: Path) -> None:
identity: Final = f"logging-scope-k2-{uuid.uuid4().hex}"
marker: Final = uuid.uuid4().hex

View file

@ -30,7 +30,7 @@ from _s3_v2_support import (
from integration._support.client import Gateway, JsonValue, Scenario, eventually, object_value
from integration._support.database import read_rows, scratch_database
from integration._support.database_relay import database_relay
from integration._support.process import OwnedProxy, owned_proxy_process
from integration._support.process import OwnedProxy, graceful_stop_seconds, owned_proxy_process
from integration._support.wire import Reply, Request, wire_server
pytestmark: Final = pytest.mark.shares_worker_state
@ -1160,7 +1160,7 @@ def test_s3_v2_hour_worker_kill_mid_burst_keeps_the_other_worker_logging(gateway
assert _outside_layout(objects, "hour") == ()
@pytest.mark.timeout(180)
@pytest.mark.timeout(4 * graceful_stop_seconds() + 120)
def test_s3_v2_hour_proxy_restart_mid_burst_keeps_the_layout_without_duplicates(
gateway: Gateway, tmp_path: Path
) -> None: