From 05a83e703cb55abb2475c6f2c4d95fdc5a1fd54e Mon Sep 17 00:00:00 2001 From: whning Date: Fri, 4 Sep 2026 19:37:00 +0800 Subject: [PATCH] test: use injected backend websocket --- .../test_litellm/litellm_core_utils/test_realtime_streaming.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_litellm/litellm_core_utils/test_realtime_streaming.py b/tests/test_litellm/litellm_core_utils/test_realtime_streaming.py index e3218d96cb0..09ce0baeef0 100644 --- a/tests/test_litellm/litellm_core_utils/test_realtime_streaming.py +++ b/tests/test_litellm/litellm_core_utils/test_realtime_streaming.py @@ -977,11 +977,10 @@ async def test_provider_path_completed_event_without_guardrails_skips_response_c provider_config=provider_config, model="gemini-2.5-flash", ) - streaming._send_to_backend = AsyncMock() # type: ignore[method-assign] await streaming._handle_provider_config_message("{}") - assert streaming._send_to_backend.await_count == 0 + assert backend_ws.send.await_count == 0 def test_client_session_update_marks_transcription_session():