From aa2baa85904194e3261fa88e0f91636386018fb0 Mon Sep 17 00:00:00 2001 From: Tin Chi Lo Date: Tue, 7 Jul 2026 10:55:33 -0700 Subject: [PATCH] test(mcp): pin mock server attrs and new kwarg in single-server listing tests 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 --- tests/mcp_tests/test_mcp_server.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/mcp_tests/test_mcp_server.py b/tests/mcp_tests/test_mcp_server.py index 321a17fbb03..6a6bf390fa6 100644 --- a/tests/mcp_tests/test_mcp_server.py +++ b/tests/mcp_tests/test_mcp_server.py @@ -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"]