Преглед на файлове

Fixed #25108 -- Fixed a test which failed on Pillow 2.9+

Yosuke Yasuda преди 9 години
родител
ревизия
a2b999dfca
променени са 1 файла, в които са добавени 14 реда и са изтрити 10 реда
  1. 14 10
      tests/forms_tests/tests/test_fields.py

+ 14 - 10
tests/forms_tests/tests/test_fields.py

@@ -847,19 +847,23 @@ class FieldsTests(SimpleTestCase):
         This also tests the situation when Pillow doesn't detect the MIME type
         of the image (#24948).
         """
-        f = ImageField()
-
-        img_path = os.path.dirname(upath(__file__)) + '/filepath_test_files/1x1.bmp'
-        with open(img_path, 'rb') as img_file:
-            img_data = img_file.read()
+        from PIL.BmpImagePlugin import BmpImageFile
+        try:
+            Image.register_mime(BmpImageFile.format, None)
+            f = ImageField()
+            img_path = os.path.dirname(upath(__file__)) + '/filepath_test_files/1x1.bmp'
+            with open(img_path, 'rb') as img_file:
+                img_data = img_file.read()
 
-        img_file = SimpleUploadedFile('1x1.bmp', img_data)
-        img_file.content_type = 'text/plain'
+            img_file = SimpleUploadedFile('1x1.bmp', img_data)
+            img_file.content_type = 'text/plain'
 
-        uploaded_file = f.clean(img_file)
+            uploaded_file = f.clean(img_file)
 
-        self.assertEqual('BMP', uploaded_file.image.format)
-        self.assertIsNone(uploaded_file.content_type)
+            self.assertEqual('BMP', uploaded_file.image.format)
+            self.assertIsNone(uploaded_file.content_type)
+        finally:
+            Image.register_mime(BmpImageFile.format, 'image/bmp')
 
     # URLField ##################################################################