def test_resize_no_upscaling(self): """ Ensure upscaling is forbidden in resize action. """ renderer = LazyThumbRenderer() mock_img = MockImg() img = renderer.resize(width=2000, height=2000, img=mock_img) self.assertEqual(img.size[0], 1000) self.assertEqual(img.size[1], 1000) self.assertEqual(len(mock_img.called), 0)
def test_resize(self): """ Test behavior of resize action. """ renderer = LazyThumbRenderer() mock_img = MockImg() img = renderer.resize(width=48, height=50, img=mock_img) self.assertEqual(img.size[1], 50) self.assertEqual(len(mock_img.called), 2) self.assertTrue('crop' in mock_img.called) self.assertTrue('resize' in mock_img.called)
def test_thumbnail_noop(self): """ Test that no image operations occur if the desired w/h match image's existing w/h """ renderer = LazyThumbRenderer() mock_img = MockImg() mock_img.size = (100, 100) img = renderer.thumbnail(width=100, img=mock_img) self.assertEqual(img.size[0], 100) self.assertEqual(img.size[1], 100) self.assertEqual(len(mock_img.called), 0)
def test_thumbnail_no_upscaling(self): """ Ensure that upscaling is forbidden in thumbnail action. """ renderer = LazyThumbRenderer() mock_img = MockImg() mock_img.size = (100, 100) img = renderer.thumbnail(width=20000, img=mock_img) self.assertEqual(img.size[0], 100) self.assertEqual(img.size[1], 100) self.assertEqual(len(mock_img.called), 0)
def test_thumbnail_square(self): """ Test behavior of thumbnail action when no width == height """ renderer = LazyThumbRenderer() mock_img = MockImg() mock_img.size = (100, 100) img = renderer.thumbnail(width=50, img=mock_img) self.assertEqual(img.size[0], 50) self.assertEqual(img.size[1], 50) self.assertEqual(len(mock_img.called), 1) self.assertTrue('resize' in mock_img.called)
def test_aresize_height(self): """ Ensure landscape aresize action thumbnails to height and center-crops. """ renderer = LazyThumbRenderer() mock_img = MockImg(width=2000, height=1000) mock_Image = Mock() # aresize 2000x1000 => 1000x800 should thumbnail to 1600x800 and then # paste into the center of a new image, losing some left/right content. with patch('lazythumbs.views.Image', mock_Image): img = renderer.aresize(width=1000, height=800, img=mock_img) self.assertEqual(mock_img.called, ['resize']) self.assertEqual(img, mock_Image.new.return_value) self.assertEqual(mock_Image.new.call_args[1]['size'], (1000, 800)) img.paste.assert_called_once_with(mock_img, (-300, 0))
def test_aresize_width(self): """ Ensure landscape aresize action thumbnails to width and center-crops. """ renderer = LazyThumbRenderer() mock_img = MockImg(width=1500, height=1000) mock_Image = Mock() # aresize 1500x1000 => 750x400 should thumbnail to 750x500 and then # paste into the center of a new image, losing some top/bottom content. with patch('lazythumbs.views.Image', mock_Image): img = renderer.aresize(width=750, height=400, img=mock_img) self.assertEqual(mock_img.called, ['resize']) self.assertEqual(img, mock_Image.new.return_value) self.assertEqual(mock_Image.new.call_args[1]['size'], (750, 400)) img.paste.assert_called_once_with(mock_img, (0, -50))
def test_aresize_no_crop_small(self): """ Ensure aresize_no_crop action does not grow a small image. """ renderer = LazyThumbRenderer() mock_img = MockImg(width=100, height=80) mock_Image = Mock() # aresize_no_crop 100x80 => 300x200 should not resize, but should paste # into the center of a new image, leaving matte background content # on the sides. with patch('lazythumbs.views.Image', mock_Image): img = renderer.aresize(width=300, height=200, img=mock_img) self.assertEqual(mock_img.called, []) self.assertEqual(img, mock_Image.new.return_value) self.assertEqual(mock_Image.new.call_args[1]['size'], (300, 200)) img.paste.assert_called_once_with(mock_img, (100, 60))
def test_aresize_portrait(self): """ Ensure aresize action from portrait to landscape shrinks image and mattes sides. """ renderer = LazyThumbRenderer() mock_img = MockImg(width=1000, height=2000) mock_Image = Mock() # aresize 1000x2000 => 600x500 should thumbnail to 250x500 and then # paste into the center of a new image, leaving matte background # content on the sides. with patch('lazythumbs.views.Image', mock_Image): img = renderer.aresize(width=600, height=500, img=mock_img) self.assertEqual(mock_img.called, ['resize']) self.assertEqual(img, mock_Image.new.return_value) self.assertEqual(mock_Image.new.call_args[1]['size'], (600, 500)) img.paste.assert_called_once_with(mock_img, (175, 0))
def test_aresize_no_crop_opposite_orientation_landscape(self): """ Ensure aresize_no_crop action when the source and target are the opposite orientation, and the source is landscape, that it scales according to width and mattes the top and bottom. """ renderer = LazyThumbRenderer() mock_img = MockImg(width=2000, height=1000) mock_Image = Mock() # aresize_no_crop 2000x1000 => 500x600 should thumbnail to 500x250 and # then paste into the center of a new image, leaving matte background # content on the top and bottom. with patch('lazythumbs.views.Image', mock_Image): img = renderer.aresize_no_crop(width=500, height=600, img=mock_img) self.assertEqual(mock_img.called, ['resize']) self.assertEqual(img, mock_Image.new.return_value) self.assertEqual(mock_Image.new.call_args[1]['size'], (500, 600)) img.paste.assert_called_once_with(mock_img, (0, 175))
def test_aresize_no_crop_opposite_orientation_portrait(self): """ Ensure aresize_no_crop action when the source and target are the opposite orientation, and the source is portrait, that it scales according to height and mattes sides. """ renderer = LazyThumbRenderer() mock_img = MockImg(width=1000, height=2000) mock_Image = Mock() # aresize_no_crop 1000x2000 => 600x500 should thumbnail to 250x500 and # then paste into the center of a new image, leaving matte background # content on the sides. with patch('lazythumbs.views.Image', mock_Image): img = renderer.aresize_no_crop(width=600, height=500, img=mock_img) self.assertEqual(mock_img.called, ['resize']) self.assertEqual(img, mock_Image.new.return_value) self.assertEqual(mock_Image.new.call_args[1]['size'], (600, 500)) img.paste.assert_called_once_with(mock_img, (175, 0))
def test_extensionless_gif(self): """If the image file is a GIF without an extension, we can produce a valid thumbnail for it.""" # Note: this test image file was breaking thumbnails if lazythumbs didn't # see the .gif extension. I tried creating a gif on the fly using # PIL but didn't hit the same problem, so it might be something about # this image that's special, maybe that it has a transparent background. # (The error was "cannot write mode P as JPEG"; the symptom was a 404 # response.) MEDIA_ROOT = tempfile.gettempdir() # Need to override MEDIA_ROOT in both django file storage and lazythumbs views # and Django doesn't provide override_settings until 1.4 with patch('django.core.files.storage.settings') as settings1: settings1.MEDIA_ROOT = MEDIA_ROOT with patch('lazythumbs.views.settings') as settings2: settings2.MEDIA_ROOT = MEDIA_ROOT testfile = TEST_IMG_GIF filename = None try: filename = os.path.join(MEDIA_ROOT, "gif_without_extension") shutil.copy(testfile, filename) # Now we have a gif file in a filename that doesn't end in .gif renderer = LazyThumbRenderer() source_path = os.path.relpath(filename, MEDIA_ROOT) rsp = renderer.get( request=Mock(path="/thumbnail/x50/" + source_path), action="thumbnail", geometry="x50", source_path=source_path ) # if you get 404, jpeg encoder is probably missing for Pillow self.assertEqual(200, rsp.status_code) finally: if filename: os.remove(filename)
""" {% lazythumb image.url thumbnail '48' as img_tag %} <img src="{{img_tag.src}}" width="{{img_tag.width}}" height="{{img_tag.height}} /> {% endlazythumb %} {% lazythumb image.url resize '150x200' %} <img src="{{img_tag.src}}" width="{{img_tag.width}}" height="{{img_tag.height}} /> {% endlazythumb %} """ import logging from django.template import TemplateSyntaxError, Library, Node, Variable from lazythumbs.util import compute_img, get_attr_string from lazythumbs.views import LazyThumbRenderer SUPPORTED_ACTIONS = LazyThumbRenderer().allowed_actions register = Library() logger = logging.getLogger() register.tag('lazythumb', lambda p, t: LazythumbNode(p, t)) class LazythumbNode(Node): usage = 'Expected invocation is {% lazythumb url|ImageFile|Object action geometry [**kwargs] as variable %}' def __init__(self, parser, token): # simple alias tse = lambda m: TemplateSyntaxError('lazythumb: %s' % m) bits = token.contents.split() try:
def test_thumbnail_no_img(self): renderer = LazyThumbRenderer() self.assertRaises(ValueError, renderer.thumbnail, 200, 200)
def test_thumbnail_no_width_and_no_height_specified(self): renderer = LazyThumbRenderer() mock_img = MockImg() mock_img.size = (100, 100) self.assertRaises(ValueError, renderer.thumbnail, img=mock_img)
def test_no_img(self): renderer = LazyThumbRenderer() self.assertRaises(ValueError, renderer.scale, 200, 200)
def test_maximum_width_and_height(self): renderer = LazyThumbRenderer() new_img = renderer.scale(1000, 1000, img_path=TEST_IMG_GIF) self.assertEqual(new_img.size, (399, 499))
def test_new_img(self): renderer = LazyThumbRenderer() new_img = renderer.matte(200, 200, img_path=TEST_IMG_GIF) self.assertEqual(new_img.size, (200, 200))
def setUp(self): self.renderer = LazyThumbRenderer() self.mock_Image = Mock() self.mock_img = Mock() self.mock_Image.open = Mock(return_value=self.mock_img) self.mock_img.size = [1,1]