scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 75

test(mcp): patch stateless/stateful managers in concurrency init test

Browse source 
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
This commit is contained in:
Sameer Kankute 2026-04-30 16:53:59 +05:30
parent 1a2aa62c0f
commit c245cae1e9
No known key found for this signature in database
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py
View file

@ -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,