fix(immichup): Add images to studio albums

immichup_test.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env -S uv run --script
+# /// script
+# dependencies = ["pytest", "httpx>=0.27,<1"]
+# ///
+
+import sys
+from pathlib import Path
+from unittest.mock import Mock, patch
+
+import pytest
+
+import immichup
+
+
+@pytest.mark.parametrize(
+    "filename, expected_studio",
+    [
+        ("Irina Dreyt -- @Mavrin__0007-upscaled.jpg", "Mavrin"),
+        ("Irina Dreyt -- @Mavrin__0155.jpg", "Mavrin"),
+        ("Irina Dreyt -- @MAVRIN__0007-UPSCALED.jpg", "MAVRIN"),
+        ("Irina Dreyt -- @Mavrin -- Shoot__0007-upscaled.jpg", "Mavrin"),
+        ("Irina Dreyt -- @Studio_Name-2.jpg", "Studio"),
+        ("Irina Dreyt -- Shoot__0007-upscaled.jpg", None),
+    ],
+)
+def test_studio_excludes_image_number_and_upscale_suffix(
+    filename: str,
+    expected_studio: str | None,
+) -> None:
+    assert immichup.parse_filename(image_path=Path(filename)) == (["Irina Dreyt"], expected_studio)
+
+
+@pytest.mark.parametrize(
+    "filename",
+    [
+        "Irina Dreyt -- @Mavrin__0007-upscaled.jpg",
+        "Irina Dreyt -- @MAVRIN__0155.jpg",
+    ],
+)
+def test_studio_album_and_tag_use_lowercase_name_without_image_suffix(filename: str) -> None:
+    path = Path(filename)
+    client = Mock(spec=immichup.ImmichClient)
+    with patch.object(immichup, "read_jpeg_dimensions", return_value=(600, 800)):
+        assert immichup.process_image(path=path, client=client, dry_run=False)
+    client.upload_image.assert_called_once_with(
+        path,
+        ["irina dreyt", "@mavrin"],
+        ["irina dreyt", "mavrin", "aspect:portrait"],
+    )
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(args=[str(Path(__file__).resolve()), *sys.argv[1:]]))