first commit

2026-07-22 13:48:46 +08:00
commit c87751c3dc
2820 changed files with 726976 additions and 0 deletions
@@ -0,0 +1,95 @@
+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_metadata/src")
+
+from cadpy_metadata.generator import (
+    GenerationOutput,
+    PythonSourceIdentity,
+    track_generation_run,
+    xml_with_text_to_cad_metadata,
+)
+
+
+class MetadataGenerationStatusTests(unittest.TestCase):
+    def test_generation_status_uses_hidden_lock_next_to_output_and_cleans_up(self) -> None:
+        with tempfile.TemporaryDirectory(prefix="cadpy-metadata-generation-status-") as tempdir:
+            workspace = Path(tempdir) / "workspace"
+            model_dir = workspace / "models"
+            skill_dir = Path(tempdir) / "installed-skill"
+            source_path = model_dir / "robot_urdf.py"
+            urdf_path = model_dir / "robot.urdf"
+            model_dir.mkdir(parents=True)
+            source_path.write_text("def gen_urdf():\n    return '<robot />'\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_urdf",
+                outputs=[GenerationOutput(urdf_path, "urdf")],
+                repo_root=skill_dir,
+            ):
+                lock_paths = tuple(model_dir.glob(".robot.urdf.*.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_urdf", payload["generator"])
+                self.assertEqual("robot_urdf.py", payload["sourcePath"])
+                self.assertEqual("robot.urdf", 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(self) -> None:
+        with tempfile.TemporaryDirectory(prefix="cadpy-metadata-generation-status-") as tempdir:
+            model_dir = Path(tempdir) / "models"
+            urdf_path = model_dir / "robot.urdf"
+            srdf_path = model_dir / "robot.srdf"
+            model_dir.mkdir(parents=True)
+
+            with track_generation_run(
+                source_path=None,
+                generator="gen_srdf",
+                outputs=[
+                    GenerationOutput(urdf_path, "urdf"),
+                    GenerationOutput(srdf_path, "srdf"),
+                ],
+                repo_root=Path(tempdir),
+            ):
+                urdf_locks = tuple(model_dir.glob(".robot.urdf.*.generation.lock.json"))
+                srdf_locks = tuple(model_dir.glob(".robot.srdf.*.generation.lock.json"))
+                self.assertEqual(1, len(urdf_locks))
+                self.assertEqual(1, len(srdf_locks))
+                self.assertNotEqual(urdf_locks[0], srdf_locks[0])
+
+            self.assertFalse(tuple(model_dir.glob("*.generation.lock.json")))
+
+    def test_xml_metadata_source_path_is_relative_to_output(self) -> None:
+        with tempfile.TemporaryDirectory(prefix="cadpy-metadata-xml-") as tempdir:
+            workspace = Path(tempdir)
+            source_path = workspace / "src" / "robot.py"
+            output_path = workspace / "out" / "robot.urdf"
+            identity = PythonSourceIdentity(
+                source_path="src/robot.py",
+                source_hash="source-hash",
+            )
+
+            xml = xml_with_text_to_cad_metadata(
+                "<robot name=\"sample\" />",
+                identity,
+                output_path=output_path,
+                source_path=source_path,
+            )
+
+            self.assertIn("<!-- cadpy:sourcePath=../src/robot.py -->", xml)
+
+
+if __name__ == "__main__":
+    unittest.main()