scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

test_get_tools_from_mcp_servers

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-22 09:55:14 -08:00
parent b2812af0a0
commit fc0eac2d10
1 changed files with 11 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
tests/mcp_tests/test_mcp_server.py
View file

@ -789,7 +789,7 @@ async def test_get_tools_from_mcp_servers():
mock_manager.get_allowed_mcp_servers = AsyncMock(
return_value=["server1_id", "server2_id"]
)
mock_manager.get_mcp_servers_from_ids = MagicMock(return_value=[mock_server_1, mock_server_2])
mock_manager.get_mcp_server_by_id = lambda server_id: mock_server_1 if server_id == "server1_id" else mock_server_2
mock_manager._get_tools_from_server = AsyncMock(return_value=[mock_tool_1])
with patch(
@ -839,7 +839,7 @@ async def test_get_tools_from_mcp_servers():
mock_manager.get_allowed_mcp_servers = AsyncMock(
return_value=["server1_id", "server2_id", "server3_id"]
)
mock_manager.get_mcp_servers_from_ids = MagicMock(return_value=[mock_server_1, mock_server_2, mock_server_3])
mock_manager.get_mcp_server_by_id = lambda server_id: mock_server_1 if server_id == "server1_id" else (mock_server_2 if server_id == "server2_id" else mock_server_3)
mock_manager._get_tools_from_server = AsyncMock(return_value=[mock_tool_1])
with patch(
@ -2040,7 +2040,7 @@ async def test_filter_tools_by_allowed_tools_integration():
mock_manager.get_allowed_mcp_servers = AsyncMock(
return_value=["test-server-123"]
)
mock_manager.get_mcp_servers_from_ids = MagicMock(return_value=[mock_server])
mock_manager.get_mcp_server_by_id = lambda server_id: mock_server
# Mock the _get_tools_from_server method to return all tools
mock_manager._get_tools_from_server = AsyncMock(return_value=mock_tools)
@ -2078,7 +2078,9 @@ async def test_filter_tools_by_allowed_tools_integration():
# Verify the manager methods were called correctly
mock_manager.get_allowed_mcp_servers.assert_called_once_with(mock_user_auth)
mock_manager.get_mcp_servers_from_ids.assert_called_once_with(["test-server-123"])
# Note: get_mcp_server_by_id is now called for each server ID instead of batch
# Verify it was called with the correct server ID
assert mock_manager.get_mcp_server_by_id.call_count > 0
mock_manager._get_tools_from_server.assert_called_once()
@ -2148,7 +2150,7 @@ async def test_filter_tools_by_disallowed_tools_integration():
mock_manager.get_allowed_mcp_servers = AsyncMock(
return_value=["test-server-456"]
)
mock_manager.get_mcp_servers_from_ids = MagicMock(return_value=[mock_server])
mock_manager.get_mcp_server_by_id = lambda server_id: mock_server
# Mock the _get_tools_from_server method to return all tools
mock_manager._get_tools_from_server = AsyncMock(return_value=mock_tools)
@ -2185,7 +2187,9 @@ async def test_filter_tools_by_disallowed_tools_integration():
# Verify the manager methods were called correctly
mock_manager.get_allowed_mcp_servers.assert_called_once_with(mock_user_auth)
mock_manager.get_mcp_servers_from_ids.assert_called_once_with(["test-server-456"])
# Note: get_mcp_server_by_id is now called for each server ID instead of batch
# Verify it was called with the correct server ID
assert mock_manager.get_mcp_server_by_id.call_count > 0
mock_manager._get_tools_from_server.assert_called_once()
@ -2242,7 +2246,7 @@ async def test_filter_tools_no_restrictions_integration():
mock_manager.get_allowed_mcp_servers = AsyncMock(
return_value=["test-server-000"]
)
mock_manager.get_mcp_servers_from_ids = MagicMock(return_value=[mock_server])
mock_manager.get_mcp_server_by_id = lambda server_id: mock_server
# Mock the _get_tools_from_server method to return all tools
mock_manager._get_tools_from_server = AsyncMock(return_value=mock_tools)