Ticket #9591: 9591-test.diff

tests/regressiontests/file_uploads/views.py

@@ -4 +4 @@
 from django.utils import simplejson
 from models import FileModel
 from uploadhandler import QuotaUploadHandler, ErroringUploadHandler
+from django.core.files.uploadhandler import MemoryFileUploadHandler
 from django.utils.hashcompat import sha_constructor
 from tests import UNICODE_FILENAME
 
@@ -111 +112 @@
 def file_upload_errors(request):
     request.upload_handlers.insert(0, ErroringUploadHandler())
     return file_upload_echo(request)
+
+def file_upload_memory_position(request):
+    request.upload_handlers.insert(0, MemoryFileUploadHandler())
+    return HttpResponse(str(request.FILES['file_field'].tell()))
+

tests/regressiontests/file_uploads/tests.py

@@ -249 +249 @@
             # CustomUploadError is the error that should have been raised
             self.assertEqual(err.__class__, uploadhandler.CustomUploadError)
 
+    def test_memory_position(self):
+        post_data = {
+            'name': 'Ringo',
+            'file_field': open(__file__),
+        }
+        response = self.client.post('/file_uploads/memory_position/', post_data)
+        self.assertEqual(response.content, "0")
+
 class DirectoryCreationTests(unittest.TestCase):
     """
     Tests for error handling during directory creation

tests/regressiontests/file_uploads/urls.py

@@ -10 +10 @@
     (r'^quota/broken/$',    views.file_upload_quota_broken),
     (r'^getlist_count/$',   views.file_upload_getlist_count),
     (r'^upload_errors/$',   views.file_upload_errors),
+    (r'^memory_position/$', views.file_upload_memory_position),
 )
+