From 97b7fbfc3f9cb040584fe220a276ff749f103097 Mon Sep 17 00:00:00 2001
From: Nate Armstrong <narmstrong@berri.ai>
Date: Thu, 12 Nov 2026 10:44:13 -0800
Subject: [PATCH] test(providers): expect rust for anthropic messages after
 rust-only cutover (#51351)

Anthropic /v1/messages always runs on Rust since #51119, so the safeguards
contracts no longer assert those calls stay on Python, and the idle reconnect
stream row expects the Rust route header like the keepalive row after #51346.
Vertex and DeepSeek safeguards calls keep their Python assertions.
---
 .../providers/test_rust_anthropic_safeguards_beta_wire.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/tests/integration/providers/test_rust_anthropic_safeguards_beta_wire.py b/tests/integration/providers/test_rust_anthropic_safeguards_beta_wire.py
index 2941f8003ee..7a5431489db 100644
--- a/tests/integration/providers/test_rust_anthropic_safeguards_beta_wire.py
+++ b/tests/integration/providers/test_rust_anthropic_safeguards_beta_wire.py
@@ -155,7 +155,6 @@ def test_sdk_messages_with_safeguards_go_upstream_with_the_dangerous_tool_use_be
             extra_body={"safeguards": list(SAFEGUARDS), "tools": [dict(BASH_TOOL)]},
         )
     message: Final = raw.parse()
-    assert not rust_routed(raw.headers), dict(raw.headers)
     received: Final = json_object(message.model_dump_json())
     assert message.id == f"msg_{marker}", received
     assert received.get("safeguard_results") == list(SAFEGUARD_RESULTS), received
@@ -192,7 +191,6 @@ async def test_sdk_streamed_messages_with_safeguards_go_upstream_with_the_beta(
         ) as stream:
             text: Final = "".join([piece async for piece in stream.text_stream])
             final: Final = await stream.get_final_message()
-            assert not rust_routed(stream.response.headers), dict(stream.response.headers)
     assert (final.id, text) == (f"msg_{marker}", REPLY_TEXT), (final, text)
     sent: Final = one_delivery(rig.wire, marker)
     assert (sent.target, sent.beta, sent.safeguards) == (
@@ -209,7 +207,6 @@ def test_raw_messages_merge_the_safeguards_beta_into_the_client_betas(rig: _Rig,
     body: Final = safeguarded(messages_body(ANTHROPIC_DEPLOYMENT, marker, stream=stream), list(SAFEGUARDS))
     response: Final = _post(rig.gateway, body, beta=CLIENT_BETAS)
     assert response.status_code == 200, response.text
-    assert not rust_routed(response.headers), dict(response.headers)
     assert f"msg_{marker}" in response.text and "safeguard_results" in response.text, response.text
     sent: Final = one_delivery(rig.wire, marker)
     assert (sent.target, sent.beta) == (ANTHROPIC_TARGET, MERGED_BETAS), sent
@@ -221,7 +218,6 @@ def test_null_or_absent_safeguards_add_no_beta(rig: _Rig, extra: Mapping[str, Js
     marker: Final = new_marker()
     response: Final = _post(rig.gateway, {**messages_body(ANTHROPIC_DEPLOYMENT, marker), **extra})
     assert response.status_code == 200, response.text
-    assert not rust_routed(response.headers), dict(response.headers)
     assert f"msg_{marker}" in response.text and "safeguard_results" not in response.text, response.text
     sent: Final = one_delivery(rig.wire, marker)
     assert (sent.beta, sent.safeguards) == (None, None), sent
@@ -253,7 +249,6 @@ def test_malformed_safeguards_still_add_the_beta_and_relay_the_vendor_verdict(
     control: Final = new_marker()
     after: Final = _post(rig.gateway, messages_body(ANTHROPIC_DEPLOYMENT, control))
     assert after.status_code == 200, after.text
-    assert not rust_routed(after.headers), dict(after.headers)
     assert one_delivery(rig.wire, control).beta is None
     assert rig.gateway.request("GET", "/health/liveliness").status_code == 200
 
@@ -276,7 +271,6 @@ def test_duplicated_safeguards_keys_resolve_to_the_last_value(rig: _Rig, last: s
         },
     )
     assert response.status_code == 200, response.text
-    assert not rust_routed(response.headers), dict(response.headers)
     assert ("safeguard_results" in response.text) == (last == "list"), response.text
     sent: Final = one_delivery(rig.wire, marker)
     assert sent.beta == (DANGEROUS_TOOL_USE if last == "list" else None), sent
@@ -370,7 +364,7 @@ def _wave(pool: ThreadPoolExecutor, gateway: Gateway, markers: tuple[str, ...],
 def _assert_served(answer: _Answer) -> None:
     assert answer.status == 200, answer
     assert REPLY_TEXT in answer.text, answer
-    assert not answer.rust, answer
+    assert not answer.rust or answer.path == "/v1/messages", answer
     if answer.path == "/v1/messages":
         assert f"msg_{answer.marker}" in answer.text and "safeguard_results" in answer.text, answer