From 89ce8091bb6ceac6e10850319c2ddb94e02dfe2e Mon Sep 17 00:00:00 2001 From: Yuta Saito Date: Fri, 2 Jan 2026 16:32:33 +0900 Subject: [PATCH] fix: test --- tests/store_model_in_db_tests/test_mcp_servers.py | 4 ++-- .../management_endpoints/test_mcp_management_endpoints.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/store_model_in_db_tests/test_mcp_servers.py b/tests/store_model_in_db_tests/test_mcp_servers.py index 8197ce2693b..b9c63256050 100644 --- a/tests/store_model_in_db_tests/test_mcp_servers.py +++ b/tests/store_model_in_db_tests/test_mcp_servers.py @@ -139,7 +139,7 @@ async def test_create_mcp_server_direct(): mock_get_prisma.return_value = mock_prisma # Mock server manager - mock_manager.add_update_server = mock.AsyncMock() + mock_manager.add_server = mock.AsyncMock() mock_manager.reload_servers_from_database = mock.AsyncMock() # Set up test data @@ -195,7 +195,7 @@ async def test_create_mcp_server_direct(): # Verify mocks were called mock_get_server.assert_called_once_with(mock_prisma, server_id) mock_create.assert_called_once() - mock_manager.add_update_server.assert_called_once_with(expected_response) + mock_manager.add_server.assert_called_once_with(expected_response) @pytest.mark.asyncio diff --git a/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py b/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py index cd1a1f5e10d..cc268ab9925 100644 --- a/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py +++ b/tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py @@ -957,7 +957,7 @@ class TestUpdateMCPServer: "litellm.proxy.management_endpoints.mcp_management_endpoints.update_mcp_server", AsyncMock(return_value=updated_server), ) as update_mock, patch( - "litellm.proxy.management_endpoints.mcp_management_endpoints.global_mcp_server_manager.add_update_server", + "litellm.proxy.management_endpoints.mcp_management_endpoints.global_mcp_server_manager.add_server", AsyncMock(), ), patch( "litellm.proxy.management_endpoints.mcp_management_endpoints.global_mcp_server_manager.reload_servers_from_database",