Ticket #7654: file_uploads_tests.patch

tests/regressiontests/file_uploads/views.py

@@ -67 +67 @@
     """
     response = file_upload_echo(request)
     request.upload_handlers.insert(0, QuotaUploadHandler())
+    return response
+
+def file_upload_quota_broken_when_looking_at_post(request):
+    """
+    You can't change handlers after reading POST; this view shouldn't work.
+    """
+    post_copy = request.POST.copy()
+    request.upload_handlers.insert(0, QuotaUploadHandler())
+    response = file_upload_echo(request)
     return response
+ No newline at end of file

tests/regressiontests/file_uploads/tests.py

@@ -154 +154 @@
             self.client.post,
             '/file_uploads/quota/broken/', 
             {'f': open(f.name)}
-        )        
+        )
+    
+    def test_broken_custom_upload_handler_when_examining_post(self):
+        f = tempfile.NamedTemporaryFile()
+        f.write('a' * (2 ** 21))
+        
+        # AttributeError: You cannot alter upload handlers after the upload has been processed.
+        self.assertRaises(
+            AttributeError,
+            self.client.post,
+            '/file_uploads/quota/broken/post/', 
+            {'f': open(f.name)}
+        )
         
+ No newline at end of file

tests/regressiontests/file_uploads/urls.py

@@ -7 +7 @@
     (r'^echo/$',            views.file_upload_echo),
     (r'^quota/$',           views.file_upload_quota),
     (r'^quota/broken/$',    views.file_upload_quota_broken),
+    (r'^quota/broken/post/$',    views.file_upload_quota_broken_when_looking_at_post),
 )