first commit

2026-07-22 13:48:46 +08:00
commit c87751c3dc
2820 changed files with 726976 additions and 0 deletions
@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+import json
+import tempfile
+import unittest
+from pathlib import Path
+
+from tests.python.support.paths import add_repo_path
+
+add_repo_path("packages/cadpy/src")
+
+from cadpy.generation_status import GenerationOutput, track_generation_run
+
+
+class GenerationStatusTests(unittest.TestCase):
+    def test_generation_status_uses_hidden_lock_next_to_output_and_cleans_up(self) -> None:
+        with tempfile.TemporaryDirectory(prefix="cadpy-generation-status-") as tempdir:
+            workspace = Path(tempdir) / "workspace"
+            model_dir = workspace / "models"
+            skill_dir = Path(tempdir) / "installed-skill"
+            source_path = model_dir / "part.py"
+            step_path = model_dir / "part.step"
+            model_dir.mkdir(parents=True)
+            source_path.write_text("def gen_step():\n    return None\n", encoding="utf-8")
+            (skill_dir / "scripts").mkdir(parents=True)
+            (skill_dir / "SKILL.md").write_text("# skill\n", encoding="utf-8")
+
+            with track_generation_run(
+                source_path=source_path,
+                generator="gen_step",
+                outputs=[GenerationOutput(step_path, "step")],
+                repo_root=skill_dir,
+            ):
+                lock_paths = tuple(model_dir.glob(".part.step.*.generation.lock.json"))
+                self.assertEqual(1, len(lock_paths))
+                payload = json.loads(lock_paths[0].read_text(encoding="utf-8"))
+                self.assertEqual("running", payload["status"])
+                self.assertEqual("gen_step", payload["generator"])
+                self.assertEqual("part.py", payload["sourcePath"])
+                self.assertEqual("part.step", payload["outputs"][0]["path"])
+                self.assertFalse(tuple(skill_dir.rglob("*.generation.lock.json")))
+
+            self.assertFalse(lock_paths[0].exists())
+
+    def test_generation_status_writes_one_lock_for_each_output_without_blocking(self) -> None:
+        with tempfile.TemporaryDirectory(prefix="cadpy-generation-status-") as tempdir:
+            model_dir = Path(tempdir) / "models"
+            step_path = model_dir / "part.step"
+            glb_path = model_dir / ".part.step.glb"
+            model_dir.mkdir(parents=True)
+
+            with track_generation_run(
+                source_path=None,
+                generator="gen_step",
+                outputs=[
+                    GenerationOutput(step_path, "step"),
+                    GenerationOutput(glb_path, "glb"),
+                ],
+                repo_root=Path(tempdir),
+            ):
+                step_locks = tuple(model_dir.glob(".part.step.*.generation.lock.json"))
+                glb_locks = tuple(model_dir.glob("..part.step.glb.*.generation.lock.json"))
+                self.assertEqual(1, len(step_locks))
+                self.assertEqual(1, len(glb_locks))
+                self.assertNotEqual(step_locks[0], glb_locks[0])
+
+            self.assertFalse(tuple(model_dir.glob("*.generation.lock.json")))
+
+
+if __name__ == "__main__":
+    unittest.main()