class RedisFilesTest(TestCase):
def setUp(self):
self.source_name = "tests.png"
self.size = "small"
self.instance = TestModel.objects.create()
with open('thumbnails/tests/tests.png', 'rb') as image_file:
self.instance.avatar = File(image_file)
self.instance.save()
self.backend = RedisBackend()
self.instance.avatar.thumbnails.metadata_backend = self.backend
self.avatar_folder = \
os.path.join(self.instance.avatar.storage.temporary_location, conf.BASE_DIR, 'avatars')
self.instance.avatar.thumbnails.small
self.basename = os.path.basename(self.instance.avatar.path)
self.filename, self.ext = os.path.splitext(self.basename)
def tearDown(self):
key = self.backend.get_thumbnail_key(self.instance.avatar.name)
self.backend.redis.delete(key)
self.instance.avatar.storage.delete_temporary_storage()
super(RedisFilesTest, self).tearDown()
def test_delete_with_thumbnails(self):
avatar_path = self.instance.avatar.path
# ensure file still exists
self.assertTrue(os.path.exists(avatar_path))
key = self.backend.get_thumbnail_key(self.instance.avatar.name)
self.assertTrue(self.backend.redis.exists(key))
self.instance.avatar.delete(with_thumbnails=True)
# image file is deleted
self.assertFalse(os.path.exists(avatar_path))
# thumbnails and their metadata are also deleted
self.assertEqual(len(os.listdir(self.avatar_folder)), 0)
self.assertFalse(self.backend.redis.exists(key))
def test_delete_all(self):
thumbnails = self.instance.avatar.thumbnails
key = self.backend.get_thumbnail_key(self.instance.avatar.name)
self.assertTrue(self.backend.redis.exists(key))
thumbnails.delete_all()
# thumbnails and their metadata are deleted
self.assertEqual(len(os.listdir(self.avatar_folder)), 0)
self.assertFalse(self.backend.redis.exists(key))
class RedisBackendTest(TestCase):
def setUp(self):
self.backend = RedisBackend()
self.redis = StrictRedis()
def test_get_source_key(self):
self.assertEqual(self.backend.get_source_key("a.jpg"), "djthumbs-test:sources:a.jpg")
def test_get_thumbnail_key(self):
self.assertEqual(self.backend.get_thumbnail_key("a.jpg"), "djthumbs-test:thumbnails:a.jpg")
def test_add_delete_source(self):
source_name = "test-thumbnail.jpg"
source_key = self.backend.get_source_key(source_name)
self.backend.add_source(source_name)
self.assertTrue(self.redis.hexists(source_key, source_name))
self.backend.delete_source(source_name)
self.assertFalse(self.redis.hexists(source_key, source_name))
def test_get_source(self):
source_name = "test-thumbnail.jpg"
source_key = self.backend.get_source_key(source_name)
self.redis.hset(source_key, source_name, source_name)
self.assertEqual(self.backend.get_source(source_name), source_name)
# Delete Source
self.redis.hdel(source_key, source_name)
def test_add_delete_thumbnail(self):
source_name = "test-thumbnail.jpg"
size = "small"
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, size, "test-thumbnail_small.jpg")
self.assertTrue(self.redis.hexists(thumbnail_key, size))
self.backend.delete_thumbnail(source_name, size)
self.assertFalse(self.redis.hexists(thumbnail_key, size))
# Delete Source
self.redis.hdel(self.backend.get_source_key(source_name), source_name)
def test_get_thumbnail(self):
source_name = "test-thumbnail.jpg"
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, "small", "test-thumbnail_small.jpg")
self.assertEqual(
self.backend.get_thumbnail(source_name, "small"),
ImageMeta(source_name, "test-thumbnail_small.jpg", "small"),
)
self.backend.add_thumbnail(source_name, "large", "test-thumbnail_large.jpg")
expected = ["test-thumbnail_large.jpg", "test-thumbnail_small.jpg"]
result = [image_meta.name for image_meta in self.backend.get_thumbnails(source_name)]
self.assertEqual(result.sort(), expected.sort())
# Delete Source & Thumbnails
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.redis.hdel(self.backend.get_source_key(source_name), source_name)
self.redis.hdel(thumbnail_key, "small")
self.redis.hdel(thumbnail_key, "large")
class RedisBackendTest(TestCase):
def setUp(self):
self.backend = RedisBackend()
self.redis = StrictRedis()
def test_get_source_key(self):
self.assertEqual(self.backend.get_source_key('a.jpg'), 'djthumbs-test:sources:a.jpg')
def test_get_thumbnail_key(self):
self.assertEqual(self.backend.get_thumbnail_key('a.jpg'), 'djthumbs-test:thumbnails:a.jpg')
def test_add_delete_source(self):
source_name = 'test-thumbnail.jpg'
source_key = self.backend.get_source_key(source_name)
self.backend.add_source(source_name)
self.assertTrue(self.redis.hexists(source_key, source_name))
self.backend.delete_source(source_name)
self.assertFalse(self.redis.hexists(source_key, source_name))
def test_get_source(self):
source_name = 'test-thumbnail.jpg'
source_key = self.backend.get_source_key(source_name)
self.redis.hset(source_key, source_name, source_name)
self.assertEqual(self.backend.get_source(source_name), source_name)
# Delete Source
self.redis.hdel(source_key, source_name)
def test_add_delete_thumbnail(self):
source_name = 'test-thumbnail.jpg'
size = 'small'
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, size, 'test-thumbnail_small.jpg')
self.assertTrue(self.redis.hexists(thumbnail_key, size))
self.backend.delete_thumbnail(source_name, size)
self.assertFalse(self.redis.hexists(thumbnail_key, size))
# Delete Source
self.redis.hdel(self.backend.get_source_key(source_name), source_name)
def test_get_thumbnail(self):
source_name = 'test-thumbnail.jpg'
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, 'small', 'test-thumbnail_small.jpg')
self.assertEqual(self.backend.get_thumbnail(source_name, 'small'), ImageMeta(source_name, 'test-thumbnail_small.jpg', 'small'))
self.backend.add_thumbnail(source_name, 'large', 'test-thumbnail_large.jpg')
expected = ['test-thumbnail_large.jpg', 'test-thumbnail_small.jpg']
result = [image_meta.name for image_meta in self.backend.get_thumbnails(source_name)]
self.assertEqual(result.sort(), expected.sort())
# Delete Source & Thumbnails
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.redis.hdel(self.backend.get_source_key(source_name), source_name)
self.redis.hdel(thumbnail_key, 'small')
self.redis.hdel(thumbnail_key, 'large')
class RedisBackendTest(TestCase):
def setUp(self):
self.backend = RedisBackend()
self.redis = StrictRedis()
def test_get_source_key(self):
self.assertEqual(self.backend.get_source_key('a.jpg'),
'djthumbs-test:sources:a.jpg')
def test_get_thumbnail_key(self):
self.assertEqual(self.backend.get_thumbnail_key('a.jpg'),
'djthumbs-test:thumbnails:a.jpg')
def test_add_delete_source(self):
source_name = 'test-thumbnail.jpg'
source_key = self.backend.get_source_key(source_name)
self.backend.add_source(source_name)
self.assertTrue(self.redis.hexists(source_key, source_name))
self.backend.delete_source(source_name)
self.assertFalse(self.redis.hexists(source_key, source_name))
def test_get_source(self):
source_name = 'test-thumbnail.jpg'
source_key = self.backend.get_source_key(source_name)
self.redis.hset(source_key, source_name, source_name)
self.assertEqual(self.backend.get_source(source_name), source_name)
# Delete Source
self.redis.hdel(source_key, source_name)
def test_add_delete_thumbnail(self):
source_name = 'test-thumbnail.jpg'
size = 'small'
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, size,
'test-thumbnail_small.jpg')
self.assertTrue(self.redis.hexists(thumbnail_key, size))
self.backend.delete_thumbnail(source_name, size)
self.assertFalse(self.redis.hexists(thumbnail_key, size))
# Delete Source
self.redis.hdel(self.backend.get_source_key(source_name), source_name)
def test_get_thumbnail(self):
source_name = 'test-thumbnail.jpg'
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, 'small',
'test-thumbnail_small.jpg')
self.assertEqual(
self.backend.get_thumbnail(source_name, 'small'),
ImageMeta(source_name, 'test-thumbnail_small.jpg', 'small'))
self.backend.add_thumbnail(source_name, 'large',
'test-thumbnail_large.jpg')
expected = ['test-thumbnail_large.jpg', 'test-thumbnail_small.jpg']
result = [
image_meta.name
for image_meta in self.backend.get_thumbnails(source_name)
]
# sort is replacing the variable in place, not returning new value, it will always return None
result.sort()
expected.sort()
self.assertEqual(result, expected)
# Delete Source & Thumbnails
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.redis.hdel(self.backend.get_source_key(source_name), source_name)
self.redis.hdel(thumbnail_key, 'small')
self.redis.hdel(thumbnail_key, 'large')
def test_flush_thumbnails(self):
source_name = 'test-thumbnail.jpg'
thumbnail_key = self.backend.get_thumbnail_key(source_name)
self.backend.add_source(source_name)
self.backend.add_thumbnail(source_name, "small",
'test-thumbnail_small.jpg')
self.backend.add_thumbnail(source_name, "default",
'test-thumbnail_default.jpg')
self.backend.flush_thumbnails(source_name)
self.assertFalse(self.redis.exists(thumbnail_key))
