def test_image_resize_success(
self,
mock_default_category: Category,
INITIAL_SIZE: Tuple[int, int],
FILE_EXTENTION: str,
):
"""
Test that the resizeImage function works as expected
when saving a Category object
"""
mock_default_category.image = f"dummy_image_base_name.{FILE_EXTENTION}"
create_dummy_png_image(
mock_default_category.image.name, IMAGE_SIZE=INITIAL_SIZE
)
check_image_attributes(
mock_default_category.image,
size_check=INITIAL_SIZE,
ext_check=f".{FILE_EXTENTION}",
)
mock_default_category.save()
check_image_attributes(
mock_default_category.image,
size_check=BaseModelMixin.CROP_SIZE,
ext_check=".jpg",
)
shutil.rmtree(Path(MEDIA_URL))
def test_image_resize_failed(
self,
mock_default_category: Category,
FILE_EXTENTION: str,
EXCEPTION: Exception,
):
"""
Test the expected Exception is raised when an invalid file format is submitted.
Clean up created test images
"""
mock_default_category.image = f"dummy_image_base_name.{FILE_EXTENTION}"
create_dummy_file(mock_default_category.image.name)
with pytest.raises(EXCEPTION):
mock_default_category.save()
shutil.rmtree(Path(MEDIA_URL))
