first commit

2026-07-22 13:48:46 +08:00
commit c87751c3dc
2820 changed files with 726976 additions and 0 deletions
@@ -0,0 +1,48 @@
+from __future__ import annotations
+
+import tempfile
+import unittest
+from pathlib import Path
+from unittest import mock
+
+from tests.python.support.paths import add_repo_path
+
+add_repo_path("skills/cad/scripts")
+
+from cadpy import assembly_spec, catalog, generation, glb_topology, metadata, render, step_scene, step_targets
+from tests.python.support.tmp_root import CAD_TEST_TMP_ROOT, temporary_directory
+
+
+IGNORED_TEST_ROOT = CAD_TEST_TMP_ROOT
+
+
+class IsolatedCadRoots:
+    def __init__(self, testcase: unittest.TestCase, *, prefix: str) -> None:
+        self._tempdir = temporary_directory(prefix=prefix)
+        testcase.addCleanup(self._tempdir.cleanup)
+
+        self.root = Path(self._tempdir.name)
+        self.cad_root = self.root / "workspace"
+        self.cad_root.mkdir(parents=True, exist_ok=True)
+
+        patches = []
+        for module in (
+            assembly_spec,
+            catalog,
+            render,
+            generation,
+            glb_topology,
+            metadata,
+            step_scene,
+            step_targets,
+        ):
+            if hasattr(module, "CAD_ROOT"):
+                patches.append(mock.patch.object(module, "CAD_ROOT", self.cad_root))
+            if hasattr(module, "REPO_ROOT"):
+                patches.append(mock.patch.object(module, "REPO_ROOT", self.cad_root))
+        for patcher in patches:
+            patcher.start()
+            testcase.addCleanup(patcher.stop)
+
+    def temporary_cad_directory(self, *, prefix: str) -> tempfile.TemporaryDirectory[str]:
+        return tempfile.TemporaryDirectory(prefix=prefix, dir=self.cad_root)