scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-14 23:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 156

test(mcp): pin mock server attrs and new kwarg in single-server listing tests

Browse source 
The MagicMock servers in tests/mcp_tests leaked auto-generated short_prefix
and server_name mocks into the prefix normalization added for the disabled
tool annotation, and the call assertion was missing the new
drop_overlong_names kwarg
This commit is contained in:
Tin Chi Lo 2026-07-07 10:55:33 -07:00
parent fed5479113
commit aa2baa8590
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/mcp_tests/test_mcp_server.py
View file

@ -1946,6 +1946,8 @@ async def test_get_tools_for_single_server():
mock_server.mcp_info = {"server_name": "zapier"}
mock_server.server_id = "zapier_id"
mock_server.alias = "zapier_alias"
mock_server.server_name = "zapier"
mock_server.short_prefix = None
mock_server.allowed_tools = None
mock_server.disallowed_tools = None
@ -1974,6 +1976,7 @@ async def test_get_tools_for_single_server():
add_prefix=False,
raw_headers=None,
user_api_key_auth=None,
drop_overlong_names=False,
)
# Verify the result
@ -1998,6 +2001,9 @@ async def test_get_tools_for_single_server_applies_disallowed_tools_without_allo
mock_server.mcp_info = {"server_name": "zapier"}
mock_server.name = "zapier"
mock_server.server_id = "zapier"
mock_server.alias = "zapier"
mock_server.server_name = "zapier"
mock_server.short_prefix = None
mock_server.allowed_tools = None
mock_server.disallowed_tools = ["send_email"]