|
@@ -239,6 +239,22 @@ class InconsistentGetImageDimensionsBug(unittest.TestCase):
|
|
|
self.assertEqual(size, Image.open(fh).size)
|
|
|
|
|
|
|
|
|
+class GetImageDimensionsOnInvalidImages(unittest.TestCase):
|
|
|
+ @unittest.skipUnless(Image, "Pillow not installed")
|
|
|
+ def test_invalid_image(self):
|
|
|
+ """
|
|
|
+ get_image_dimensions() should return (None, None) for the dimensions of
|
|
|
+ invalid images (#24441).
|
|
|
+
|
|
|
+ brokenimg.png is not a valid image and it has been generated by:
|
|
|
+ $ echo "123" > brokenimg.png
|
|
|
+ """
|
|
|
+ img_path = os.path.join(os.path.dirname(upath(__file__)), "brokenimg.png")
|
|
|
+ with open(img_path, 'rb') as fh:
|
|
|
+ size = images.get_image_dimensions(fh)
|
|
|
+ self.assertEqual(size, (None, None))
|
|
|
+
|
|
|
+
|
|
|
class FileMoveSafeTests(unittest.TestCase):
|
|
|
def test_file_move_overwrite(self):
|
|
|
handle_a, self.file_a = tempfile.mkstemp()
|