scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

style(tests): wrap MCP health regression setup

Browse source
This commit is contained in:
Joshua Valluru 2026-09-17 09:19:05 -07:00
parent e21db01d67
commit 664b1f16bb
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py
View file

@ -4106,12 +4106,20 @@ async def test_health_discovery_respects_route_restricted_key_grants(
user_role=LitellmUserRoles.INTERNAL_USER,
api_key="test-health-key",
allowed_routes=["/v1/mcp/server", "/v1/mcp/server/health"] if restricted else [],
object_permission=LiteLLM_ObjectPermissionTable(object_permission_id="health-permissions", mcp_servers=list(grants)),
object_permission=LiteLLM_ObjectPermissionTable(
object_permission_id="health-permissions", mcp_servers=list(grants),
),
)
with (
patch.object(mgmt_endpoints, "global_mcp_server_manager", manager), # test-quality-ok: TQ008 inject real registry into legacy route binding
patch.object(mcp_server_manager, "global_mcp_server_manager", manager), # test-quality-ok: TQ008 share real registry with unchanged permission resolver
patch("litellm.proxy.proxy_server.general_settings", {"user_mcp_management_mode": mode}), # test-quality-ok: TQ008 configure mode without mocking authorization
patch.object( # test-quality-ok: TQ008 inject real registry into legacy route binding
mgmt_endpoints, "global_mcp_server_manager", manager,
),
patch.object( # test-quality-ok: TQ008 inject shared registry without mocking permission policy
mcp_server_manager, "global_mcp_server_manager", manager,
),
patch( # test-quality-ok: TQ008 configure mode without mocking authorization
"litellm.proxy.proxy_server.general_settings", {"user_mcp_management_mode": mode},
),
):
result: Final = await mgmt_endpoints.health_check_servers(
server_ids=list(requested) if requested is not None else None,