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

test(mcp): seed tool ownership from the specific server under test

Browse source 
The call_tool and list_tools mock tests seeded a tool's owning-server set from every id
in the registry. That holds only while the fixture has exactly one server; adding a second
would make the tool look multi-owned and resolve as ambiguous, failing the test for an
unrelated reason. Each site now derives the id from the server it actually loaded, matching
the pattern already used in the alias-prefixing tests and the logging tests.
This commit is contained in:
Tin Chi Lo 2026-07-20 18:57:08 -07:00
parent cd3a922732
commit b0908d5345
1 changed files with 14 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
tests/mcp_tests/test_mcp_server.py
View file

@ -226,17 +226,20 @@ async def test_mcp_http_transport_list_tools_mock():
# Verify tool mapping was updated
expected_prefix = "test_http_server"
test_http_server = test_manager.get_mcp_server_by_name("test_http_server")
assert test_http_server is not None
expected_owner = frozenset({test_http_server.server_id})
assert (
test_manager.tool_name_to_mcp_server_ids_mapping[
f"{expected_prefix}-gmail_send_email"
]
== frozenset(allowed_server_ids)
== expected_owner
)
assert (
test_manager.tool_name_to_mcp_server_ids_mapping[
f"{expected_prefix}-calendar_create_event"
]
== frozenset(allowed_server_ids)
== expected_owner
)
@ -281,8 +284,10 @@ async def test_mcp_http_transport_call_tool_mock():
)
# Manually set up tool mapping (normally done by list_tools)
test_http_server = test_manager.get_mcp_server_by_name("test_http_server")
assert test_http_server is not None
test_manager.tool_name_to_mcp_server_ids_mapping["gmail_send_email"] = frozenset(
test_manager.get_registry().keys()
{test_http_server.server_id}
)
# Call the tool
@ -347,8 +352,10 @@ async def test_mcp_http_transport_call_tool_error_mock():
)
# Manually set up tool mapping
test_http_server = test_manager.get_mcp_server_by_name("test_http_server")
assert test_http_server is not None
test_manager.tool_name_to_mcp_server_ids_mapping["gmail_send_email"] = frozenset(
test_manager.get_registry().keys()
{test_http_server.server_id}
)
# Call the tool with invalid data
@ -389,8 +396,10 @@ async def test_mcp_http_transport_tool_not_found():
)
# Mapping populated for this server but not for the requested tool
test_http_server = test_manager.get_mcp_server_by_name("test_http_server")
assert test_http_server is not None
test_manager.tool_name_to_mcp_server_ids_mapping["gmail_send_email"] = frozenset(
test_manager.get_registry().keys()
{test_http_server.server_id}
)
# Try to call a tool that doesn't exist in mapping