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

fix: test

Browse source
This commit is contained in:
Yuta Saito 2026-01-02 16:32:33 +09:00
parent 35f9a75d55
commit 89ce8091bb
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/store_model_in_db_tests/test_mcp_servers.py
View file

@ -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

2
tests/test_litellm/proxy/management_endpoints/test_mcp_management_endpoints.py
View file

@ -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",