feat(download_imageset): process ZIP archives

download_imageset_test.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env -S uv run --script
+# /// script
+# requires-python = ">=3.11"
+# dependencies = ["pytest", "httpx", "pillow", "click"]
+# ///
+import sys
+import zipfile
+from pathlib import Path
+from unittest.mock import Mock
+
+import pytest
+from click.testing import CliRunner
+from PIL import Image
+
+import download_imageset
+
+
+@pytest.mark.parametrize("extension", [".zip", ".ZIP"])
+def test_process_zip_and_directory(
+    tmp_path: Path, monkeypatch: pytest.MonkeyPatch, extension: str
+) -> None:
+    directory = tmp_path / "existing"
+    nested = directory / "nested"
+    nested.mkdir(parents=True)
+    image_path = nested / "photo.png"
+    Image.new(mode="RGB", size=(10, 10)).save(fp=image_path)
+    archive_path = tmp_path / f"gallery{extension}"
+    with zipfile.ZipFile(file=archive_path, mode="w") as archive:
+        archive.write(filename=image_path, arcname="album/photo.png")
+    trash = Mock()
+    monkeypatch.setattr(download_imageset, "move_to_trash", trash)
+
+    result = CliRunner().invoke(
+        cli=download_imageset.cli,
+        args=["process", str(archive_path), str(directory)],
+    )
+
+    assert result.exit_code == 0, result.output
+    assert (tmp_path / "gallery" / "gallery__0000.png").is_file()
+    assert (directory / "existing__0000.png").is_file()
+    assert not (tmp_path / "gallery" / "album").exists()
+    assert not nested.exists()
+    trash.assert_called_once_with(file_paths=[archive_path])
+
+
+@pytest.mark.parametrize("filename", ["missing", "unsupported.txt", "broken.zip"])
+def test_process_skips_invalid_inputs(
+    tmp_path: Path, monkeypatch: pytest.MonkeyPatch, filename: str
+) -> None:
+    source = tmp_path / filename
+    if filename != "missing":
+        source.write_text(data="not an archive")
+    trash = Mock()
+    monkeypatch.setattr(download_imageset, "move_to_trash", trash)
+
+    result = CliRunner().invoke(
+        cli=download_imageset.cli, args=["process", str(source)]
+    )
+
+    assert result.exit_code == 0, result.output
+    trash.assert_not_called()
+    if filename != "missing":
+        assert source.read_text() == "not an archive"
+    assert not (tmp_path / "broken").exists()
+
+
+def test_process_keeps_zip_when_processing_fails(
+    tmp_path: Path, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    archive_path = tmp_path / "gallery.zip"
+    with zipfile.ZipFile(file=archive_path, mode="w") as archive:
+        archive.writestr(zinfo_or_arcname="photo.png", data=b"invalid image")
+    trash = Mock()
+    monkeypatch.setattr(download_imageset, "move_to_trash", trash)
+
+    result = CliRunner().invoke(
+        cli=download_imageset.cli, args=["process", str(archive_path)]
+    )
+
+    assert result.exit_code != 0
+    assert archive_path.is_file()
+    trash.assert_not_called()
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(args=[str(Path(__file__).resolve()), *sys.argv[1:]]))