1 files changed, 113 insertions, 0 deletions
diff --git a/servers/hello_world/tests/test_cli.py b/servers/hello_world/tests/test_cli.py
new file mode 100644
index 0000000..48fa89c
--- /dev/null
+++ b/servers/hello_world/tests/test_cli.py
@@ -0,0 +1,113 @@
+import pytest
+import sys
+from unittest.mock import patch, MagicMock
+import argparse
+from mcp_server_hello_world.cli import run_server, parse_args, validate_args
+
+def test_cli_argument_parsing():
+    """Test the argument parsing in the CLI module."""
+    # Test default arguments
+    with patch('argparse.ArgumentParser.parse_args') as mock_parse_args:
+        
+        # Set up the mock to return default values
+        mock_args = argparse.Namespace(
+            transport="stdio",
+            host="0.0.0.0",
+            port=8080,
+            log_level="INFO"
+        )
+        mock_parse_args.return_value = mock_args
+        
+        # Call the parse_args function
+        args = parse_args()
+        
+        # Check the arguments
+        assert args.transport == "stdio"
+        assert args.host == "0.0.0.0"
+        assert args.port == 8080
+        assert args.log_level == "INFO"
+    
+    # Test custom arguments
+    with patch('argparse.ArgumentParser.parse_args') as mock_parse_args:
+        
+        # Set up the mock to return custom values
+        mock_args = argparse.Namespace(
+            transport="remote",
+            host="127.0.0.1",
+            port=9090,
+            log_level="DEBUG"
+        )
+        mock_parse_args.return_value = mock_args
+        
+        # Call the parse_args function
+        args = parse_args()
+        
+        # Check the arguments
+        assert args.transport == "remote"
+        assert args.host == "127.0.0.1"
+        assert args.port == 9090
+        assert args.log_level == "DEBUG"
+
+def test_run_server():
+    """Test the run_server function."""
+    with patch('mcp_server_hello_world.cli.parse_args') as mock_parse_args, \
+         patch('mcp_server_hello_world.cli.validate_args', return_value=mock_parse_args.return_value) as mock_validate_args, \
+         patch('mcp_server_hello_world.cli.setup_logging') as mock_setup_logging, \
+         patch('asyncio.run') as mock_run:
+        
+        # Set up the mock to return default values
+        mock_args = argparse.Namespace(
+            transport="stdio",
+            host="0.0.0.0",
+            port=8080,
+            log_level="INFO"
+        )
+        mock_parse_args.return_value = mock_args
+        
+        # Call the run_server function
+        run_server()
+        
+        # Check that the functions were called with the correct arguments
+        mock_parse_args.assert_called_once()
+        mock_validate_args.assert_called_once_with(mock_args)
+        # The setup_logging function is called with the log_level attribute from the args
+        mock_setup_logging.assert_called_once()
+        mock_run.assert_called_once()
+
+def test_argument_validation():
+    """Test that the argument parser validates arguments correctly."""
+    # Test valid transport values
+    with patch('sys.argv', ['mcp-server-hello-world', '--transport', 'stdio']):
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--transport', choices=["stdio", "remote"])
+        args = parser.parse_args()
+        assert args.transport == "stdio"
+    
+    with patch('sys.argv', ['mcp-server-hello-world', '--transport', 'remote']):
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--transport', choices=["stdio", "remote"])
+        args = parser.parse_args()
+        assert args.transport == "remote"
+    
+    # Test invalid transport value
+    with patch('sys.argv', ['mcp-server-hello-world', '--transport', 'invalid']), \
+         patch('sys.stderr', MagicMock()), \
+         pytest.raises(SystemExit):
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--transport', choices=["stdio", "remote"])
+        parser.parse_args()
+    
+    # Test port validation
+    with patch('sys.argv', ['mcp-server-hello-world', '--port', '8080']):
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--port', type=int)
+        args = parser.parse_args()
+        assert args.port == 8080
+    
+    # Test invalid port (non-integer)
+    with patch('sys.argv', ['mcp-server-hello-world', '--port', 'invalid']), \
+         patch('sys.stderr', MagicMock()), \
+         pytest.raises(SystemExit):
+        parser = argparse.ArgumentParser()
+        parser.add_argument('--port', type=int)
+        parser.parse_args()