Pinry/django_images/tests.py

import mock
import qrcode
from django.test import TestCase
from django.core.files.images import ImageFile
from django.conf import settings
from django.utils.six import BytesIO
from django.core.urlresolvers import reverse
from django_images.models import Image, Thumbnail
from django_images.templatetags.images import at_size
from django_images.utils import scale_and_crop


class ImageModelTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))

    def test_get_by_size(self):
        size = list(settings.IMAGE_SIZES.keys())[0]
        thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)
        self.image.get_by_size(size)

    def test_get_absolute_url(self):
        url = self.image.get_absolute_url()
        self.assertEqual(url, self.image.image.url)
        # For thumbnail
        size = list(settings.IMAGE_SIZES.keys())[0]
        thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)
        url = self.image.get_absolute_url(size)
        self.assertEqual(url, thumb.image.url)
        # Fallback on creation url
        size = list(settings.IMAGE_SIZES.keys())[1]
        url = self.image.get_absolute_url(size)
        fallback_url = reverse('image-thumbnail', args=(self.image.id, size))
        self.assertEqual(url, fallback_url)


class ThumbnailManagerModelTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))
        self.size = list(settings.IMAGE_SIZES.keys())[0]

    def test_unknown_size(self):
        self.assertRaises(ValueError, Thumbnail.objects.get_or_create_at_size,
                          self.image.id, 'foo')

    # TODO: Test the image object and data
    def test_create(self):
        thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, self.size)
        self.assertEqual(self.image.thumbnail_set.count(), 1)

    def test_get(self):
        thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, self.size)
        thumb2 = Thumbnail.objects.get_or_create_at_size(self.image.id, self.size)
        self.assertEqual(thumb.id, thumb2.id)


class ThumbnailModelTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))
        size = list(settings.IMAGE_SIZES.keys())[0]
        self.thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)

    def test_get_absolute_url(self):
        url = self.thumb.get_absolute_url()
        self.assertEqual(url, self.thumb.image.url)


class PostSaveSignalOriginalChangedTestCase(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))
        size = list(settings.IMAGE_SIZES.keys())[0]
        self.thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)

    def test_post_save_signal_original_changed(self):
        size = list(settings.IMAGE_SIZES.keys())[0]
        thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)
        self.image.delete()
        self.assertFalse(Thumbnail.objects.exists())


class PostDeleteSignalDeleteImageFileTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))
        size = list(settings.IMAGE_SIZES.keys())[0]
        self.thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)

    @mock.patch('django_images.models.IMAGE_AUTO_DELETE', True)
    def test_post_delete_signal_delete_image_files_enabled(self):
        storage = self.image.image.storage
        image_name = self.image.image.name
        thumb_name = self.thumb.image.name
        self.image.delete()
        self.assertFalse(storage.exists(image_name))
        self.assertFalse(storage.exists(thumb_name))

    @mock.patch('django_images.models.IMAGE_AUTO_DELETE', False)
    def test_post_delete_signal_delete_image_files_disabled(self):
        storage = self.image.image.storage
        image_name = self.image.image.name
        thumb_name = self.thumb.image.name
        # Delete thumb
        self.thumb.delete()
        self.assertTrue(storage.exists(image_name))
        self.assertTrue(storage.exists(thumb_name))
        # Delete image
        self.image.delete()
        self.assertTrue(storage.exists(image_name))
        self.assertTrue(storage.exists(thumb_name))


class AtSizeTemplateTagTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))
        size = list(settings.IMAGE_SIZES.keys())[0]
        self.thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, size)

    def test_at_size(self):
        size = list(settings.IMAGE_SIZES.keys())[0]
        url = at_size(self.image, size)
        self.assertEqual(url, self.thumb.image.url)


class ThumbnailViewTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.image = Image.objects.create(width=370, height=370,
                                          image=ImageFile(image_obj, '01.png'))
        self.size = list(settings.IMAGE_SIZES.keys())[0]
        self.thumb = Thumbnail.objects.get_or_create_at_size(self.image.id, self.size)

    def test_redirect(self):
        url = reverse('image-thumbnail', args=[self.image.id, self.size])
        response = self.client.get(url)
        self.assertRedirects(response, self.thumb.image.url)

    def test_not_found(self):
        url = reverse('image-thumbnail', args=['42', self.size])
        response = self.client.get(url)
        self.assertEqual(response.status_code, 404)

    def test_size_not_found(self):
        url = reverse('image-thumbnail', args=[self.image.id, '42'])
        response = self.client.get(url)
        self.assertEqual(response.status_code, 404)


class UtilsScaleAndDropTest(TestCase):
    def setUp(self):
        image_obj = BytesIO()
        qrcode_obj = qrcode.make('https://mirumee.com/')
        qrcode_obj.save(image_obj)
        self.imagefile = ImageFile(image_obj, '01.png')

    def test_change_size(self):
        new_size = (10, 10)
        image = scale_and_crop(self.imagefile, new_size)
        self.assertEqual(new_size, image.im.size)

    def test_crop(self):
        new_size = (10, 10)
        image = scale_and_crop(self.imagefile, new_size, crop=True)
        self.assertEqual(new_size, image.im.size)

    def test_disabled_upscale(self):
        image = scale_and_crop(self.imagefile, (740, 740), upscale=False)
        self.assertLess(image.im.size[0], 371)
        self.assertLess(image.im.size[1], 371)

    def test_enaabled_upscale(self):
        image = scale_and_crop(self.imagefile, (740, 740), upscale=True)
        self.assertGreater(image.im.size[0], 371)
        self.assertGreater(image.im.size[1], 371)

    def test_not_change_quality(self):
        image = scale_and_crop(self.imagefile, (10, 10), quality=None)
        self.assertEqual(image.info.get('quality'), None)

    def test_change_quality(self):
        image = scale_and_crop(self.imagefile, (10, 10), quality=50)
        self.assertEqual(image.info.get('quality'), 50)