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(