class ThumbnailerFieldTest(BaseTest):
def setUp(self):
BaseTest.setUp(self)
self.storage = TemporaryStorage()
# Save a test image.
data = StringIO()
Image.new('RGB', (800, 600)).save(data, 'JPEG')
data.seek(0)
image_file = ContentFile(data.read())
self.storage.save('avatars/avatar.jpg', image_file)
# Set the test model to use the current temporary storage.
TestModel._meta.get_field('avatar').storage = self.storage
TestModel._meta.get_field('avatar').thumbnail_storage = self.storage
def tearDown(self):
self.storage.delete_temporary_storage()
BaseTest.tearDown(self)
def test_generate_thumbnail(self):
instance = TestModel(avatar='avatars/avatar.jpg')
thumb = instance.avatar.generate_thumbnail({'size': (300, 300)})
self.assertEqual((thumb.width, thumb.height), (300, 225))
def test_delete(self):
instance = TestModel(avatar='avatars/avatar.jpg')
instance.avatar.get_thumbnail({'size': (300, 300)})
instance.avatar.get_thumbnail({'size': (200, 200)})
instance.avatar.get_thumbnail({'size': (100, 100)})
self.assertEqual(len(self.storage.listdir('avatars')[1]), 4)
instance.avatar.delete(save=False)
self.assertEqual(len(self.storage.listdir('avatars')[1]), 0)
def test_delete_thumbnails(self):
instance = TestModel(avatar='avatars/avatar.jpg')
instance.avatar.get_thumbnail({'size': (300, 300)})
instance.avatar.get_thumbnail({'size': (200, 200)})
instance.avatar.get_thumbnail({'size': (100, 100)})
self.assertEqual(len(self.storage.listdir('avatars')[1]), 4)
instance.avatar.delete_thumbnails()
self.assertEqual(len(self.storage.listdir('avatars')[1]), 1)
def test_get_thumbnails(self):
instance = TestModel(avatar='avatars/avatar.jpg')
instance.avatar.get_thumbnail({'size': (300, 300)})
instance.avatar.get_thumbnail({'size': (200, 200)})
self.assertEqual(len(list(instance.avatar.get_thumbnails())), 2)
def test_thumbnail_tag(self):
instance = TestModel(avatar='avatars/avatar.jpg')
thumb = instance.avatar.get_thumbnail({'size': (300, 300)})
self.assertTrue(len(thumb.tag()) > 0)
class ThumbnailerFieldTest(BaseTest):
def setUp(self):
BaseTest.setUp(self)
self.storage = TemporaryStorage()
# Save a test image.
data = StringIO()
Image.new('RGB', (800, 600)).save(data, 'JPEG')
data.seek(0)
image_file = ContentFile(data.read())
self.storage.save('avatars/avatar.jpg', image_file)
# Set the test model to use the current temporary storage.
TestModel._meta.get_field('avatar').storage = self.storage
TestModel._meta.get_field('avatar').thumbnail_storage = self.storage
def tearDown(self):
self.storage.delete_temporary_storage()
BaseTest.tearDown(self)
def test_generate_thumbnail(self):
instance = TestModel(avatar='avatars/avatar.jpg')
thumb = instance.avatar.generate_thumbnail({'size': (300, 300)})
self.assertEqual((thumb.width, thumb.height), (300, 225))
def test_generate_thumbnail_type_error(self):
text_file = ContentFile("Lorem ipsum dolor sit amet. Not an image.")
self.storage.save('avatars/invalid.jpg', text_file)
instance = TestModel(avatar='avatars/invalid.jpg')
generate = lambda: instance.avatar.generate_thumbnail(
{'size': (300, 300)})
self.assertRaises(InvalidImageFormatError, generate)
def test_delete(self):
instance = TestModel(avatar='avatars/avatar.jpg')
thumb = instance.avatar.get_thumbnail({'size': (300, 300)})
self.assertEqual((thumb.width, thumb.height), (300, 225))
instance.avatar.delete(save=False)
self.assertEqual(self.storage.listdir('avatars')[1], [])
def test_get_thumbnails(self):
instance = TestModel(avatar='avatars/avatar.jpg')
instance.avatar.get_thumbnail({'size': (300, 300)})
instance.avatar.get_thumbnail({'size': (200, 200)})
self.assertEqual(len(list(instance.avatar.get_thumbnails())), 2)
