scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-03 02:22:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 140

update mcp tools unit test

Browse source
This commit is contained in:
kigland 2026-06-07 11:06:48 +08:00
parent cc66bda2ce
commit 0a0e4fea0d
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/mcp_tests/test_mcp_client_unit.py
View file

@ -179,14 +179,19 @@ class TestMCPClientUnitTests:
]
mock_result = MagicMock()
mock_result.tools = mock_tools
mock_session_instance.list_tools.return_value = mock_result
async def send_request(_request, result_type):
assert result_type is mcp_client_module._LenientListToolsResult
return mock_result
mock_session_instance.send_request = AsyncMock(side_effect=send_request)
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.send_request.assert_awaited_once()
@pytest.mark.asyncio
@patch.object(mcp_client_module, "streamable_http_client")