Ticket #27777: 27777-test.diff

tests/file_storage/tests.py

@@ -670 +670 @@
         obj.normal.file.seek(0)
         obj.normal.close()
 
+    def test_filefield_context_manager(self):
+        obj = Storage.objects.create(normal=SimpleUploadedFile('rewritten.txt', b'content'))
+        with obj.normal as normal:
+            self.assertEqual(normal.size, 7)
+
+        with obj.normal as normal:
+            self.assertEqual(normal.size, 7)
+
     def test_duplicate_filename(self):
         # Multiple files with the same name get _(7 random chars) appended to them.
         objs = [Storage() for i in range(2)]