yamlgraph 0.1.1__py3-none-any.whl

tests/unit/test_generic_report.py

@@ -0,0 +1,200 @@
+"""Tests for Phase 6.5: Generic Report Schema.
+
+Tests for flexible GenericReport model that works for most analysis/summary tasks.
+"""
+
+import pytest
+from pydantic import ValidationError
+
+
+class TestGenericReportSchema:
+    """Tests for GenericReport model."""
+
+    def test_generic_report_exists(self):
+        """GenericReport model is importable."""
+        from yamlgraph.models.schemas import GenericReport
+
+        assert GenericReport is not None
+
+    def test_minimal_report(self):
+        """Report works with just title and summary."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Test Report",
+            summary="A brief summary of findings.",
+        )
+
+        assert report.title == "Test Report"
+        assert report.summary == "A brief summary of findings."
+
+    def test_report_with_sections(self):
+        """Sections field accepts arbitrary dict content."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Report",
+            summary="Summary",
+            sections={
+                "overview": "First section content",
+                "details": {"nested": "data", "count": 42},
+                "items": ["a", "b", "c"],
+            },
+        )
+
+        assert report.sections["overview"] == "First section content"
+        assert report.sections["details"]["count"] == 42
+        assert len(report.sections["items"]) == 3
+
+    def test_report_with_findings(self):
+        """Findings field is list of strings."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Report",
+            summary="Summary",
+            findings=["Finding 1", "Finding 2", "Finding 3"],
+        )
+
+        assert len(report.findings) == 3
+        assert "Finding 1" in report.findings
+
+    def test_report_with_recommendations(self):
+        """Recommendations field is list of strings."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Report",
+            summary="Summary",
+            recommendations=["Action 1", "Action 2"],
+        )
+
+        assert len(report.recommendations) == 2
+
+    def test_report_with_metadata(self):
+        """Metadata field accepts arbitrary key-value data."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Report",
+            summary="Summary",
+            metadata={
+                "author": "Test Author",
+                "version": 1.0,
+                "tags": ["a", "b"],
+            },
+        )
+
+        assert report.metadata["author"] == "Test Author"
+        assert report.metadata["version"] == 1.0
+
+    def test_defaults_are_empty(self):
+        """Optional fields default to empty collections."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(title="Report", summary="Summary")
+
+        assert report.sections == {}
+        assert report.findings == []
+        assert report.recommendations == []
+        assert report.metadata == {}
+
+    def test_title_is_required(self):
+        """Title field is required."""
+        from yamlgraph.models.schemas import GenericReport
+
+        with pytest.raises(ValidationError) as exc_info:
+            GenericReport(summary="Summary")
+
+        errors = exc_info.value.errors()
+        assert any(e["loc"] == ("title",) for e in errors)
+
+    def test_summary_is_required(self):
+        """Summary field is required."""
+        from yamlgraph.models.schemas import GenericReport
+
+        with pytest.raises(ValidationError) as exc_info:
+            GenericReport(title="Title")
+
+        errors = exc_info.value.errors()
+        assert any(e["loc"] == ("summary",) for e in errors)
+
+    def test_model_serializes_to_dict(self):
+        """Report serializes to dictionary."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Test",
+            summary="Summary",
+            findings=["A", "B"],
+        )
+
+        data = report.model_dump()
+
+        assert data["title"] == "Test"
+        assert data["summary"] == "Summary"
+        assert data["findings"] == ["A", "B"]
+
+    def test_model_serializes_to_json(self):
+        """Report serializes to JSON string."""
+        import json
+
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(title="Test", summary="Summary")
+
+        json_str = report.model_dump_json()
+        data = json.loads(json_str)
+
+        assert data["title"] == "Test"
+
+
+class TestGenericReportUseCases:
+    """Verify GenericReport works for common analysis patterns."""
+
+    def test_git_analysis_report(self):
+        """GenericReport works for git analysis output."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="Git Repository Analysis",
+            summary="Analysis of recent activity in the repository.",
+            sections={
+                "commit_summary": "15 commits in last 7 days",
+                "authors": ["alice", "bob", "carol"],
+            },
+            findings=[
+                "High activity in src/core module",
+                "No tests for new features",
+                "Breaking changes in v2.0",
+            ],
+            recommendations=[
+                "Add tests for recent changes",
+                "Review breaking changes before release",
+            ],
+            metadata={
+                "repo": "langgraph-showcase",
+                "analyzed_at": "2024-01-01T00:00:00",
+            },
+        )
+
+        assert "Git Repository" in report.title
+        assert len(report.findings) == 3
+        assert report.metadata["repo"] == "langgraph-showcase"
+
+    def test_api_analysis_report(self):
+        """GenericReport works for API analysis output."""
+        from yamlgraph.models.schemas import GenericReport
+
+        report = GenericReport(
+            title="API Performance Report",
+            summary="Performance analysis of API endpoints.",
+            sections={
+                "latency": {"p50": 45, "p95": 120, "p99": 250},
+                "errors": {"rate": 0.02, "top_errors": ["500", "429"]},
+            },
+            findings=["High latency on /search endpoint"],
+            recommendations=["Add caching for /search"],
+        )
+
+        assert report.sections["latency"]["p95"] == 120

