From 6996a1789cb594d188d2904b6ddedac288b1adba Mon Sep 17 00:00:00 2001 From: mateo Date: Fri, 10 Jul 2026 04:43:06 +0000 Subject: [PATCH] test(mcp): mock draft cleanup in DB integration test Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/store_model_in_db_tests/test_mcp_servers.py | 5 +++++ 1 file changed, 5 insertions(+) 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)