Add gated changeset application and graph controls

2026-07-25 16:00:19 -04:00 · 2026-07-25 16:00:19 -04:00 · 78335c8973
commit 78335c8973
parent 3c15e26283
20 changed files with 1813 additions and 453 deletions
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@ -0,0 +1,107 @@
+from __future__ import annotations
+
+import os
+import shutil
+import tempfile
+import threading
+import time
+import unittest
+from pathlib import Path
+from unittest import mock
+
+from docforge.changesets import ChangesetStore
+from docforge.cli import _parser, _run
+from docforge.project import Project
+from docforge.viewer_manager import ViewerManager
+
+ROOT = Path(__file__).resolve().parents[1]
+FIXTURES = ROOT / "tests" / "fixtures"
+
+
+class DocForgeCliTests(unittest.TestCase):
+    def copy_fixture(self, destination: Path) -> Path:
+        root = destination / "alpha"
+        shutil.copytree(FIXTURES / "alpha", root)
+        return root
+
+    def test_reindex_apply_and_visualization_commands_are_self_service(self) -> None:
+        with tempfile.TemporaryDirectory() as directory:
+            parent = Path(directory)
+            root = self.copy_fixture(parent)
+            parser = _parser()
+            reindexed = _run(parser.parse_args(["--project-root", str(root), "reindex"]))
+            self.assertTrue(reindexed["reindexed"])
+
+            project = Project.open(root)
+            store = ChangesetStore(project, "alpha-editor")
+            node = next(node for node in project.load().nodes if node.node_id == "guide.workflow")
+            created = store.create("cli-apply")
+            proposed = store.propose_update(
+                changeset_id="cli-apply",
+                expected_changeset_hash=str(created["changeset_hash"]),
+                node_id=node.node_id,
+                expected_content_hash=node.content_hash,
+                metadata={"summary": "Applied through the CLI."},
+                content=None,
+                relationship_changes=[],
+                rationale="Verify the direct application command.",
+            )
+            applied = _run(
+                parser.parse_args(
+                    [
+                        "--project-root",
+                        str(root),
+                        "apply",
+                        "cli-apply",
+                        "--changeset-hash",
+                        str(proposed["changeset_hash"]),
+                        "--applier",
+                        "alpha-editor",
+                    ]
+                )
+            )
+            self.assertTrue(applied["applied"])
+
+            state_path = parent / "viewer-manager.json"
+            manager = ViewerManager(state_path, check_interval_seconds=0.02)
+            thread = threading.Thread(target=manager.serve_forever, daemon=True)
+            previous = os.environ.get("DOCFORGE_VIEWER_MANAGER_STATE")
+            os.environ["DOCFORGE_VIEWER_MANAGER_STATE"] = str(state_path)
+            thread.start()
+            deadline = time.monotonic() + 2
+            while not state_path.exists() and time.monotonic() < deadline:
+                time.sleep(0.01)
+            try:
+                with mock.patch("docforge.cli.webbrowser.open", return_value=True) as opened:
+                    visualized = _run(
+                        parser.parse_args(
+                            [
+                                "--project-root",
+                                str(root),
+                                "visualize",
+                                "--node",
+                                "guide.workflow",
+                            ]
+                        )
+                    )
+                self.assertTrue(visualized["opened_browser"])
+                opened.assert_called_once()
+                status = _run(
+                    parser.parse_args(["--project-root", str(root), "visualization-status"])
+                )
+                self.assertEqual("running", status["state"])
+                stopped = _run(
+                    parser.parse_args(["--project-root", str(root), "visualization-stop"])
+                )
+                self.assertEqual("stopped", stopped["state"])
+            finally:
+                manager.shutdown()
+                thread.join(timeout=2)
+                if previous is None:
+                    os.environ.pop("DOCFORGE_VIEWER_MANAGER_STATE", None)
+                else:
+                    os.environ["DOCFORGE_VIEWER_MANAGER_STATE"] = previous
+
+
+if __name__ == "__main__":
+    unittest.main()