tests/unit/test_graph_commands.py

@@ -0,0 +1,327 @@
+"""Tests for universal graph runner (Phase 7.2).
+
+TDD tests for `yamlgraph graph run <path>` command.
+"""
+
+import argparse
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+# =============================================================================
+# graph subcommand tests
+# =============================================================================
+
+
+class TestGraphSubcommand:
+    """Tests for graph subcommand group."""
+
+    def test_graph_subparser_exists(self):
+        """graph subparser should be configured."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        # Parse with graph command
+        args = parser.parse_args(["graph", "list"])
+        assert args.command == "graph"
+
+    def test_graph_run_subcommand_exists(self):
+        """graph run subcommand should exist."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(
+            ["graph", "run", "graphs/yamlgraph.yaml", "--var", "topic=AI"]
+        )
+        assert args.graph_command == "run"
+        assert args.graph_path == "graphs/yamlgraph.yaml"
+
+    def test_graph_list_subcommand_exists(self):
+        """graph list subcommand should exist."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(["graph", "list"])
+        assert args.graph_command == "list"
+
+    def test_graph_info_subcommand_exists(self):
+        """graph info subcommand should exist."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(["graph", "info", "graphs/yamlgraph.yaml"])
+        assert args.graph_command == "info"
+        assert args.graph_path == "graphs/yamlgraph.yaml"
+
+
+# =============================================================================
+# graph run argument parsing tests
+# =============================================================================
+
+
+class TestGraphRunArgs:
+    """Tests for graph run argument parsing."""
+
+    def test_var_single_value(self):
+        """--var key=value should parse correctly."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(
+            ["graph", "run", "graphs/test.yaml", "--var", "topic=AI"]
+        )
+        assert args.var == ["topic=AI"]
+
+    def test_var_multiple_values(self):
+        """Multiple --var flags should accumulate."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(
+            [
+                "graph",
+                "run",
+                "graphs/test.yaml",
+                "--var",
+                "topic=AI",
+                "--var",
+                "style=casual",
+            ]
+        )
+        assert args.var == ["topic=AI", "style=casual"]
+
+    def test_thread_argument(self):
+        """--thread should set thread ID."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(
+            ["graph", "run", "graphs/test.yaml", "--thread", "abc123"]
+        )
+        assert args.thread == "abc123"
+
+    def test_export_flag(self):
+        """--export flag should enable export."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(["graph", "run", "graphs/test.yaml", "--export"])
+        assert args.export is True
+
+    def test_graph_path_required(self):
+        """graph run requires a path argument."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        with pytest.raises(SystemExit):
+            parser.parse_args(["graph", "run"])
+
+
+# =============================================================================
+# parse_vars helper tests
+# =============================================================================
+
+
+class TestParseVars:
+    """Tests for --var parsing helper."""
+
+    def test_parse_single_var(self):
+        """Single var should parse to dict."""
+        from yamlgraph.cli.graph_commands import parse_vars
+
+        result = parse_vars(["topic=AI"])
+        assert result == {"topic": "AI"}
+
+    def test_parse_multiple_vars(self):
+        """Multiple vars should parse to dict."""
+        from yamlgraph.cli.graph_commands import parse_vars
+
+        result = parse_vars(["topic=AI", "style=casual", "count=5"])
+        assert result == {"topic": "AI", "style": "casual", "count": "5"}
+
+    def test_parse_empty_list(self):
+        """Empty list returns empty dict."""
+        from yamlgraph.cli.graph_commands import parse_vars
+
+        result = parse_vars([])
+        assert result == {}
+
+    def test_parse_none_returns_empty(self):
+        """None returns empty dict."""
+        from yamlgraph.cli.graph_commands import parse_vars
+
+        result = parse_vars(None)
+        assert result == {}
+
+    def test_parse_value_with_equals(self):
+        """Value containing = should preserve it."""
+        from yamlgraph.cli.graph_commands import parse_vars
+
+        result = parse_vars(["equation=a=b+c"])
+        assert result == {"equation": "a=b+c"}
+
+    def test_parse_invalid_format_raises(self):
+        """Invalid format (no =) should raise ValueError."""
+        from yamlgraph.cli.graph_commands import parse_vars
+
+        with pytest.raises(ValueError, match="Invalid"):
+            parse_vars(["invalid"])
+
+
+# =============================================================================
+# cmd_graph_run tests
+# =============================================================================
+
+
+class TestCmdGraphRun:
+    """Tests for cmd_graph_run function."""
+
+    def test_cmd_graph_run_exists(self):
+        """cmd_graph_run function should exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_run
+
+        assert callable(cmd_graph_run)
+
+    def test_graph_not_found_error(self):
+        """Should error if graph file doesn't exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_run
+
+        args = argparse.Namespace(
+            graph_path="nonexistent.yaml",
+            var=[],
+            thread=None,
+            export=False,
+        )
+
+        with pytest.raises(SystemExit):
+            cmd_graph_run(args)
+
+    @patch("yamlgraph.graph_loader.load_and_compile")
+    def test_invokes_graph_with_vars(self, mock_load):
+        """Should invoke graph with parsed vars as initial state."""
+        from yamlgraph.cli.graph_commands import cmd_graph_run
+
+        mock_graph = MagicMock()
+        mock_app = MagicMock()
+        mock_app.invoke.return_value = {"result": "success"}
+        mock_graph.compile.return_value = mock_app
+        mock_load.return_value = mock_graph
+
+        args = argparse.Namespace(
+            graph_path="graphs/yamlgraph.yaml",
+            var=["topic=AI", "style=casual"],
+            thread=None,
+            export=False,
+        )
+
+        # Mock Path.exists
+        with patch.object(Path, "exists", return_value=True):
+            cmd_graph_run(args)
+
+        mock_app.invoke.assert_called_once()
+        call_args = mock_app.invoke.call_args[0][0]
+        assert call_args["topic"] == "AI"
+        assert call_args["style"] == "casual"
+
+
+# =============================================================================
+# cmd_graph_list tests
+# =============================================================================
+
+
+class TestCmdGraphList:
+    """Tests for cmd_graph_list function."""
+
+    def test_cmd_graph_list_exists(self):
+        """cmd_graph_list function should exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_list
+
+        assert callable(cmd_graph_list)
+
+    @patch("yamlgraph.cli.graph_commands.Path")
+    def test_lists_yaml_files(self, mock_path):
+        """Should list all .yaml files in graphs/."""
+        from yamlgraph.cli.graph_commands import cmd_graph_list
+
+        mock_graphs_dir = MagicMock()
+        mock_path.return_value = mock_graphs_dir
+        mock_graphs_dir.exists.return_value = True
+        mock_graphs_dir.glob.return_value = [
+            Path("graphs/yamlgraph.yaml"),
+            Path("graphs/router-demo.yaml"),
+        ]
+
+        args = argparse.Namespace()
+
+        with patch("builtins.print") as mock_print:
+            cmd_graph_list(args)
+            # Check it printed something about the graphs
+            calls = [str(c) for c in mock_print.call_args_list]
+            assert any("yamlgraph" in c for c in calls)
+
+
+# =============================================================================
+# cmd_graph_info tests
+# =============================================================================
+
+
+class TestCmdGraphInfo:
+    """Tests for cmd_graph_info function."""
+
+    def test_cmd_graph_info_exists(self):
+        """cmd_graph_info function should exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_info
+
+        assert callable(cmd_graph_info)
+
+    def test_info_file_not_found(self):
+        """Should error if graph file doesn't exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_info
+
+        args = argparse.Namespace(graph_path="nonexistent.yaml")
+
+        with pytest.raises(SystemExit):
+            cmd_graph_info(args)
+
+
+# =============================================================================
+# cmd_graph_validate tests
+# =============================================================================
+
+
+class TestCmdGraphValidate:
+    """Tests for cmd_graph_validate function."""
+
+    def test_cmd_graph_validate_exists(self):
+        """cmd_graph_validate function should exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_validate
+
+        assert callable(cmd_graph_validate)
+
+    def test_validate_file_not_found(self):
+        """Should error if graph file doesn't exist."""
+        from yamlgraph.cli.graph_commands import cmd_graph_validate
+
+        args = argparse.Namespace(graph_path="nonexistent.yaml")
+
+        with pytest.raises(SystemExit):
+            cmd_graph_validate(args)
+
+    def test_validate_valid_graph(self):
+        """Should validate a correct graph without errors."""
+        from yamlgraph.cli.graph_commands import cmd_graph_validate
+
+        args = argparse.Namespace(graph_path="graphs/yamlgraph.yaml")
+
+        # Should not raise
+        cmd_graph_validate(args)
+
+    def test_validate_subparser_exists(self):
+        """graph validate subcommand should exist."""
+        from yamlgraph.cli import create_parser
+
+        parser = create_parser()
+        args = parser.parse_args(["graph", "validate", "graphs/yamlgraph.yaml"])
+        assert args.graph_command == "validate"
+        assert args.graph_path == "graphs/yamlgraph.yaml"