From 03d4e5d3e58136d5bbc378bf99630b4b7d80d578 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 c614ce2643b..29697d529a1 100644 --- a/tests/test_litellm/litellm_core_utils/test_realtime_streaming.py +++ b/tests/test_litellm/litellm_core_utils/test_realtime_streaming.py @@ -972,11 +972,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():