coverage for mcp

Author: Zhe Yu

tests/test_mcp.py

@@ -1,6 +1,7 @@
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
+from chromadb.errors import NotFoundError
 from mcp import McpError
 
 from vectorcode.cli_utils import Config
@@ -182,3 +183,25 @@ async def test_mcp_server():
         await mcp_server()
 
         assert mock_add_tool.call_count == 2
+
+
+@pytest.mark.asyncio
+async def test_mcp_server_no_default_collection():
+    with (
+        patch(
+            "vectorcode.mcp_main.find_project_config_dir"
+        ) as mock_find_project_config_dir,
+        patch("vectorcode.mcp_main.load_config_file") as mock_load_config_file,
+        patch("vectorcode.mcp_main.get_client") as mock_get_client,
+        patch("vectorcode.mcp_main.get_collection") as mock_get_collection,
+        patch("mcp.server.fastmcp.FastMCP.add_tool") as mock_add_tool,
+    ):
+        mock_find_project_config_dir.return_value = "/path/to/config"
+        mock_load_config_file.return_value = Config(project_root="/path/to/project")
+        mock_client = AsyncMock()
+        mock_get_client.return_value = mock_client
+        mock_get_collection.side_effect = NotFoundError()
+
+        await mcp_server()
+
+        assert mock_add_tool.call_count == 2