mirror of
https://github.com/Hopiu/django-imagekit.git
synced 2026-03-17 13:50:24 +00:00
124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
from django.conf import settings
|
|
from hashlib import md5
|
|
from imagekit.cachefiles import ImageCacheFile, LazyImageCacheFile
|
|
from imagekit.cachefiles.backends import Simple, CacheFileState
|
|
from imagekit.lib import force_bytes
|
|
from mock import Mock, patch
|
|
from nose.tools import raises, eq_, assert_raises
|
|
from .imagegenerators import TestSpec
|
|
from .utils import (assert_file_is_truthy, assert_file_is_falsy,
|
|
DummyAsyncCacheFileBackend, get_unique_image_file,
|
|
get_image_file)
|
|
|
|
|
|
def test_no_source_falsiness():
|
|
"""
|
|
Ensure cache files generated from sourceless specs are falsy.
|
|
|
|
"""
|
|
spec = TestSpec(source=None)
|
|
file = ImageCacheFile(spec)
|
|
assert_file_is_falsy(file)
|
|
|
|
|
|
def test_sync_backend_truthiness():
|
|
"""
|
|
Ensure that a cachefile with a synchronous cache file backend (the default)
|
|
is truthy.
|
|
|
|
"""
|
|
spec = TestSpec(source=get_unique_image_file())
|
|
file = ImageCacheFile(spec)
|
|
assert_file_is_truthy(file)
|
|
|
|
|
|
def test_async_backend_falsiness():
|
|
"""
|
|
Ensure that a cachefile with an asynchronous cache file backend is falsy.
|
|
|
|
"""
|
|
spec = TestSpec(source=get_unique_image_file())
|
|
file = ImageCacheFile(spec, cachefile_backend=DummyAsyncCacheFileBackend())
|
|
assert_file_is_falsy(file)
|
|
|
|
|
|
@raises(TestSpec.MissingSource)
|
|
def test_no_source_error():
|
|
spec = TestSpec(source=None)
|
|
file = ImageCacheFile(spec)
|
|
file.generate()
|
|
|
|
|
|
def test_memcached_cache_key():
|
|
"""
|
|
Ensure the default cachefile backend is sanitizing its cache key for
|
|
memcached by default.
|
|
|
|
"""
|
|
|
|
class MockFile(object):
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
backend = Simple()
|
|
extra_char_count = len('state-') + len(settings.IMAGEKIT_CACHE_PREFIX)
|
|
|
|
length = 199 - extra_char_count
|
|
filename = '1' * length
|
|
file = MockFile(filename)
|
|
eq_(backend.get_key(file), '%s%s-state' %
|
|
(settings.IMAGEKIT_CACHE_PREFIX, file.name))
|
|
|
|
length = 200 - extra_char_count
|
|
filename = '1' * length
|
|
file = MockFile(filename)
|
|
eq_(backend.get_key(file), '%s%s:%s' % (
|
|
settings.IMAGEKIT_CACHE_PREFIX,
|
|
'1' * (200 - len(':') - 32 - len(settings.IMAGEKIT_CACHE_PREFIX)),
|
|
md5(force_bytes('%s%s-state' % (settings.IMAGEKIT_CACHE_PREFIX, filename))).hexdigest()))
|
|
|
|
|
|
def test_lazyfile_stringification():
|
|
file = LazyImageCacheFile('testspec', source=None)
|
|
eq_(str(file), '')
|
|
eq_(repr(file), '<ImageCacheFile: None>')
|
|
|
|
source_file = get_image_file()
|
|
file = LazyImageCacheFile('testspec', source=source_file)
|
|
file.name = 'a.jpg'
|
|
eq_(str(file), 'a.jpg')
|
|
eq_(repr(file), '<ImageCacheFile: a.jpg>')
|
|
|
|
|
|
def test_error_on_dne():
|
|
"""
|
|
Trying to access one of the fields that requires the image should error if
|
|
the image does not exist.
|
|
|
|
"""
|
|
file = LazyImageCacheFile('testspec', source=get_image_file())
|
|
get_state_mock = Mock(return_value=CacheFileState.DOES_NOT_EXIST)
|
|
noop = Mock()
|
|
with patch.object(file.cachefile_backend, 'get_state', get_state_mock):
|
|
with patch.object(file.cachefile_backend, 'generate', noop):
|
|
with assert_raises(Exception): # TODO: Use more specific type
|
|
file.url
|
|
with assert_raises(Exception): # TODO: Use more specific type
|
|
file.width
|
|
|
|
|
|
def test_error_when_generating():
|
|
"""
|
|
Trying to access one of the fields that requires the image should error if
|
|
the image is being generated.
|
|
|
|
"""
|
|
file = LazyImageCacheFile('testspec', source=get_image_file())
|
|
get_state_mock = Mock(return_value=CacheFileState.GENERATING)
|
|
noop = Mock()
|
|
with patch.object(file.cachefile_backend, 'get_state', get_state_mock):
|
|
with patch.object(file.cachefile_backend, 'generate', noop):
|
|
with assert_raises(Exception): # TODO: Use more specific type
|
|
file.url
|
|
with assert_raises(Exception): # TODO: Use more specific type
|
|
file.width
|