add test for unsupported image

mbourqui · Sep 2, 2022 · 59cf89c · 59cf89c
1 parent ab32aac
commit 59cf89c
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 1 deletion.
diff --git a/constrainedfilefield/tests/models.py b/constrainedfilefield/tests/models.py
@@ -28,7 +28,11 @@ class TestDocModel(models.Model):
 
 
 class TestImageModel(models.Model):
-    the_image = ConstrainedImageField(null=True, blank=True, upload_to="testfile",)
+    the_image = ConstrainedImageField(
+        null=True,
+        blank=True,
+        upload_to="testfile",
+        content_types=['image/png'],)
     the_image_small = ConstrainedImageField(
         null=True,
         blank=True,

diff --git a/constrainedfilefield/tests/sample_files/unsupported_image.jpg b/constrainedfilefield/tests/sample_files/unsupported_image.jpg
diff --git a/constrainedfilefield/tests/tests/backend.py b/constrainedfilefield/tests/tests/backend.py
@@ -224,6 +224,24 @@ def test_form_invalid_filetype(self):
             u"Allowed types are ['image/png'].",
         )
 
+    def test_form_invalid_image_filetype(self):
+        files = {
+            "the_image": self._create_simple_uploaded_file(
+                orig_filename="unsupported_image.jpg",
+                dest_filename="the_file.jpg",
+                content_type="image/jpeg",
+            ),
+        }
+        form = TestImageModelForm(data={}, files=files)
+        self.assertFalse(form.is_valid())
+        self.assertEqual(len(form.errors), 1)
+        self.assertEqual(len(form.errors["the_image"]), 1)
+        self.assertEqual(
+            form.errors["the_image"][0],
+            u"Unsupported file type: image/jpeg. "
+            u"Allowed types are ['image/png']."
+        )
+
     def test_form_invalid_filetype_and_size(self):
         form = self._create_bound_test_model_form(
             form_class=TestModelForm,