Add tests

Author: niechen

tests/test_remove.py

@@ -0,0 +1,105 @@
+from unittest.mock import Mock, patch
+
+from click.testing import CliRunner
+
+from mcpm.clients.client_registry import ClientRegistry
+from mcpm.commands.remove import remove
+
+
+def test_remove_server_success(windsurf_manager, monkeypatch):
+    """Test successful server removal"""
+    # Setup mocks
+    monkeypatch.setattr(ClientRegistry, "get_active_client", Mock(return_value="windsurf"))
+    monkeypatch.setattr(ClientRegistry, "get_active_client_manager", Mock(return_value=windsurf_manager))
+    monkeypatch.setattr(ClientRegistry, "get_client_info", Mock(return_value={"name": "windsurf"}))
+
+    # Mock server info
+    mock_server = Mock()
+    mock_server.command = "npx"
+    mock_server.args = ["-y", "@modelcontextprotocol/server-test"]
+    mock_server.env = {"API_KEY": "test-key"}
+    windsurf_manager.get_server = Mock(return_value=mock_server)
+    windsurf_manager.remove_server = Mock(return_value=True)
+
+    # Run the command with force flag to skip confirmation
+    runner = CliRunner()
+    result = runner.invoke(remove, ["server-test", "--force"])
+
+    assert result.exit_code == 0
+    assert "Successfully removed server: server-test" in result.output
+    windsurf_manager.remove_server.assert_called_once_with("server-test")
+
+
+def test_remove_server_not_found(windsurf_manager, monkeypatch):
+    """Test removal of non-existent server"""
+    monkeypatch.setattr(ClientRegistry, "get_active_client", Mock(return_value="windsurf"))
+    monkeypatch.setattr(ClientRegistry, "get_active_client_manager", Mock(return_value=windsurf_manager))
+    monkeypatch.setattr(ClientRegistry, "get_client_info", Mock(return_value={"name": "windsurf"}))
+
+    # Mock server not found
+    windsurf_manager.get_server = Mock(return_value=None)
+
+    runner = CliRunner()
+    result = runner.invoke(remove, ["non-existent-server"])
+
+    assert result.exit_code == 0  # Command exits successfully but with error message
+    assert "Server 'non-existent-server' not found in windsurf" in result.output
+
+
+def test_remove_server_unsupported_client(monkeypatch):
+    """Test removal with unsupported client"""
+    monkeypatch.setattr(ClientRegistry, "get_active_client_manager", Mock(return_value=None))
+    monkeypatch.setattr(ClientRegistry, "get_active_client", Mock(return_value="unsupported"))
+
+    runner = CliRunner()
+    result = runner.invoke(remove, ["server-test"])
+
+    assert result.exit_code == 0  # Command exits successfully but with error message
+    assert "Unsupported active client" in result.output
+
+
+def test_remove_server_cancelled(windsurf_manager, monkeypatch):
+    """Test removal when user cancels the confirmation"""
+    monkeypatch.setattr(ClientRegistry, "get_active_client", Mock(return_value="windsurf"))
+    monkeypatch.setattr(ClientRegistry, "get_active_client_manager", Mock(return_value=windsurf_manager))
+    monkeypatch.setattr(ClientRegistry, "get_client_info", Mock(return_value={"name": "windsurf"}))
+
+    # Mock server info
+    mock_server = Mock()
+    mock_server.command = "npx"
+    mock_server.args = ["-y", "@modelcontextprotocol/server-test"]
+    mock_server.env = {"API_KEY": "test-key"}
+    windsurf_manager.get_server = Mock(return_value=mock_server)
+    windsurf_manager.remove_server = Mock(return_value=True)
+
+    # Run the command without force flag and simulate user cancellation
+    runner = CliRunner()
+    with patch("rich.prompt.Confirm.ask", return_value=False):
+        result = runner.invoke(remove, ["server-test"])
+
+    assert result.exit_code == 0
+    assert "Removal cancelled" in result.output
+    windsurf_manager.remove_server.assert_not_called()
+
+
+def test_remove_server_failure(windsurf_manager, monkeypatch):
+    """Test removal when the removal operation fails"""
+    monkeypatch.setattr(ClientRegistry, "get_active_client", Mock(return_value="windsurf"))
+    monkeypatch.setattr(ClientRegistry, "get_active_client_manager", Mock(return_value=windsurf_manager))
+    monkeypatch.setattr(ClientRegistry, "get_client_info", Mock(return_value={"name": "windsurf"}))
+
+    # Mock server info
+    mock_server = Mock()
+    mock_server.command = "npx"
+    mock_server.args = ["-y", "@modelcontextprotocol/server-test"]
+    mock_server.env = {"API_KEY": "test-key"}
+    windsurf_manager.get_server = Mock(return_value=mock_server)
+    windsurf_manager.remove_server = Mock(return_value=False)
+
+    # Run the command with force flag
+    runner = CliRunner()
+    result = runner.invoke(remove, ["server-test", "--force"])
+
+    assert result.exit_code == 0  # Command exits successfully but with error message
+    assert "Failed to remove server 'server-test'" in result.output
+    windsurf_manager.remove_server.assert_called_once_with("server-test")

