From c245cae1e91f14b16e74013b37533b0711d94e44 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Thu, 30 Apr 2026 16:53:59 +0530 Subject: [PATCH] test(mcp): patch stateless/stateful managers in concurrency init test Update concurrent session-manager initialization test to patch session_manager_stateless and session_manager_stateful directly, matching initialize_session_managers() behavior and preventing NameError from undefined mocks. Made-with: Cursor --- .../proxy/_experimental/mcp_server/test_mcp_server.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py index 4e3c9083ecd..366e2879c78 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py @@ -865,8 +865,11 @@ async def test_concurrent_initialize_session_managers(): # Mock the session managers to avoid actual MCP initialization with ( patch( - "litellm.proxy._experimental.mcp_server.server.session_manager" - ) as mock_session_manager, + "litellm.proxy._experimental.mcp_server.server.session_manager_stateless" + ) as mock_session_manager_stateless, + patch( + "litellm.proxy._experimental.mcp_server.server.session_manager_stateful" + ) as mock_session_manager_stateful, patch( "litellm.proxy._experimental.mcp_server.server.sse_session_manager" ) as mock_sse_session_manager,