basic advanced example

tests/test_automatic_catalog.py

@@ -0,0 +1,77 @@
+"""Automatic catalogs preserve other models, partial selections and offline imports."""
+from pathlib import Path
+import sys
+import tempfile
+from types import ModuleType
+import unittest
+from unittest.mock import patch
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "Content/Python"))
+from ue_workshop import animation_catalog
+from ue_workshop.animation_clip import AnimationClip
+
+
+class AutomaticCatalogTests(unittest.TestCase):
+    def setUp(self):
+        self.folder = tempfile.TemporaryDirectory()
+        self.addCleanup(self.folder.cleanup)
+        self.output = Path(self.folder.name) / "animations.py"
+
+    def load(self):
+        namespace = {"__name__": "ue_workshop._catalog_test", "__package__": "ue_workshop"}
+        exec(compile(self.output.read_text(encoding="utf-8"), str(self.output), "exec"), namespace)
+        return namespace
+
+    def update(self, name, clips, **options):
+        return animation_catalog.update_catalog(name, clips, output=self.output, **options)
+
+    def test_new_model_produces_enum_with_durations_without_unreal(self):
+        self.assertEqual(self.update("StudentHero", {"walking": 2.0}), "StudentHeroAnimations")
+        clip = self.load()["StudentHeroAnimations"].WALKING
+        self.assertEqual(clip.key, "walking")
+        self.assertEqual(clip.seconds, 2)
+        self.assertEqual(clip.frames(24), 48)
+        self.assertNotIn("unreal", sys.modules)
+
+    def test_full_update_preserves_other_models_and_removes_deleted_clips(self):
+        self.update("Fighter", {"walk": 1.0, "attack": 2.0})
+        self.update("StudentHero", {"jump": 3.0})
+        self.update("Fighter", {"walk": 1.5})
+        namespace = self.load()
+        self.assertEqual([clip.key for clip in namespace["FighterAnimations"]], ["walk"])
+        self.assertEqual(namespace["StudentHeroAnimations"].JUMP.seconds, 3.0)
+
+    def test_partial_selection_updates_duration_without_erasing_other_clips(self):
+        self.update("Hero", {"walk": 1.0, "attack": 2.0})
+        self.update("Hero", {"walk": 1.5}, replace=False)
+        catalog = self.load()["HeroAnimations"]
+        self.assertEqual(catalog.WALK.seconds, 1.5)
+        self.assertEqual(catalog.ATTACK.seconds, 2.0)
+
+    def test_unchanged_metadata_does_not_rewrite_file(self):
+        self.update("Hero", {"walk": 1.0})
+        with patch.object(animation_catalog.tempfile, "NamedTemporaryFile", side_effect=AssertionError("rewritten")):
+            self.update("Hero", {"walk": 1.0})
+
+    def test_empty_model_is_valid(self):
+        self.update("Hero", {})
+        self.assertEqual(list(self.load()["HeroAnimations"]), [])
+
+    def test_invalid_metadata_does_not_damage_existing_catalog(self):
+        self.update("Hero", {"walk": 1.0})
+        before = self.output.read_bytes()
+        for data in ({"walk": 0}, {"walk": float("nan")}, {"walk": 1.0, "WALK": 2.0}):
+            with self.subTest(data=data), self.assertRaises(ValueError):
+                self.update("Hero", data)
+            self.assertEqual(self.output.read_bytes(), before)
+
+    def test_publish_adds_class_to_loaded_module_without_replacing_other_enums(self):
+        module = ModuleType("ue_workshop.animations")
+        module.AnimationClip = AnimationClip
+        module.ExistingAnimations = object()
+        original = module.ExistingAnimations
+        source = animation_catalog._class_source("StudentHeroAnimations", {"walk": 2.0})
+        with patch.dict(sys.modules, {"ue_workshop.animations": module}):
+            animation_catalog._publish_class(source, self.output)
+            self.assertEqual(module.StudentHeroAnimations.WALK.frames(24), 48)
+            self.assertIs(module.ExistingAnimations, original)