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

error

Browse source
This commit is contained in:
Yug 2026-05-01 14:04:52 +05:30
parent 6ac2f957e5
commit fc53adb2ce
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -627,6 +627,7 @@ async def test_get_tools_from_mcp_servers_continues_when_one_server_fails():
extra_headers=None,
add_prefix=True,
raw_headers=None,
**kwargs,
):
if server.name == "working_server":
# Working server returns tools
@ -732,6 +733,7 @@ async def test_get_tools_from_mcp_servers_handles_all_servers_failing():
extra_headers=None,
add_prefix=True,
raw_headers=None,
**kwargs,
):
# All servers fail
raise Exception(f"Server {server.name} connection failed")
@ -1173,6 +1175,7 @@ async def test_list_tools_single_server_unprefixed_names():
extra_headers=None,
add_prefix=False,
raw_headers=None,
**kwargs,
):
tool = MagicMock()
tool.name = f"{server.alias}-toolA" if add_prefix else "toolA"
@ -1254,6 +1257,7 @@ async def test_list_tools_multiple_servers_prefixed_names():
extra_headers=None,
add_prefix=True,
raw_headers=None,
**kwargs,
):
tool = MagicMock()
# When multiple servers, add_prefix should be True -> prefixed names
@ -1520,6 +1524,7 @@ async def test_list_tools_filters_by_key_team_permissions():
extra_headers=None,
add_prefix=False,
raw_headers=None,
**kwargs,
):
# Return 4 tools, but only 2 should be allowed
tool1 = MagicMock()
@ -1629,6 +1634,7 @@ async def test_list_tools_with_team_tool_permissions_inheritance():
extra_headers=None,
add_prefix=False,
raw_headers=None,
**kwargs,
):
# Return 4 tools
tool1 = MagicMock()
@ -1724,6 +1730,7 @@ async def test_list_tools_with_no_tool_permissions_shows_all():
extra_headers=None,
add_prefix=False,
raw_headers=None,
**kwargs,
):
# Return 3 tools
tool1 = MagicMock()
@ -1822,6 +1829,7 @@ async def test_list_tools_strips_prefix_when_matching_permissions():
extra_headers=None,
add_prefix=True,
raw_headers=None,
**kwargs,
):
# Return tools WITH prefix (as they come from MCP server)
tool1 = MagicMock()