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 84

Merge pull request #21536 from BerriAI/fix/mcp-test-failures-from-user-permissions

Browse source 
fix(tests): update MCP tests broken by user permissions commit (#21462)
This commit is contained in:
jquinter 2026-02-19 02:17:49 -03:00 • committed by GitHub
commit 53cfb93a06
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tests/test_litellm/proxy/_experimental/mcp_server/test_jwt_mcp_enforcement.py
View file

@ -184,6 +184,7 @@ async def test_e2e_jwt_team_mcp_permissions_enforced(monkeypatch):
proxy_server_module.prisma_client = MagicMock() # Mock prisma client
proxy_server_module.user_api_key_cache = DualCache()
proxy_server_module.proxy_logging_obj = MagicMock()
proxy_server_module.general_settings = {}
monkeypatch.setitem(sys.modules, "litellm.proxy.proxy_server", proxy_server_module)
# Team "ABC" has MCP servers assigned via object_permission
@ -389,6 +390,7 @@ async def test_e2e_jwt_team_mcp_key_intersection(monkeypatch):
proxy_server_module.prisma_client = MagicMock()
proxy_server_module.user_api_key_cache = DualCache()
proxy_server_module.proxy_logging_obj = MagicMock()
proxy_server_module.general_settings = {}
monkeypatch.setitem(sys.modules, "litellm.proxy.proxy_server", proxy_server_module)
# Team MCP servers

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

@ -332,7 +332,7 @@ async def test_mcp_get_prompt_success():
mock_manager.get_prompt_from_server = AsyncMock(return_value=prompt_result)
result = await mcp_get_prompt(
name="hello",
name="server_a-hello", # prefixed name since server prefixes are always added
arguments={"foo": "bar"},
user_api_key_auth=user_api_key_auth,
)
@ -1006,7 +1006,7 @@ async def test_oauth2_headers_passed_to_mcp_client():
@pytest.mark.asyncio
async def test_list_tools_single_server_unprefixed_names():
"""When only one MCP server is allowed, list tools should return unprefixed names."""
"""When only one MCP server is allowed, list tools should return prefixed names (server prefix is always added)."""
try:
from litellm.proxy._experimental.mcp_server.server import (
_get_tools_from_mcp_servers,
@ -1063,9 +1063,9 @@ async def test_list_tools_single_server_unprefixed_names():
mcp_server_auth_headers=None,
)
# Should be unprefixed since only one server is allowed
# Server prefix is always added regardless of number of allowed servers
assert len(tools) == 1
assert tools[0].name == "toolA"
assert tools[0].name == "zapier-toolA"
@pytest.mark.asyncio