Add test suite

tests/test_cli.py

@@ -0,0 +1,74 @@
+import pytest
+from click.testing import CliRunner
+from unittest.mock import Mock, patch, AsyncMock
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
+
+from promptforge.cli.main import main
+
+
+@pytest.fixture
+def cli_runner():
+    return CliRunner()
+
+
+def test_main_help(cli_runner):
+    result = cli_runner.invoke(main, ["--help"])
+    assert result.exit_code == 0
+    assert "PromptForge" in result.output
+
+
+def test_main_version(cli_runner):
+    result = cli_runner.invoke(main, ["--version"])
+    assert result.exit_code == 0
+    assert "0.1.0" in result.output
+
+
+def test_prompt_group(cli_runner):
+    result = cli_runner.invoke(main, ["prompt", "--help"])
+    assert result.exit_code == 0
+    assert "create" in result.output
+    assert "list" in result.output
+    assert "show" in result.output
+
+
+def test_version_group(cli_runner):
+    result = cli_runner.invoke(main, ["version", "--help"])
+    assert result.exit_code == 0
+    assert "history" in result.output
+    assert "create" in result.output
+
+
+def test_registry_group(cli_runner):
+    result = cli_runner.invoke(main, ["registry", "--help"])
+    assert result.exit_code == 0
+    assert "list" in result.output
+    assert "search" in result.output
+
+
+def test_run_command(cli_runner, tmp_path):
+    with patch("promptforge.cli.commands.run.Prompt.list") as mock_list:
+        mock_prompt = Mock()
+        mock_prompt.name = "TestPrompt"
+        mock_prompt.content = "Hello {{name}}"
+        mock_prompt.variables = []
+        mock_prompt.provider = None
+        mock_prompt.validation_rules = []
+        mock_list.return_value = [mock_prompt]
+
+        with patch("promptforge.cli.commands.run.ProviderFactory.create") as mock_factory:
+            mock_provider = Mock()
+            mock_provider.stream_complete = AsyncMock(return_value=iter(["Hello ", "World"]))
+            mock_factory.return_value = mock_provider
+
+            with patch("promptforge.cli.commands.run.get_config") as mock_config:
+                mock_cfg = Mock()
+                mock_cfg.providers = {}
+                mock_cfg.defaults.provider = "openai"
+                mock_config.return_value = mock_cfg
+
+                result = cli_runner.invoke(main, ["run", "TestPrompt", "-v", "name=World", "--no-stream"])
+                assert result.exit_code == 0
+                assert mock_factory.called