294 lines
11 KiB
Python
294 lines
11 KiB
Python
|
import os
|
||
|
|
import unittest
|
||
|
|
from io import BytesIO
|
||
|
|
from unittest.mock import patch
|
||
|
|
|
||
|
|
from django import VERSION
|
||
|
|
from django.conf import settings
|
||
|
|
from django.core.files.base import ContentFile
|
||
|
|
from django.core.files.storage import default_storage
|
||
|
|
|
||
|
|
from ..models import PHOTOLOGUE_CACHEDIRTAG, PHOTOLOGUE_DIR, Image, Photo
|
||
|
|
from .factories import (LANDSCAPE_IMAGE_PATH, NONSENSE_IMAGE_PATH, QUOTING_IMAGE_PATH, UNICODE_IMAGE_PATH,
|
||
|
|
GalleryFactory, PhotoEffectFactory, PhotoFactory)
|
||
|
|
from .helpers import PhotologueBaseTest
|
||
|
|
|
||
|
|
|
||
|
|
class PhotoTest(PhotologueBaseTest):
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
"""Delete any extra test files (if created)."""
|
||
|
|
super().tearDown()
|
||
|
|
try:
|
||
|
|
self.pl2.delete()
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_new_photo(self):
|
||
|
|
self.assertEqual(Photo.objects.count(), 1)
|
||
|
|
self.assertTrue(self.pl.image.storage.exists(self.pl.image.name))
|
||
|
|
self.assertEqual(self.pl.image.storage.size(self.pl.image.name),
|
||
|
|
os.path.getsize(LANDSCAPE_IMAGE_PATH))
|
||
|
|
|
||
|
|
# def test_exif(self):
|
||
|
|
# self.assertTrue(len(self.pl.EXIF.keys()) > 0)
|
||
|
|
|
||
|
|
def test_paths(self):
|
||
|
|
self.assertEqual(os.path.normpath(str(self.pl.cache_path())).lower(),
|
||
|
|
os.path.normpath(os.path.join(PHOTOLOGUE_DIR,
|
||
|
|
'photos',
|
||
|
|
'cache')).lower())
|
||
|
|
self.assertEqual(self.pl.cache_url(),
|
||
|
|
settings.MEDIA_URL + PHOTOLOGUE_DIR + '/photos/cache')
|
||
|
|
|
||
|
|
def test_cachedir_tag(self):
|
||
|
|
self.assertTrue(default_storage.exists(PHOTOLOGUE_CACHEDIRTAG))
|
||
|
|
|
||
|
|
content = default_storage.open(PHOTOLOGUE_CACHEDIRTAG).read()
|
||
|
|
self.assertEqual(content, b"Signature: 8a477f597d28d172789f06886806bc55")
|
||
|
|
|
||
|
|
def test_count(self):
|
||
|
|
for i in range(5):
|
||
|
|
self.pl.get_testPhotoSize_url()
|
||
|
|
self.assertEqual(self.pl.view_count, 0)
|
||
|
|
self.s.increment_count = True
|
||
|
|
self.s.save()
|
||
|
|
for i in range(5):
|
||
|
|
self.pl.get_testPhotoSize_url()
|
||
|
|
self.assertEqual(self.pl.view_count, 5)
|
||
|
|
|
||
|
|
def test_precache(self):
|
||
|
|
# set the thumbnail photo size to pre-cache
|
||
|
|
self.s.pre_cache = True
|
||
|
|
self.s.save()
|
||
|
|
# make sure it created the file
|
||
|
|
self.assertTrue(self.pl.image.storage.exists(
|
||
|
|
self.pl.get_testPhotoSize_filename()))
|
||
|
|
self.s.pre_cache = False
|
||
|
|
self.s.save()
|
||
|
|
# clear the cache and make sure the file's deleted
|
||
|
|
self.pl.clear_cache()
|
||
|
|
self.assertFalse(self.pl.image.storage.exists(
|
||
|
|
self.pl.get_testPhotoSize_filename()))
|
||
|
|
|
||
|
|
def test_accessor_methods(self):
|
||
|
|
self.assertEqual(self.pl.get_testPhotoSize_photosize(), self.s)
|
||
|
|
self.assertEqual(self.pl.get_testPhotoSize_size(),
|
||
|
|
Image.open(self.pl.image.storage.open(
|
||
|
|
self.pl.get_testPhotoSize_filename())).size)
|
||
|
|
self.assertEqual(self.pl.get_testPhotoSize_url(),
|
||
|
|
self.pl.cache_url() + '/' + self.pl._get_filename_for_size(self.s))
|
||
|
|
self.assertEqual(self.pl.get_testPhotoSize_filename(),
|
||
|
|
os.path.join(self.pl.cache_path(),
|
||
|
|
self.pl._get_filename_for_size(self.s)))
|
||
|
|
|
||
|
|
def test_quoted_url(self):
|
||
|
|
"""Test for issue #29 - filenames of photos are incorrectly quoted when
|
||
|
|
building a URL."""
|
||
|
|
|
||
|
|
# Create a Photo with a name that needs quoting.
|
||
|
|
self.pl2 = PhotoFactory(image__from_path=QUOTING_IMAGE_PATH)
|
||
|
|
# Quoting method filepath_to_uri has changed in Django 1.9 - so the string that we're looking
|
||
|
|
# for depends on the Django version.
|
||
|
|
if VERSION[0] == 1 and VERSION[1] <= 8:
|
||
|
|
quoted_string = 'test_photologue_%26quoting_testPhotoSize.jpg'
|
||
|
|
else:
|
||
|
|
quoted_string = 'test_photologue_quoting_testPhotoSize.jpg'
|
||
|
|
self.assertIn(quoted_string,
|
||
|
|
self.pl2.get_testPhotoSize_url(),
|
||
|
|
self.pl2.get_testPhotoSize_url())
|
||
|
|
|
||
|
|
def test_unicode(self):
|
||
|
|
"""Trivial check that unicode titles work.
|
||
|
|
(I was trying to track down an elusive unicode issue elsewhere)"""
|
||
|
|
self.pl2 = PhotoFactory(title='É',
|
||
|
|
slug='é')
|
||
|
|
|
||
|
|
@patch('photologue.models.ImageModel.resize_image')
|
||
|
|
def test_update_crop_applied(self, mock_resize_image):
|
||
|
|
self.assertEqual(1, Photo.objects.count())
|
||
|
|
self.assertTrue(self.pl.crop_from != 'right')
|
||
|
|
self.pl.crop_from = 'right'
|
||
|
|
self.pl.save()
|
||
|
|
self.assertTrue(mock_resize_image.called)
|
||
|
|
|
||
|
|
@patch('photologue.models.ImageModel.resize_image')
|
||
|
|
@patch('photologue.models.PhotoEffect.pre_process')
|
||
|
|
@patch('photologue.models.PhotoEffect.post_process')
|
||
|
|
def test_update_effect_applied(self, mock_post_process, mock_pre_process, mock_resize_image):
|
||
|
|
self.assertEqual(1, Photo.objects.count())
|
||
|
|
self.assertIsNone(self.pl.effect)
|
||
|
|
self.pl.effect = PhotoEffectFactory()
|
||
|
|
self.pl.save()
|
||
|
|
self.assertTrue(mock_pre_process.called)
|
||
|
|
self.assertTrue(mock_resize_image.called)
|
||
|
|
self.assertTrue(mock_post_process.called)
|
||
|
|
|
||
|
|
|
||
|
|
class PhotoManagerTest(PhotologueBaseTest):
|
||
|
|
"""Some tests for the methods on the Photo manager class."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Create 2 photos."""
|
||
|
|
super().setUp()
|
||
|
|
self.pl2 = PhotoFactory()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super().tearDown()
|
||
|
|
self.pl2.delete()
|
||
|
|
|
||
|
|
def test_public(self):
|
||
|
|
"""Method 'is_public' should only return photos flagged as public."""
|
||
|
|
self.assertEqual(Photo.objects.is_public().count(), 2)
|
||
|
|
self.pl.is_public = False
|
||
|
|
self.pl.save()
|
||
|
|
self.assertEqual(Photo.objects.is_public().count(), 1)
|
||
|
|
|
||
|
|
|
||
|
|
class PreviousNextTest(PhotologueBaseTest):
|
||
|
|
"""Tests for the methods that provide the previous/next photos in a gallery."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Create a test gallery with 2 photos."""
|
||
|
|
super().setUp()
|
||
|
|
self.test_gallery = GalleryFactory()
|
||
|
|
self.pl1 = PhotoFactory()
|
||
|
|
self.pl2 = PhotoFactory()
|
||
|
|
self.pl3 = PhotoFactory()
|
||
|
|
self.test_gallery.photos.add(self.pl1)
|
||
|
|
self.test_gallery.photos.add(self.pl2)
|
||
|
|
self.test_gallery.photos.add(self.pl3)
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super().tearDown()
|
||
|
|
self.pl1.delete()
|
||
|
|
self.pl2.delete()
|
||
|
|
self.pl3.delete()
|
||
|
|
|
||
|
|
def test_previous_simple(self):
|
||
|
|
# Previous in gallery.
|
||
|
|
self.assertEqual(self.pl1.get_previous_in_gallery(self.test_gallery),
|
||
|
|
None)
|
||
|
|
self.assertEqual(self.pl2.get_previous_in_gallery(self.test_gallery),
|
||
|
|
self.pl1)
|
||
|
|
self.assertEqual(self.pl3.get_previous_in_gallery(self.test_gallery),
|
||
|
|
self.pl2)
|
||
|
|
|
||
|
|
def test_previous_public(self):
|
||
|
|
"""What happens if one of the photos is not public."""
|
||
|
|
self.pl2.is_public = False
|
||
|
|
self.pl2.save()
|
||
|
|
|
||
|
|
self.assertEqual(self.pl1.get_previous_in_gallery(self.test_gallery),
|
||
|
|
None)
|
||
|
|
self.assertRaisesMessage(ValueError,
|
||
|
|
'Cannot determine neighbours of a non-public photo.',
|
||
|
|
self.pl2.get_previous_in_gallery,
|
||
|
|
self.test_gallery)
|
||
|
|
self.assertEqual(self.pl3.get_previous_in_gallery(self.test_gallery),
|
||
|
|
self.pl1)
|
||
|
|
|
||
|
|
def test_previous_gallery_mismatch(self):
|
||
|
|
"""Photo does not belong to the gallery."""
|
||
|
|
self.pl4 = PhotoFactory()
|
||
|
|
|
||
|
|
self.assertRaisesMessage(ValueError,
|
||
|
|
'Photo does not belong to gallery.',
|
||
|
|
self.pl4.get_previous_in_gallery,
|
||
|
|
self.test_gallery)
|
||
|
|
|
||
|
|
self.pl4.delete()
|
||
|
|
|
||
|
|
def test_next_simple(self):
|
||
|
|
# Next in gallery.
|
||
|
|
self.assertEqual(self.pl1.get_next_in_gallery(self.test_gallery),
|
||
|
|
self.pl2)
|
||
|
|
self.assertEqual(self.pl2.get_next_in_gallery(self.test_gallery),
|
||
|
|
self.pl3)
|
||
|
|
self.assertEqual(self.pl3.get_next_in_gallery(self.test_gallery),
|
||
|
|
None)
|
||
|
|
|
||
|
|
def test_next_public(self):
|
||
|
|
"""What happens if one of the photos is not public."""
|
||
|
|
self.pl2.is_public = False
|
||
|
|
self.pl2.save()
|
||
|
|
|
||
|
|
self.assertEqual(self.pl1.get_next_in_gallery(self.test_gallery),
|
||
|
|
self.pl3)
|
||
|
|
self.assertRaisesMessage(ValueError,
|
||
|
|
'Cannot determine neighbours of a non-public photo.',
|
||
|
|
self.pl2.get_next_in_gallery,
|
||
|
|
self.test_gallery)
|
||
|
|
self.assertEqual(self.pl3.get_next_in_gallery(self.test_gallery),
|
||
|
|
None)
|
||
|
|
|
||
|
|
def test_next_gallery_mismatch(self):
|
||
|
|
"""Photo does not belong to the gallery."""
|
||
|
|
self.pl4 = PhotoFactory()
|
||
|
|
|
||
|
|
self.assertRaisesMessage(ValueError,
|
||
|
|
'Photo does not belong to gallery.',
|
||
|
|
self.pl4.get_next_in_gallery,
|
||
|
|
self.test_gallery)
|
||
|
|
|
||
|
|
self.pl4.delete()
|
||
|
|
|
||
|
|
|
||
|
|
class ImageModelTest(PhotologueBaseTest):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super().setUp()
|
||
|
|
|
||
|
|
# Unicode image has unicode in the path
|
||
|
|
# self.pu = TestPhoto(name='portrait')
|
||
|
|
self.pu = PhotoFactory()
|
||
|
|
self.pu.image.save(os.path.basename(UNICODE_IMAGE_PATH),
|
||
|
|
ContentFile(open(UNICODE_IMAGE_PATH, 'rb').read()))
|
||
|
|
|
||
|
|
# Nonsense image contains nonsense
|
||
|
|
# self.pn = TestPhoto(name='portrait')
|
||
|
|
self.pn = PhotoFactory()
|
||
|
|
self.pn.image.save(os.path.basename(NONSENSE_IMAGE_PATH),
|
||
|
|
ContentFile(open(NONSENSE_IMAGE_PATH, 'rb').read()))
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super().tearDown()
|
||
|
|
self.pu.delete()
|
||
|
|
self.pn.delete()
|
||
|
|
|
||
|
|
@unittest.skipUnless(os.path.exists(UNICODE_IMAGE_PATH),
|
||
|
|
'Test relies on a file with a non-ascii filename - this cannot be distributed as it breaks '
|
||
|
|
'under Python 2.7, so the distribution does not include that test file.')
|
||
|
|
def test_create_size(self):
|
||
|
|
"""Nonsense image must not break scaling"""
|
||
|
|
self.pn.create_size(self.s)
|
||
|
|
|
||
|
|
|
||
|
|
def raw_image(mode='RGB', fmt='JPEG'):
|
||
|
|
"""Create raw image.
|
||
|
|
"""
|
||
|
|
data = BytesIO()
|
||
|
|
Image.new(mode, (100, 100)).save(data, fmt)
|
||
|
|
data.seek(0)
|
||
|
|
return data
|
||
|
|
|
||
|
|
|
||
|
|
class ImageTransparencyTest(PhotologueBaseTest):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super().setUp()
|
||
|
|
self.png = PhotoFactory()
|
||
|
|
self.png.image.save(
|
||
|
|
'trans.png', ContentFile(raw_image('RGBA', 'PNG').read()))
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
super().tearDown()
|
||
|
|
self.png.clear_cache()
|
||
|
|
os.unlink(os.path.join(settings.MEDIA_ROOT, self.png.image.path))
|
||
|
|
|
||
|
|
def test_create_size_png_keep_alpha_channel(self):
|
||
|
|
thumbnail = self.png.get_thumbnail_filename()
|
||
|
|
im = Image.open(
|
||
|
|
os.path.join(settings.MEDIA_ROOT, thumbnail))
|
||
|
|
self.assertEqual('RGBA', im.mode)
|