scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-22 00:31:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 51

test(mcp): suppress TQ008 on new MCP routing tests

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Devin AI 2026-09-15 18:49:29 +00:00
parent 608840b823
commit 3ca534b1c0
1 changed files with 21 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -2165,23 +2165,29 @@ async def test_mcp_routing_stashes_peeked_body_for_auth():
stateless_called.append(1)
with (
patch(
patch( # test-quality-ok: the ASGI handler reads auth from a module-level helper; the suite's only seam
"litellm.proxy._experimental.mcp_server.server.extract_mcp_auth_context",
new_callable=AsyncMock,
return_value=(MagicMock(), None, None, None, None, None),
),
patch(
patch( # test-quality-ok: registry is empty in unit tests; key owns one server
"litellm.proxy._experimental.mcp_server.server._get_allowed_mcp_servers",
new_callable=AsyncMock,
return_value=[MagicMock()],
),
patch(
patch( # test-quality-ok: init flag is a module global; no injection seam
"litellm.proxy._experimental.mcp_server.server._SESSION_MANAGERS_INITIALIZED",
True,
),
patch.object(session_manager_stateless, "handle_request", side_effect=stateless_handle),
patch.object(session_manager_stateless, "_server_instances", {}),
patch.object(session_manager_stateful, "_server_instances", {}),
patch.object( # test-quality-ok: session managers are module-level singletons; the suite's only seam
session_manager_stateless, "handle_request", side_effect=stateless_handle
),
patch.object( # test-quality-ok: session managers are module-level singletons; the suite's only seam
session_manager_stateless, "_server_instances", {}
),
patch.object( # test-quality-ok: session managers are module-level singletons; the suite's only seam
session_manager_stateful, "_server_instances", {}
),
):
await handle_streamable_http_mcp(scope, receive, send)
@ -2223,22 +2229,26 @@ async def test_mcp_routing_batch_body_is_not_stashed_for_auth():
send = AsyncMock()
with (
patch(
patch( # test-quality-ok: the ASGI handler reads auth from a module-level helper; the suite's only seam
"litellm.proxy._experimental.mcp_server.server.extract_mcp_auth_context",
new_callable=AsyncMock,
return_value=(MagicMock(), None, None, None, None, None),
),
patch(
patch( # test-quality-ok: registry is empty in unit tests; key owns one server
"litellm.proxy._experimental.mcp_server.server._get_allowed_mcp_servers",
new_callable=AsyncMock,
return_value=[MagicMock()],
),
patch(
patch( # test-quality-ok: init flag is a module global; no injection seam
"litellm.proxy._experimental.mcp_server.server._SESSION_MANAGERS_INITIALIZED",
True,
),
patch.object(session_manager_stateless, "handle_request", new=AsyncMock()),
patch.object(session_manager_stateless, "_server_instances", {}),
patch.object( # test-quality-ok: session managers are module-level singletons; the suite's only seam
session_manager_stateless, "handle_request", new=AsyncMock()
),
patch.object( # test-quality-ok: session managers are module-level singletons; the suite's only seam
session_manager_stateless, "_server_instances", {}
),
):
await handle_streamable_http_mcp(scope, receive, send)