MCP Claude Code

"""Tests for the server module.""" from unittest.mock import MagicMock, patch import pytest from mcp_claude_code.server import ClaudeCodeServer class TestClaudeCodeServer: """Test the ClaudeCodeServer class.""" @pytest.fixture def server(self) -> tuple[ClaudeCodeServer, MagicMock]: """Create a ClaudeCodeServer instance for testing.""" with patch("mcp.server.fastmcp.FastMCP") as mock_fastmcp: # Create a mock FastMCP instance mock_mcp = MagicMock() mock_fastmcp.return_value = mock_mcp # Create the server with the mock MCP server = ClaudeCodeServer(name="test-server", mcp_instance=mock_mcp) # Return both the server and the mock MCP yield server, mock_mcp def test_initialization(self, server: tuple[ClaudeCodeServer, MagicMock]) -> None: """Test initializing ClaudeCodeServer.""" server_instance, mock_mcp = server # Verify components were initialized assert server_instance.mcp is mock_mcp assert server_instance.permission_manager is not None def test_initialization_with_allowed_paths(self) -> None: """Test initializing with allowed paths.""" allowed_paths = ["/test/path1", "/test/path2"] with ( patch("mcp.server.fastmcp.FastMCP") as mock_fastmcp, patch("mcp_claude_code.tools.register_all_tools") as mock_register, ): # Create mock fastmcp mock_mcp = MagicMock() mock_fastmcp.return_value = mock_mcp # Direct mock of the permission manager perm_manager = MagicMock() # Create the server server = ClaudeCodeServer(name="test-server", mcp_instance=mock_mcp) # Inject our mock server.permission_manager = perm_manager # Manually call register_all_tools from mcp_claude_code.tools import register_all_tools register_all_tools( mcp_server=mock_mcp, permission_manager=perm_manager, agent_model=None, agent_max_tokens=None, agent_api_key=None, agent_base_url=None, agent_max_iterations=10, agent_max_tool_uses=30, enable_agent_tool=False, ) # Now call the code that would add the paths for path in allowed_paths: server.permission_manager.add_allowed_path(path) # Verify paths were added assert perm_manager.add_allowed_path.call_count == len(allowed_paths) # Verify each path was passed for path in allowed_paths: perm_manager.add_allowed_path.assert_any_call(path) # Verify tools were registered mock_register.assert_called_once() @pytest.mark.skip(reason="Cannot run stdio server in a test environment") def test_run(self, server: tuple[ClaudeCodeServer, MagicMock]) -> None: """Test running the server.""" server_instance, mock_mcp = server # Run the server server_instance.run() # Verify the MCP server was run mock_mcp.run.assert_called_once_with(transport="stdio") # def test_run_with_transport(self, server): # """Test running the server with a specific transport.""" # server_instance, mock_mcp = server # # # Run the server with SSE transport # server_instance.run(transport="sse") # # # Verify the MCP server was run with the specified transport # mock_mcp.run.assert_called_once_with(transport="sse") @pytest.mark.skip(reason="Cannot run stdio server in a test environment") def test_run_with_allowed_paths( self, server: tuple[ClaudeCodeServer, MagicMock] ) -> None: """Test running the server with additional allowed paths.""" server_instance, mock_mcp = server # Replace permission_manager with mock server_instance.permission_manager = MagicMock() # Run the server with allowed paths additional_paths = ["/additional/path1", "/additional/path2"] server_instance.run(allowed_paths=additional_paths) # Verify paths were added by checking call counts assert server_instance.permission_manager.add_allowed_path.call_count == len( additional_paths ) # Verify each path was passed to the add methods for path in additional_paths: server_instance.permission_manager.add_allowed_path.assert_any_call(path) # Verify the MCP server was run mock_mcp.run.assert_called_once()