Initial public release

tests/anyimage/server/test_input_validation.py

@@ -0,0 +1,74 @@
+
+import pytest
+from PIL import Image
+
+from tests.anyimage.server.support import load_server_module
+
+
+@pytest.fixture
+def media():
+    return load_server_module("media.input")
+
+
+@pytest.mark.parametrize("suffix", [".PNG", ".JPG", ".webp", ".jpeg"])
+def test_supported_format_is_case_insensitive(media, tmp_path, suffix):
+    path = tmp_path / f"input{suffix}"
+    path.touch()
+    assert media.validate_input_path(path) == path
+
+
+@pytest.mark.parametrize("suffix,kind", [(".png", "image"), (".mp4", "image")])
+def test_missing_path_precedes_format_validation(media, tmp_path, suffix, kind):
+    with pytest.raises(RuntimeError, match=f"Source {kind} file does not exist"):
+        media.validate_input_path(tmp_path / f"missing{suffix}")
+
+
+def test_image_decode_error_keeps_cause(media, tmp_path):
+    path = tmp_path / "broken.png"
+    path.write_bytes(b"broken")
+    with pytest.raises(RuntimeError, match="Unable to read image") as error:
+        with media.open_image(path):
+            pytest.fail("Broken image must not reach processing")
+    assert isinstance(error.value.__cause__, OSError)
+
+
+def test_processing_error_is_not_relabelled(media, tmp_path):
+    path = tmp_path / "valid.png"
+    Image.new("RGB", (2, 2)).save(path)
+    failure = OSError("inference failed")
+    with pytest.raises(OSError) as error:
+        with media.open_image(path):
+            raise failure
+    assert error.value is failure
+
+
+
+
+@pytest.mark.parametrize("suffix", [".mp4", ".mts", ".gif", ".tiff", ""])
+def test_unsupported_formats_are_rejected(media, tmp_path, suffix):
+    path = tmp_path / f"input{suffix}"
+    path.touch()
+    with pytest.raises(RuntimeError, match="Unsupported input format"):
+        media.validate_input_path(path)
+
+
+def test_frame_directory_is_rejected(media, tmp_path):
+    Image.new("RGB", (2, 2)).save(tmp_path / "frame.png")
+    with pytest.raises(RuntimeError, match="Source image file does not exist"):
+        media.validate_input_path(tmp_path)
+
+
+def test_animated_png_is_rejected(media, tmp_path):
+    path = tmp_path / "animated.png"
+    Image.new("RGB", (2, 2), "red").save(
+        path, save_all=True, append_images=[Image.new("RGB", (2, 2), "blue")], duration=100,
+    )
+    with pytest.raises(RuntimeError, match="Only still images"):
+        with media.open_image(path):
+            pytest.fail("Animated images must not reach processing")
+
+
+def test_ai_tasks_support_cooperative_cancellation(media):
+    with pytest.raises(media.TaskCancelled):
+        media.check_cancelled(lambda: True)
+    media.check_cancelled(lambda: False)