80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
"""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
|
|
|