diff --git a/tests/mcp_tests/test_mcp_server.py b/tests/mcp_tests/test_mcp_server.py index e8a1231c6fb..f56e8321b54 100644 --- a/tests/mcp_tests/test_mcp_server.py +++ b/tests/mcp_tests/test_mcp_server.py @@ -1105,9 +1105,16 @@ async def mock_get_allowed_servers(user_auth=None): test_manager.get_allowed_mcp_servers = mock_get_allowed_servers - # Test the method (this tests our second fix) - import asyncio + # Mock _create_mcp_client to return a client that completes immediately + # This avoids network calls while preserving the actual conversion logic + def mock_create_mcp_client(*args, **kwargs): + mock_client = MagicMock() + mock_client.run_with_session = AsyncMock(return_value="ok") + return mock_client + + test_manager._create_mcp_client = mock_create_mcp_client + # Test the method (this tests our second fix) servers_list = await test_manager.get_all_mcp_servers_with_health_and_teams( user_api_key_auth=mock_user_auth )