Merge pull request #10 from PromptExecution/copilot/sub-pr-7-another-one

Add unit tests for --mcp flag functionality

Author: elasticdotventures

tests/test_entrypoint.py

@@ -2,6 +2,7 @@
 
 from pathlib import Path
 from typing import Sequence
+from unittest.mock import MagicMock
 
 import pytest
 
@@ -37,3 +38,65 @@ def fake_run(_cmd: Sequence[str], stdin_path: Path | None = None) -> str:
     rc = entrypoint.main(["--agent", "amp", "1"])
     assert rc == 0
     assert (tmp_path / "progress.txt").exists()
+
+
+def test_parse_args_mcp_without_agent() -> None:
+    """Test that --mcp flag works without requiring --agent."""
+    args = entrypoint._parse_args(["--mcp"])
+    assert args.mcp is True
+    assert args.agent is None
+
+
+def test_parse_args_mcp_with_default_transport() -> None:
+    """Test that --mcp uses stdio as default transport."""
+    args = entrypoint._parse_args(["--mcp"])
+    assert args.mcp is True
+    assert args.transport == "stdio"
+
+
+def test_parse_args_mcp_with_http_transport() -> None:
+    """Test that --mcp accepts HTTP transport with host and port."""
+    args = entrypoint._parse_args([
+        "--mcp",
+        "--transport", "http",
+        "--host", "0.0.0.0",
+        "--port", "9000",
+    ])
+    assert args.mcp is True
+    assert args.transport == "http"
+    assert args.host == "0.0.0.0"
+    assert args.port == 9000
+
+
+def test_parse_args_mcp_with_default_host_port() -> None:
+    """Test that --mcp uses default host and port values."""
+    args = entrypoint._parse_args(["--mcp", "--transport", "http"])
+    assert args.host == "127.0.0.1"
+    assert args.port == 8000
+
+
+def test_main_mcp_calls_run_with_stdio(monkeypatch: pytest.MonkeyPatch) -> None:
+    """Test that main() calls mcp.run() with correct parameters for stdio."""
+    mock_mcp_run = MagicMock()
+    monkeypatch.setattr("ralph.entrypoint.mcp.run", mock_mcp_run)
+
+    rc = entrypoint.main(["--mcp"])
+
+    assert rc == 0
+    mock_mcp_run.assert_called_once_with(transport="stdio")
+
+
+def test_main_mcp_calls_run_with_http(monkeypatch: pytest.MonkeyPatch) -> None:
+    """Test that main() calls mcp.run() with correct parameters for HTTP."""
+    mock_mcp_run = MagicMock()
+    monkeypatch.setattr("ralph.entrypoint.mcp.run", mock_mcp_run)
+
+    rc = entrypoint.main([
+        "--mcp",
+        "--transport", "http",
+        "--host", "0.0.0.0",
+        "--port", "9000",
+    ])
+
+    assert rc == 0
+    mock_mcp_run.assert_called_once_with(transport="http", host="0.0.0.0", port=9000)