From e81a958e90f60df7ad52c8ca7d5204fa7b1e45f1 Mon Sep 17 00:00:00 2001 From: kigland Date: Mon, 26 Oct 2026 15:25:34 +0800 Subject: [PATCH] test(mcp): model listing post-processing in session mocks --- tests/unit/proxy/mcp_server/test_mcp_client_unit.py | 3 +++ tests/unit/proxy/mcp_server/test_rest_endpoints.py | 1 + 2 files changed, 4 insertions(+) diff --git a/tests/unit/proxy/mcp_server/test_mcp_client_unit.py b/tests/unit/proxy/mcp_server/test_mcp_client_unit.py index 7e4601211ef..110fdfac12d 100644 --- a/tests/unit/proxy/mcp_server/test_mcp_client_unit.py +++ b/tests/unit/proxy/mcp_server/test_mcp_client_unit.py @@ -187,6 +187,7 @@ class TestMCPClientUnitTests: ) ] mock_session_instance.send_request.return_value = ListToolsResult(tools=mock_tools) + mock_session_instance._absorb_tool_listing = MagicMock(side_effect=lambda result, **_kwargs: result) client = MCPClient("http://example.com") result = await client.list_tools() @@ -231,6 +232,7 @@ class TestMCPClientUnitTests: ListToolsResult(tools=first_page_tools, nextCursor="page-2"), ListToolsResult(tools=[second_page_tool]), ] + mock_session_instance._absorb_tool_listing = MagicMock(side_effect=lambda result, **_kwargs: result) client = MCPClient("http://example.com") result = await client.list_tools() @@ -272,6 +274,7 @@ class TestMCPClientUnitTests: ), RuntimeError("transient upstream failure"), ] + mock_session_instance._absorb_tool_listing = MagicMock(side_effect=lambda result, **_kwargs: result) client = MCPClient("http://example.com") result = await client.list_tools() diff --git a/tests/unit/proxy/mcp_server/test_rest_endpoints.py b/tests/unit/proxy/mcp_server/test_rest_endpoints.py index e05dd71a515..30bb4de615b 100644 --- a/tests/unit/proxy/mcp_server/test_rest_endpoints.py +++ b/tests/unit/proxy/mcp_server/test_rest_endpoints.py @@ -1637,6 +1637,7 @@ class TestListToolsRestAPI: ] ), ] + mock_session_instance._absorb_tool_listing = MagicMock(side_effect=lambda result, **_kwargs: result) mock_session_ctx.__aenter__ = AsyncMock(return_value=mock_session_instance) mock_session_ctx.__aexit__ = AsyncMock(return_value=None) monkeypatch.setattr(