Ticket #7812: r8161-seek-testcase-fixed.patch

tests/regressiontests/file_uploads/views.py

@@ -78 +78 @@
     for key in request.FILES.keys():
         file_counts[key] = len(request.FILES.getlist(key))
     return HttpResponse(simplejson.dumps(file_counts))
+
+def file_upload_not_empty_read(request):
+    """
+    Check that the first .read() returns the whole file (because we don't specify
+    the size to read, so we read the entire file), while the second returns an
+    empty string
+    """
+    for key in request.FILES.keys():
+        if request.FILES[key].read() == '':
+            return HttpResponseServerError()
+        if request.FILES[key].read() != '':
+            return HttpResponseServerError()
+
+    return HttpResponse('')

tests/regressiontests/file_uploads/tests.py

@@ -187 +187 @@
         self.assertEqual(got.get('file1'), 1)
         self.assertEqual(got.get('file2'), 2)
 
+    def test_fileupload_seek(self):
+        # small file (under the 5M quota)
+        small_f = tempfile.NamedTemporaryFile()
+        small_f.write('a' * (2 ** 10))
+        small_f.seek(0)
+
+        response = self.client.post('/file_uploads/notempty/', {
+            'small_f': open(small_f.name)
+        })
+        self.assertEqual(response.status_code, 200)
+
+        # big file (over the 5M quota)
+        big_f = tempfile.NamedTemporaryFile()
+        big_f.write('a' * (10 * 2 ** 20))
+        big_f.seek(0)
+
+        response = self.client.post('/file_uploads/notempty/', {
+            'big_f': open(big_f.name)
+        })
+        self.assertEqual(response.status_code, 200)
+
 class DirectoryCreationTests(unittest.TestCase):
     """
     Tests for error handling during directory creation

tests/regressiontests/file_uploads/urls.py

@@ -8 +8 @@
     (r'^quota/$',           views.file_upload_quota),
     (r'^quota/broken/$',    views.file_upload_quota_broken),
     (r'^getlist_count/$',   views.file_upload_getlist_count),
+    (r'^notempty/$',        views.file_upload_not_empty_read),
 )