summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorBenjamin Kagia <kagia@server.fake>2013-04-19 20:20:23 +0300
committerTim Graham <timograham@gmail.com>2013-07-11 09:11:59 -0400
commitb0953dc91385fb2823294a76d3c99e01c7b7e4ee (patch)
tree86f129ff99ca6f8276e71eb7f811fc28596badc3 /tests
parentecd746191c428866e621b9d13ac90a116b1369dc (diff)
Fixed #13721 -- Added UploadedFile.content_type_extra.
Thanks Waldemar Kornewald and mvschaik for work on the patch.
Diffstat (limited to 'tests')
-rw-r--r--tests/file_uploads/tests.py21
-rw-r--r--tests/file_uploads/urls.py1
-rw-r--r--tests/file_uploads/views.py13
3 files changed, 34 insertions, 1 deletions
diff --git a/tests/file_uploads/tests.py b/tests/file_uploads/tests.py
index 66bc4465a2..f5a9c10be4 100644
--- a/tests/file_uploads/tests.py
+++ b/tests/file_uploads/tests.py
@@ -187,6 +187,27 @@ class FileUploadTests(TestCase):
got = json.loads(self.client.request(**r).content.decode('utf-8'))
self.assertTrue(len(got['file']) < 256, "Got a long file name (%s characters)." % len(got['file']))
+ def test_content_type_extra(self):
+ """Uploaded files may have content type parameters available."""
+ tdir = tempfile.gettempdir()
+
+ no_content_type = tempfile.NamedTemporaryFile(suffix=".ctype_extra", dir=tdir)
+ no_content_type.write(b'something')
+ no_content_type.seek(0)
+
+ simple_file = tempfile.NamedTemporaryFile(suffix=".ctype_extra", dir=tdir)
+ simple_file.write(b'something')
+ simple_file.seek(0)
+ simple_file.content_type = 'text/plain; test-key=test_value'
+
+ response = self.client.post('/file_uploads/echo_content_type_extra/', {
+ 'no_content_type': no_content_type,
+ 'simple_file': simple_file,
+ })
+ received = json.loads(response.content.decode('utf-8'))
+ self.assertEqual(received['no_content_type'], {})
+ self.assertEqual(received['simple_file'], {'test-key': 'test_value'})
+
def test_truncated_multipart_handled_gracefully(self):
"""
If passed an incomplete multipart message, MultiPartParser does not
diff --git a/tests/file_uploads/urls.py b/tests/file_uploads/urls.py
index fc5576828f..96efaaa5d8 100644
--- a/tests/file_uploads/urls.py
+++ b/tests/file_uploads/urls.py
@@ -10,6 +10,7 @@ urlpatterns = patterns('',
(r'^verify/$', views.file_upload_view_verify),
(r'^unicode_name/$', views.file_upload_unicode_name),
(r'^echo/$', views.file_upload_echo),
+ (r'^echo_content_type_extra/$', views.file_upload_content_type_extra),
(r'^echo_content/$', views.file_upload_echo_content),
(r'^quota/$', views.file_upload_quota),
(r'^quota/broken/$', views.file_upload_quota_broken),
diff --git a/tests/file_uploads/views.py b/tests/file_uploads/views.py
index eb7b654c09..d1bd2cf44e 100644
--- a/tests/file_uploads/views.py
+++ b/tests/file_uploads/views.py
@@ -7,7 +7,7 @@ import os
from django.core.files.uploadedfile import UploadedFile
from django.http import HttpResponse, HttpResponseServerError
from django.utils import six
-from django.utils.encoding import force_bytes
+from django.utils.encoding import force_bytes, smart_str
from .models import FileModel
from .tests import UNICODE_FILENAME, UPLOAD_TO
@@ -136,3 +136,14 @@ def file_upload_filename_case_view(request):
obj = FileModel()
obj.testfile.save(file.name, file)
return HttpResponse('%d' % obj.pk)
+
+def file_upload_content_type_extra(request):
+ """
+ Simple view to echo back extra content-type parameters.
+ """
+ params = {}
+ for file_name, uploadedfile in request.FILES.items():
+ params[file_name] = dict([
+ (k, smart_str(v)) for k, v in uploadedfile.content_type_extra.items()
+ ])
+ return HttpResponse(json.dumps(params))