Add tests for image layer, image plane, inventory, saved layout, commands, documentation, manifest, and packaging

- Implement tests for image layer functionality including alpha curve and depth layer behavior. - Create tests for image plane evaluation and modifier input handling. - Add tests to verify the saved asset inventory and its persistence. - Introduce tests for the documentation command to ensure correct project version deployment. - Validate the extension manifest structure and required permissions. - Develop tests for the packaging process, ensuring correct platform handling and dependency management. - Enhance image depth plane group functionality with new split threshold input and surface processing.
2026-09-09 22:18:59 +08:00
parent 89087fc2e2
commit f48bbb3d05
97 changed files with 436 additions and 705 deletions
--- a/tests/anyimage/test_image_menu.py
+++ b/tests/anyimage/test_image_menu.py
@@ -0,0 +1,130 @@
+from types import SimpleNamespace
+from unittest.mock import patch
+from tests.support.blender import BlenderTestCase
+
+
+class ImageMenuTest(BlenderTestCase):
+    def test_image_menu_keeps_ai_features_clickable_and_offers_models(self):
+        calls = []
+        labels = {}
+
+        class Layout:
+            operator_context = ""
+            enabled = True
+
+            def operator(self, identifier, **options):
+                calls.append((identifier, self.enabled))
+                labels[identifier] = options.get("text")
+                return SimpleNamespace()
+
+            def separator(self):
+                calls.append(("separator", self.enabled))
+
+        menu = self.anyimage.AnyImageImageMenu()
+        menu.layout = Layout()
+        with patch(
+            "anyimage.properties.ai_status",
+            return_value={
+                "environment_ready": True,
+                "missing_models": ("BEN2_BASE",),
+                "ready": False,
+            },
+        ):
+            menu.draw(
+                SimpleNamespace(
+                    object=SimpleNamespace(
+                        type="EMPTY",
+                        empty_display_type="IMAGE",
+                        data=SimpleNamespace(size=(640, 480)),
+                    )
+                )
+            )
+
+        self.assertEqual(
+            calls,
+            [
+                ("anyimage.convert_to_plane", True),
+                ("anyimage.convert_to_depth_plane", True),
+                ("anyimage.convert_to_relief_plane", True),
+                ("separator", True),
+                ("anyimage.remove_image_background", True),
+                ("anyimage.upscale_image", True),
+                ("separator", True),
+                ("anyimage.setup_ai_environment", True),
+            ],
+        )
+        self.assertEqual(
+            labels["anyimage.setup_ai_environment"],
+            "Download Required Models…",
+        )
+
+        self.assertEqual(
+            labels["anyimage.upscale_image"],
+            "Upscale (1280 × 960)",
+        )
+
+
+    def test_image_menu_opens_environment_settings_when_ai_is_ready(self):
+        calls = []
+
+        class Layout:
+            operator_context = ""
+            enabled = True
+
+            def operator(self, identifier, **_options):
+                calls.append((identifier, self.enabled))
+                return SimpleNamespace()
+
+            def separator(self):
+                calls.append(("separator", self.enabled))
+
+        menu = self.anyimage.AnyImageImageMenu()
+        menu.layout = Layout()
+        with patch(
+            "anyimage.properties.ai_status",
+            return_value={
+                "environment_ready": True,
+                "missing_models": (),
+                "ready": True,
+            },
+        ):
+            menu.draw(None)
+
+        self.assertIn(("anyimage.open_ai_environment_settings", True), calls)
+        self.assertNotIn(("anyimage.setup_ai_environment", True), calls)
+
+
+    def test_image_operators_stay_clickable_until_server_is_busy(self):
+        context = SimpleNamespace(
+            object=SimpleNamespace(data=SimpleNamespace()),
+        )
+        operators = (
+            (self.convert_to_plane, self.convert_to_plane.ConvertToDepthPlane),
+            (self.convert_to_plane, self.convert_to_plane.ConvertToReliefPlane),
+            (self.remove_background, self.remove_background.RemoveImageBackground),
+        )
+
+        for busy, expected in ((False, True), (True, False)):
+            with (
+                self.subTest(busy=busy),
+                patch.object(
+                    self.anyimage.runtime,
+                    "server_busy",
+                    return_value=busy,
+                ),
+            ):
+                for module, operator in operators:
+                    with (
+                        self.subTest(operator=operator.__name__),
+                        patch.object(
+                            module,
+                            "is_image_empty",
+                            return_value=True,
+                        ),
+                        patch.object(
+                            module,
+                            "video_total_frames",
+                            return_value=1,
+                        ),
+                    ):
+                        self.assertEqual(operator.poll(context), expected)