From eea7cdb6a97283b4316085dca54a54b3ca408b07 Mon Sep 17 00:00:00 2001 From: yucheng Date: Fri, 2 Oct 2026 00:22:33 +0000 Subject: [PATCH] test(straiker): assert a saved api_version v1 with an sk_agt_ key routes to v3 Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- .../test_straiker_v3_platform.py | 22 +++++++++++++------ 1 file changed, 15 insertions(+), 7 deletions(-) diff --git a/tests/integration/observability/test_straiker_v3_platform.py b/tests/integration/observability/test_straiker_v3_platform.py index c4d34b1a9a71..f6f85ea62dbb 100644 --- a/tests/integration/observability/test_straiker_v3_platform.py +++ b/tests/integration/observability/test_straiker_v3_platform.py @@ -784,13 +784,21 @@ def test_v1_post_call_sends_response_envelope(rig: Rig) -> None: # E: explicit api_version v1 with a v3-shaped key follows the configuration, not the key -def test_explicit_api_version_v1_overrides_key_prefix(rig: Rig) -> None: - marker: Final = rig.marker() - response: Final = _chat(rig, "explicit " + marker, guardrails=["straiker-v3-as-v1"]) - assert response.status_code == 200, response.text - calls: Final = _v1_calls(rig, marker, V3_KEY) - assert len(calls) == 1, rig.sink_calls(marker) - assert calls[0].headers["x-straiker-webhook-format"] == "litellm" +def test_saved_api_version_v1_with_v3_key_routes_to_v3_not_the_v1_webhook(rig: Rig) -> None: + allowed_marker: Final = rig.marker() + allowed: Final = _chat(rig, "saved v1 " + allowed_marker, guardrails=["straiker-v3-as-v1"]) + assert allowed.status_code == 200, allowed.text + assert len(_v3_request_calls(rig, allowed_marker, agent=None)) == 1, rig.sink_calls(allowed_marker) + assert _v1_calls(rig, allowed_marker, V3_KEY) == () + assert len(rig.provider_calls(allowed_marker, rig.provider_drain())) == 1 + + blocked_marker: Final = rig.marker() + blocked: Final = _chat(rig, f"{STRAY_V3_BLOCK_MARK} {blocked_marker}", guardrails=["straiker-v3-as-v1"]) + assert blocked.status_code == 400, blocked.text + assert blocked.json()["error"]["message"] == BLOCK_MESSAGE, blocked.text + assert len(_v3_request_calls(rig, blocked_marker, agent=None)) == 1, rig.sink_calls(blocked_marker) + assert _v1_calls(rig, blocked_marker, V3_KEY) == () + assert rig.provider_calls(blocked_marker, rig.provider_drain()) == () def test_stray_api_version_with_v3_key_still_enforces_on_v3(rig: Rig) -> None: