mirrors/wagtail
mirrors/wagtail
0
0
Fork 0
mirror of https://github.com/wagtail/wagtail.git synced 2024-12-01 11:41:20 +01:00
Code Issues Releases Wiki Activity

Added tests for jpegquality filter

Browse Source
This commit is contained in:
Karl Hobley 2016-06-17 09:39:02 +01:00 committed by Matt Westcott
parent af9d4ae5a3
commit a449e2d8b2
2 changed files with 94 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

90
wagtail/wagtailimages/tests/test_image_operations.py
View File

@ -2,16 +2,16 @@ from __future__ import absolute_import, unicode_literals
import warnings import warnings
from django.test import TestCase from django.test import TestCase, override_settings
from django.utils.six import BytesIO from django.utils.six import BytesIO
from mock import Mock from mock import Mock, patch
from wagtail.utils.deprecation import RemovedInWagtail19Warning from wagtail.utils.deprecation import RemovedInWagtail19Warning
from wagtail.wagtailcore import hooks from wagtail.wagtailcore import hooks
from wagtail.wagtailimages import image_operations from wagtail.wagtailimages import image_operations
from wagtail.wagtailimages.exceptions import InvalidFilterSpecError from wagtail.wagtailimages.exceptions import InvalidFilterSpecError
from wagtail.wagtailimages.models import Filter, Image from wagtail.wagtailimages.models import Filter, Image
from wagtail.wagtailimages.tests.utils import get_test_image_file from wagtail.wagtailimages.tests.utils import get_test_image_file, get_test_image_file_jpeg
class WillowOperationRecorder(object): class WillowOperationRecorder(object):
@ -487,3 +487,87 @@ class TestFormatFilter(TestCase):
file=get_test_image_file(), file=get_test_image_file(),
) )
self.assertRaises(InvalidFilterSpecError, fil.run, image, BytesIO()) self.assertRaises(InvalidFilterSpecError, fil.run, image, BytesIO())
class TestJPEGQualityFilter(TestCase):
def test_default_quality(self):
fil = Filter(spec='width-400')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
f = BytesIO()
with patch('PIL.Image.Image.save') as save:
fil.run(image, f)
save.assert_called_with(f, 'JPEG', quality=85, optimize=True, progressive=True)
def test_jpeg_quality_filter(self):
fil = Filter(spec='width-400|jpegquality-40')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
f = BytesIO()
with patch('PIL.Image.Image.save') as save:
fil.run(image, f)
save.assert_called_with(f, 'JPEG', quality=40, optimize=True, progressive=True)
def test_jpeg_quality_filter_invalid(self):
fil = Filter(spec='width-400|jpegquality-abc')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
self.assertRaises(InvalidFilterSpecError, fil.run, image, BytesIO())
def test_jpeg_quality_filter_no_value(self):
fil = Filter(spec='width-400|jpegquality')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
self.assertRaises(InvalidFilterSpecError, fil.run, image, BytesIO())
def test_jpeg_quality_filter_too_big(self):
fil = Filter(spec='width-400|jpegquality-101')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
self.assertRaises(InvalidFilterSpecError, fil.run, image, BytesIO())
@override_settings(
WAGTAILIMAGES_JPEG_QUALITY=50
)
def test_jpeg_quality_setting(self):
fil = Filter(spec='width-400')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
f = BytesIO()
with patch('PIL.Image.Image.save') as save:
fil.run(image, f)
save.assert_called_with(f, 'JPEG', quality=50, optimize=True, progressive=True)
@override_settings(
WAGTAILIMAGES_JPEG_QUALITY=50
)
def test_jpeg_quality_filter_overrides_setting(self):
fil = Filter(spec='width-400|jpegquality-40')
image = Image.objects.create(
title="Test image",
file=get_test_image_file_jpeg(),
)
f = BytesIO()
with patch('PIL.Image.Image.save') as save:
fil.run(image, f)
save.assert_called_with(f, 'JPEG', quality=40, optimize=True, progressive=True)

7
wagtail/wagtailimages/tests/utils.py
View File

@ -14,3 +14,10 @@ def get_test_image_file(filename='test.png', colour='white', size=(640, 480)):
image = PIL.Image.new('RGB', size, colour) image = PIL.Image.new('RGB', size, colour)
image.save(f, 'PNG') image.save(f, 'PNG')
return ImageFile(f, name=filename) return ImageFile(f, name=filename)
def get_test_image_file_jpeg(filename='test.jpg', colour='white', size=(640, 480)):
f = BytesIO()
image = PIL.Image.new('RGB', size, colour)
image.save(f, 'JPEG')
return ImageFile(f, name=filename)