31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
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()
|