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 e9c26221580..87e87142a42 100644 --- a/tests/store_model_in_db_tests/test_mcp_servers.py +++ b/tests/store_model_in_db_tests/test_mcp_servers.py @@ -141,6 +141,10 @@ async def test_create_mcp_server_direct(): "litellm.proxy.management_endpoints.mcp_management_endpoints.get_mcp_server", new_callable=mock.AsyncMock, ) as mock_get_server, + mock.patch( + "litellm.proxy.management_endpoints.mcp_management_endpoints._delete_draft_mcp_server", + new_callable=mock.AsyncMock, + ) as mock_delete_draft, mock.patch( "litellm.proxy.management_endpoints.mcp_management_endpoints.global_mcp_server_manager" ) as mock_manager, @@ -209,6 +213,7 @@ async def test_create_mcp_server_direct(): assert result.credentials is None # Verify mocks were called + mock_delete_draft.assert_awaited_once_with(mock_prisma, server_id) mock_get_server.assert_called_once_with(mock_prisma, server_id) mock_create.assert_called_once() mock_manager.add_server.assert_called_once_with(expected_response)