Skip to content

Commit be5010a

Browse files
committed
test: fix tests
1 parent 4f15d12 commit be5010a

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

tests/mcp_tests/test_mcp_server.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -707,9 +707,7 @@ def mock_client_constructor(*args, **kwargs):
707707

708708
assert isinstance(response, dict)
709709
assert len(response["tools"]) == 1
710-
# The server should use the server_name as prefix since no alias is provided
711-
expected_prefix = "test_server"
712-
assert response["tools"][0].name == f"{expected_prefix}-test_tool"
710+
assert response["tools"][0].name == "test_tool"
713711
finally:
714712
# Restore original state
715713
global_mcp_server_manager.registry = {}
@@ -800,7 +798,7 @@ def mock_get_server_by_id(server_id):
800798
)
801799
mock_manager_2.get_mcp_server_by_id = mock_get_server_by_id
802800
mock_manager_2._get_tools_from_server = AsyncMock(
803-
side_effect=lambda server, mcp_auth_header=None, extra_headers=None: (
801+
side_effect=lambda server, mcp_auth_header=None, extra_headers=None, add_prefix=False: (
804802
[mock_tool_1] if server.server_id == "server1_id" else [mock_tool_2]
805803
)
806804
)
@@ -1713,6 +1711,7 @@ async def test_get_tools_for_single_server():
17131711
mock_manager._get_tools_from_server.assert_called_once_with(
17141712
server=mock_server,
17151713
mcp_auth_header="Bearer test_token",
1714+
add_prefix=False,
17161715
)
17171716

17181717
# Verify the result

0 commit comments

Comments
 (0)