def test_validates_images_within_max_file_size() -> None:
one_mb_file_size = 1 * 1024 * 1024
with generate_file_streams([('image.jpeg', one_mb_file_size),
('image.png', MAX_IMAGE_FILE_SIZE)
]) as file_streams:
images = Image(file_streams)
assert images.validate_size
assert images.validate
def test_does_not_validate_unallowed_image_types() -> None:
with generate_file_streams([('image.tiff', 0)]) as file_streams:
images = Image(file_streams)
assert not images.validate_type
with pytest.raises(
ImageError,
match='One or more image files are of invalid type'):
assert not images.validate
def test_can_generate_new_filenames() -> None:
with patch('talelio_backend.app_assets.domain.image_model.uuid4',
side_effect=['mock-uuid-123', 'mock-uuid-456']):
with generate_file_streams([('Image One.jpeg', 0),
('Image Two.png', 0)]) as file_streams:
images = Image(file_streams)
renamed_images = images.generate_new_filenames
assert renamed_images[0].name == 'image_one_mock-uuid-123.jpeg'
assert renamed_images[1].name == 'image_two_mock-uuid-456.png'
def test_does_not_validate_non_image_files() -> None:
with generate_file_streams([('image.jpeg', 0),
('document.pdf', 0)]) as file_streams:
images = Image(file_streams)
assert not images.validate_type
with pytest.raises(
ImageError,
match='One or more image files are of invalid type'):
assert not images.validate
def test_does_not_validate_images_above_max_file_size() -> None:
three_mb_file_size = 3 * 1024 * 1024
with generate_file_streams([('image.jpeg', three_mb_file_size)
]) as file_streams:
images = Image(file_streams)
assert not images.validate_size
with pytest.raises(ImageError,
match='One or more image file sizes are too large'):
assert not images.validate
def test_validates_allowed_image_types() -> None:
with generate_file_streams([('image.gif', 0), ('image.jpeg', 0),
('image.png', 0)]) as file_streams:
images = Image(file_streams)
assert images.validate_type
assert images.validate