From ea50052e56a3378f3fa3c19f5c1c22ef444500ea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=86=AF=E5=9F=BA=E9=AD=81?= <1412414664@qq.com> Date: Mon, 8 Jun 2026 21:19:45 +0800 Subject: [PATCH] test: update MCP list tools unit path --- tests/mcp_tests/test_mcp_client_unit.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/tests/mcp_tests/test_mcp_client_unit.py b/tests/mcp_tests/test_mcp_client_unit.py index 43260eda1b7..5d408c153b5 100644 --- a/tests/mcp_tests/test_mcp_client_unit.py +++ b/tests/mcp_tests/test_mcp_client_unit.py @@ -14,7 +14,11 @@ sys.path.insert(0, os.path.abspath("../../..")) import litellm.experimental_mcp_client.client as mcp_client_module from litellm.experimental_mcp_client.client import MCPClient from litellm.types.mcp import MCPAuth, MCPTransport -from mcp.types import Tool as MCPTool, CallToolResult as MCPCallToolResult +from mcp.types import ( + CallToolResult as MCPCallToolResult, + ListToolsRequest, + Tool as MCPTool, +) def test_mcp_client_uses_configurable_default_timeout(): @@ -179,14 +183,17 @@ class TestMCPClientUnitTests: ] mock_result = MagicMock() mock_result.tools = mock_tools - mock_session_instance.list_tools.return_value = mock_result + mock_session_instance.send_request.return_value = mock_result client = MCPClient("http://example.com") result = await client.list_tools() assert result == mock_tools mock_session_instance.initialize.assert_called_once() - mock_session_instance.list_tools.assert_called_once() + mock_session_instance.list_tools.assert_not_called() + mock_session_instance.send_request.assert_called_once() + request, _result_type = mock_session_instance.send_request.call_args.args + assert isinstance(request.root, ListToolsRequest) @pytest.mark.asyncio @patch.object(mcp_client_module, "streamable_http_client")