Initial public release

tests/anyimage/common/test_input_preparation.py

@@ -0,0 +1,30 @@
+import tempfile
+from pathlib import Path
+from types import SimpleNamespace
+from unittest.mock import Mock, patch
+
+from tests.support.blender import BlenderTestCase
+
+
+class InputPreparationTest(BlenderTestCase):
+
+    def test_packed_export_failure_removes_temporary_directory(self):
+        for failure in (OSError("disk full"), None):
+            with tempfile.TemporaryDirectory() as directory:
+                output = Path(directory) / "export"
+                output.mkdir()
+                image = SimpleNamespace(file_format="OPEN_EXR", save=Mock(side_effect=failure))
+                with patch.object(self.image_data.tempfile, "mkdtemp", return_value=str(output)):
+                    with self.assertRaises(RuntimeError):
+                        self.image_data.save_packed_image(image, None)
+                self.assertEqual(image.file_format, "OPEN_EXR")
+                self.assertFalse(output.exists())
+
+
+    def test_animated_inputs_are_rejected_before_export(self):
+        for source in ("MOVIE", "SEQUENCE"):
+            with self.subTest(source=source), patch.object(self.image_data, "save_packed_image") as export:
+                image = SimpleNamespace(source=source, packed_file=object())
+                with self.assertRaisesRegex(RuntimeError, "Only still images"):
+                    self.image_data.prepare_image_input(image, None)
+                export.assert_not_called()