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

test: add unit testing

Browse source
This commit is contained in:
Krrish Dholakia 2025-10-03 17:27:49 -07:00
parent 6ccb20d69d
commit 3816d3b36f
1 changed files with 33 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server_manager.py
View file

@ -990,6 +990,39 @@ class TestMCPServerManager:
server_ids = [server.server_id for server in result]
assert "invalid-server-1" not in server_ids
def test_add_db_mcp_server_to_registry(self):
"""Test that add_db_mcp_server_to_registry adds a MCP server to the registry"""
manager = MCPServerManager()
server = LiteLLM_MCPServerTable(
**{
"server_id": "4c679a81-acd9-4954-9f84-30b739362498",
"server_name": "edc_mcp_server",
"alias": "edc_mcp_server",
"description": None,
"url": "fake_mcp_url",
"transport": "http",
"auth_type": "none",
"created_at": "2025-09-30T08:28:31.353000Z",
"created_by": "a1248959",
"updated_at": "2025-09-30T08:28:31.353000Z",
"updated_by": "a1248959",
"teams": [],
"mcp_access_groups": [],
"mcp_info": {
"server_name": "edc_mcp_server",
"mcp_server_cost_info": None,
},
"status": "unknown",
"last_health_check": None,
"health_check_error": None,
"command": None,
"args": [],
"env": {},
},
)
manager.add_update_server(server)
assert server.server_id in manager.get_registry()
if __name__ == "__main__":
pytest.main([__file__])