phase 5

backend/tests/images/test_processing.py

@@ -0,0 +1,79 @@
+"""Tests for image processing and thumbnail generation."""
+
+import io
+from uuid import uuid4
+
+import pytest
+from PIL import Image as PILImage
+
+from app.images.processing import generate_thumbnails
+
+
+class TestThumbnailGeneration:
+    """Tests for thumbnail generation."""
+
+    def test_generate_thumbnails_creates_all_sizes(self):
+        """Test that thumbnails are generated for all quality levels."""
+        # Create a test image
+        image_id = uuid4()
+        image = PILImage.new("RGB", (2000, 1500), color="red")
+        buffer = io.BytesIO()
+        image.save(buffer, format="JPEG")
+        contents = buffer.getvalue()
+
+        # Mock storage client to avoid actual uploads
+        from unittest.mock import MagicMock, patch
+
+        with patch("app.images.processing.get_storage_client") as mock_storage:
+            mock_storage.return_value.put_object = MagicMock()
+
+            # Generate thumbnails
+            thumbnail_paths = generate_thumbnails(image_id, "test/path.jpg", contents)
+
+            # Verify all sizes created
+            assert "low" in thumbnail_paths
+            assert "medium" in thumbnail_paths
+            assert "high" in thumbnail_paths
+
+            # Verify storage was called
+            assert mock_storage.return_value.put_object.call_count >= 2
+
+    def test_skip_thumbnail_for_small_images(self):
+        """Test that thumbnails are skipped if image is smaller than target size."""
+        # Create a small test image (smaller than low quality threshold)
+        image_id = uuid4()
+        image = PILImage.new("RGB", (500, 375), color="blue")
+        buffer = io.BytesIO()
+        image.save(buffer, format="JPEG")
+        contents = buffer.getvalue()
+
+        from unittest.mock import MagicMock, patch
+
+        with patch("app.images.processing.get_storage_client") as mock_storage:
+            mock_storage.return_value.put_object = MagicMock()
+
+            # Generate thumbnails
+            thumbnail_paths = generate_thumbnails(image_id, "test/small.jpg", contents)
+
+            # Should use original path for all sizes
+            assert thumbnail_paths["low"] == "test/small.jpg"
+
+    def test_handles_transparent_images(self):
+        """Test conversion of transparent images to RGB."""
+        # Create RGBA image
+        image_id = uuid4()
+        image = PILImage.new("RGBA", (2000, 1500), color=(255, 0, 0, 128))
+        buffer = io.BytesIO()
+        image.save(buffer, format="PNG")
+        contents = buffer.getvalue()
+
+        from unittest.mock import MagicMock, patch
+
+        with patch("app.images.processing.get_storage_client") as mock_storage:
+            mock_storage.return_value.put_object = MagicMock()
+
+            # Should not raise exception
+            thumbnail_paths = generate_thumbnails(image_id, "test/transparent.png", contents)
+
+            assert len(thumbnail_paths) > 0
+