scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

test(mcp): patch the scoped-connect resolver the preemptive challenge reads

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yucheng 2026-09-30 10:32:13 +00:00
parent 674356d3d8
commit 23eb6bd8a5
1 changed files with 32 additions and 0 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_stale_session.py
View file

@ -647,6 +647,10 @@ async def test_per_user_oauth_missing_stored_token_returns_preemptive_401():
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=oauth_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=oauth_server,
),
patch.object(
session_manager_stateless,
"handle_request",
@ -735,6 +739,10 @@ async def test_admitted_subject_missing_stored_token_challenged_with_resource_me
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=oauth_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=oauth_server,
),
patch.object(
session_manager_stateless,
"handle_request",
@ -1030,6 +1038,10 @@ async def test_per_user_oauth_with_stored_token_skips_preemptive_401():
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=oauth_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=oauth_server,
),
patch.object(
session_manager_stateless,
"handle_request",
@ -1133,6 +1145,10 @@ async def test_handle_streamable_http_mcp_delegated_server_without_token_returns
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=delegated_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=delegated_server,
),
patch.object(
session_manager_stateful,
"handle_request",
@ -1221,6 +1237,10 @@ async def test_handle_streamable_http_mcp_token_exchange_without_subject_returns
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=obo_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=obo_server,
),
patch.object(
session_manager_stateful,
"handle_request",
@ -1320,6 +1340,10 @@ async def test_handle_streamable_http_mcp_oauth_delegate_without_token_returns_g
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=od_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=od_server,
),
patch.object(
session_manager_stateful,
"handle_request",
@ -1577,6 +1601,10 @@ async def test_handle_streamable_http_mcp_true_passthrough_without_token_surface
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=tp_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=tp_server,
),
patch.object(
session_manager_stateful,
"handle_request",
@ -1645,6 +1673,10 @@ async def test_handle_streamable_http_mcp_true_passthrough_dcr_bridge_challenges
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_by_name",
return_value=bridge_server,
),
patch(
"litellm.proxy._experimental.mcp_server.operations.global_mcp_server_manager.get_mcp_server_answering_to",
return_value=bridge_server,
),
patch.object(
session_manager_stateful,
"handle_request",