diff options
| author | Benjamin Kagia <kagia@server.fake> | 2013-04-19 20:20:23 +0300 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2013-07-11 09:11:59 -0400 |
| commit | b0953dc91385fb2823294a76d3c99e01c7b7e4ee (patch) | |
| tree | 86f129ff99ca6f8276e71eb7f811fc28596badc3 /tests | |
| parent | ecd746191c428866e621b9d13ac90a116b1369dc (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.py | 21 | ||||
| -rw-r--r-- | tests/file_uploads/urls.py | 1 | ||||
| -rw-r--r-- | tests/file_uploads/views.py | 13 |
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)) |