tests/test_search.py

@@ -0,0 +1,194 @@
+from unittest.mock import Mock
+
+from click.testing import CliRunner
+
+from mcpm.commands.search import search
+
+
+def test_search_all_servers(monkeypatch):
+    """Test searching for all servers without a query"""
+    # Mock repository manager
+    mock_repo_manager = Mock()
+    mock_repo_manager.search_servers = Mock(
+        return_value=[
+            {
+                "name": "server1",
+                "display_name": "Server One",
+                "description": "First test server",
+                "categories": ["category1"],
+                "tags": ["tag1", "tag2"],
+            },
+            {
+                "name": "server2",
+                "display_name": "Server Two",
+                "description": "Second test server",
+                "categories": ["category2"],
+                "tags": ["tag3"],
+            },
+        ]
+    )
+    monkeypatch.setattr("mcpm.commands.search.repo_manager", mock_repo_manager)
+
+    # Run the command
+    runner = CliRunner()
+    result = runner.invoke(search, [])
+
+    assert result.exit_code == 0
+    assert "Listing all available MCP servers" in result.output
+    assert "Server One" in result.output
+    assert "Server Two" in result.output
+    assert "Found 2 server(s) matching search criteria" in result.output
+    mock_repo_manager.search_servers.assert_called_once_with(None)
+
+
+def test_search_with_query(monkeypatch):
+    """Test searching with a specific query"""
+    # Mock repository manager
+    mock_repo_manager = Mock()
+    mock_repo_manager.search_servers = Mock(
+        return_value=[
+            {
+                "name": "github-server",
+                "display_name": "GitHub Server",
+                "description": "GitHub integration server",
+                "categories": ["integration"],
+                "tags": ["github", "api"],
+            }
+        ]
+    )
+    monkeypatch.setattr("mcpm.commands.search.repo_manager", mock_repo_manager)
+
+    # Run the command
+    runner = CliRunner()
+    result = runner.invoke(search, ["github"])
+
+    assert result.exit_code == 0
+    assert "Searching for MCP servers matching 'github'" in result.output
+    assert "GitHub Server" in result.output
+    assert "Found 1 server(s) matching search criteria" in result.output
+    mock_repo_manager.search_servers.assert_called_once_with("github")
+
+
+def test_search_no_results(monkeypatch):
+    """Test searching with no results"""
+    # Mock repository manager
+    mock_repo_manager = Mock()
+    mock_repo_manager.search_servers = Mock(return_value=[])
+    monkeypatch.setattr("mcpm.commands.search.repo_manager", mock_repo_manager)
+
+    # Run the command
+    runner = CliRunner()
+    result = runner.invoke(search, ["nonexistent"])
+
+    assert result.exit_code == 0
+    assert "Searching for MCP servers matching 'nonexistent'" in result.output
+    assert "No matching MCP servers found" in result.output
+    mock_repo_manager.search_servers.assert_called_once_with("nonexistent")
+
+
+def test_search_detailed_view(monkeypatch):
+    """Test searching with detailed view"""
+    # Mock repository manager
+    mock_repo_manager = Mock()
+    mock_repo_manager.search_servers = Mock(
+        return_value=[
+            {
+                "name": "test-server",
+                "display_name": "Test Server",
+                "description": "A test server",
+                "categories": ["test"],
+                "tags": ["example"],
+                "license": "MIT",
+                "author": {"name": "Test Author", "email": "[email protected]"},
+                "installation": {"package": "test-package"},
+                "installations": {
+                    "npm": {
+                        "type": "npm",
+                        "description": "NPM installation",
+                        "recommended": True,
+                        "command": "npx",
+                        "args": ["-y", "test-package"],
+                        "dependencies": ["dependency1"],
+                        "env": {"API_KEY": "${API_KEY}"},
+                    }
+                },
+                "examples": [{"title": "Example Usage", "description": "How to use this server"}],
+            }
+        ]
+    )
+    monkeypatch.setattr("mcpm.commands.search.repo_manager", mock_repo_manager)
+
+    # Run the command with detailed flag
+    runner = CliRunner()
+    result = runner.invoke(search, ["--detailed"])
+
+    assert result.exit_code == 0
+    assert "Test Server" in result.output
+    assert "A test server" in result.output
+    assert "Server Information:" in result.output
+    assert "Installation Details:" in result.output
+    assert "Example:" in result.output
+    assert "Found 1 server(s) matching search criteria" in result.output
+    mock_repo_manager.search_servers.assert_called_once_with(None)
+
+
+def test_search_error_handling(monkeypatch):
+    """Test error handling during search"""
+    # Mock repository manager to raise an exception
+    mock_repo_manager = Mock()
+    mock_repo_manager.search_servers = Mock(side_effect=Exception("Test error"))
+    monkeypatch.setattr("mcpm.commands.search.repo_manager", mock_repo_manager)
+
+    # Run the command
+    runner = CliRunner()
+    result = runner.invoke(search, [])
+
+    assert result.exit_code == 0
+    assert "Error searching for servers: Test error" in result.output
+    mock_repo_manager.search_servers.assert_called_once_with(None)
+
+
+def test_search_with_query_and_detailed(monkeypatch):
+    """Test searching with both a query and detailed view"""
+    # Mock repository manager
+    mock_repo_manager = Mock()
+    mock_repo_manager.search_servers = Mock(
+        return_value=[
+            {
+                "name": "test-server",
+                "display_name": "Test Server",
+                "description": "A test server",
+                "categories": ["test"],
+                "tags": ["example"],
+                "license": "MIT",
+                "author": {"name": "Test Author", "email": "[email protected]"},
+                "installation": {"package": "test-package"},
+                "installations": {
+                    "npm": {
+                        "type": "npm",
+                        "description": "NPM installation",
+                        "recommended": True,
+                        "command": "npx",
+                        "args": ["-y", "test-package"],
+                        "dependencies": ["dependency1"],
+                        "env": {"API_KEY": "${API_KEY}"},
+                    }
+                },
+                "examples": [{"title": "Example Usage", "description": "How to use this server"}],
+            }
+        ]
+    )
+    monkeypatch.setattr("mcpm.commands.search.repo_manager", mock_repo_manager)
+
+    # Run the command with both query and detailed flag
+    runner = CliRunner()
+    result = runner.invoke(search, ["test", "--detailed"])
+
+    assert result.exit_code == 0
+    assert "Searching for MCP servers matching 'test'" in result.output
+    assert "Test Server" in result.output
+    assert "Server Information:" in result.output
+    assert "Installation Details:" in result.output
+    assert "Example:" in result.output
+    assert "Found 1 server(s) matching search criteria" in result.output
+    mock_repo_manager.search_servers.assert_called_once_with("test")