def _image_with_thumbnail(self):
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
generate_thumbnail(image, 'file', 'thumbnail')
return image
def _image_with_thumbnail(self):
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
generate_thumbnail(image, 'file', 'thumbnail')
return image
def test_thumbnail_url_if_set(self):
"""thumbnail_url_if_set() returns self.thumbnail if set, or else
returns self.file"""
img = image()
eq_(img.file.url, img.thumbnail_url_if_set())
generate_thumbnail(img, 'file', 'thumbnail')
eq_(img.thumbnail.url, img.thumbnail_url_if_set())
def test_thumbnail_url_if_set(self, create_thumbnail_mock):
"""thumbnail_url_if_set() returns self.thumbnail if set, or else
returns self.file"""
img = ImageFactory()
eq_(img.file.url, img.thumbnail_url_if_set())
create_thumbnail_mock.return_value = ContentFile("the dude")
generate_thumbnail(img, "file", "thumbnail")
eq_(img.thumbnail.url, img.thumbnail_url_if_set())
def test_generate_deleted_file(self):
"""generate_thumbnail does not fail if file doesn't actually exist."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
# The field will be set but the file isn't there.
os.remove(image.file.path)
generate_thumbnail(image, 'file', 'thumbnail')
def test_generate_deleted_file(self):
"""generate_thumbnail does not fail if file doesn't actually exist."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
# The field will be set but the file isn't there.
os.remove(image.file.path)
generate_thumbnail(image, 'file', 'thumbnail')
def test_thumbnail_url_if_set(self, create_thumbnail_mock):
"""thumbnail_url_if_set() returns self.thumbnail if set, or else
returns self.file"""
img = ImageFactory()
eq_(img.file.url, img.thumbnail_url_if_set())
create_thumbnail_mock.return_value = ContentFile('the dude')
generate_thumbnail(img, 'file', 'thumbnail')
eq_(img.thumbnail.url, img.thumbnail_url_if_set())
def test_thumbnail_if_set(self):
"""thumbnail_if_set() returns self.thumbnail if set, or else returns
self.file"""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open("kitsune/upload/tests/media/test.jpg", "rb") as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
eq_(image.file, image.thumbnail_if_set())
generate_thumbnail(image, "file", "thumbnail")
eq_(image.thumbnail, image.thumbnail_if_set())
def test_thumbnail_if_set(self):
"""thumbnail_if_set() returns self.thumbnail if set, or else returns
self.file"""
image = ImageAttachment(content_object=self.obj, creator=self.user)
with open('kitsune/upload/tests/media/test.jpg') as f:
up_file = File(f)
image.file.save(up_file.name, up_file, save=True)
eq_(image.file, image.thumbnail_if_set())
generate_thumbnail(image, 'file', 'thumbnail')
eq_(image.thumbnail, image.thumbnail_if_set())
def test_generate_thumbnail_twice(self):
"""generate_thumbnail replaces old thumbnail."""
image = self._image_with_thumbnail()
old_path = image.thumbnail.path
# The thumbnail exists.
assert os.path.isfile(old_path)
generate_thumbnail(image, 'file', 'thumbnail')
new_path = image.thumbnail.path
# The thumbnail was replaced.
# Avoid potential failure when old thumbnail was saved 1 second ago.
assert os.path.isfile(new_path)
# Old file is either replaced or deleted.
if old_path != new_path:
assert not os.path.exists(old_path)
def test_generate_thumbnail_twice(self):
"""generate_thumbnail replaces old thumbnail."""
image = self._image_with_thumbnail()
old_path = image.thumbnail.path
# The thumbnail exists.
assert os.path.isfile(old_path)
generate_thumbnail(image, 'file', 'thumbnail')
new_path = image.thumbnail.path
# The thumbnail was replaced.
# Avoid potential failure when old thumbnail was saved 1 second ago.
assert os.path.isfile(new_path)
# Old file is either replaced or deleted.
if old_path != new_path:
assert not os.path.exists(old_path)
def test_generate_no_file(self):
"""generate_thumbnail does not fail when no file is provided."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
generate_thumbnail(image, 'file', 'thumbnail')
def test_generate_no_file(self):
"""generate_thumbnail does not fail when no file is provided."""
image = ImageAttachment(content_object=self.obj, creator=self.user)
generate_thumbnail(image, 'file', 'thumbnail')
