diff options
| author | Tom Christie | 2011-05-19 08:36:55 +0100 |
|---|---|---|
| committer | Tom Christie | 2011-05-19 08:36:55 +0100 |
| commit | 8c3280f9c0d73c4e2536f1d757ad457b4a8f1de7 (patch) | |
| tree | bbcc27d1187e98c179342e0a7c75084d0c2d7934 /djangorestframework/tests/parsers.py | |
| parent | 49d4e50342cd3a6e2dce9f61d379cb54a0c1c6b5 (diff) | |
| download | django-rest-framework-8c3280f9c0d73c4e2536f1d757ad457b4a8f1de7.tar.bz2 | |
data flattening needs to go into resource
Diffstat (limited to 'djangorestframework/tests/parsers.py')
| -rw-r--r-- | djangorestframework/tests/parsers.py | 264 |
1 files changed, 132 insertions, 132 deletions
diff --git a/djangorestframework/tests/parsers.py b/djangorestframework/tests/parsers.py index 2720f4c7..764a8f5c 100644 --- a/djangorestframework/tests/parsers.py +++ b/djangorestframework/tests/parsers.py @@ -1,133 +1,133 @@ -""" -.. - >>> from djangorestframework.parsers import FormParser - >>> from djangorestframework.compat import RequestFactory - >>> from djangorestframework.views import BaseView - >>> from StringIO import StringIO - >>> from urllib import urlencode - >>> req = RequestFactory().get('/') - >>> some_view = BaseView() - >>> some_view.request = req # Make as if this request had been dispatched - -FormParser -============ - -Data flatening ----------------- - -Here is some example data, which would eventually be sent along with a post request : - - >>> inpt = urlencode([ - ... ('key1', 'bla1'), - ... ('key2', 'blo1'), ('key2', 'blo2'), - ... ]) - -Default behaviour for :class:`parsers.FormParser`, is to return a single value for each parameter : - - >>> (data, files) = FormParser(some_view).parse(StringIO(inpt)) - >>> data == {'key1': 'bla1', 'key2': 'blo1'} - True - -However, you can customize this behaviour by subclassing :class:`parsers.FormParser`, and overriding :meth:`parsers.FormParser.is_a_list` : - - >>> class MyFormParser(FormParser): - ... - ... def is_a_list(self, key, val_list): - ... return len(val_list) > 1 - -This new parser only flattens the lists of parameters that contain a single value. - - >>> (data, files) = MyFormParser(some_view).parse(StringIO(inpt)) - >>> data == {'key1': 'bla1', 'key2': ['blo1', 'blo2']} - True - -.. note:: The same functionality is available for :class:`parsers.MultiPartParser`. - -Submitting an empty list --------------------------- - -When submitting an empty select multiple, like this one :: - - <select multiple="multiple" name="key2"></select> - -The browsers usually strip the parameter completely. A hack to avoid this, and therefore being able to submit an empty select multiple, is to submit a value that tells the server that the list is empty :: - - <select multiple="multiple" name="key2"><option value="_empty"></select> - -:class:`parsers.FormParser` provides the server-side implementation for this hack. Considering the following posted data : - - >>> inpt = urlencode([ - ... ('key1', 'blo1'), ('key1', '_empty'), - ... ('key2', '_empty'), - ... ]) - -:class:`parsers.FormParser` strips the values ``_empty`` from all the lists. - - >>> (data, files) = MyFormParser(some_view).parse(StringIO(inpt)) - >>> data == {'key1': 'blo1'} - True - -Oh ... but wait a second, the parameter ``key2`` isn't even supposed to be a list, so the parser just stripped it. - - >>> class MyFormParser(FormParser): - ... - ... def is_a_list(self, key, val_list): - ... return key == 'key2' - ... - >>> (data, files) = MyFormParser(some_view).parse(StringIO(inpt)) - >>> data == {'key1': 'blo1', 'key2': []} - True - -Better like that. Note that you can configure something else than ``_empty`` for the empty value by setting :attr:`parsers.FormParser.EMPTY_VALUE`. -""" -import httplib, mimetypes -from tempfile import TemporaryFile -from django.test import TestCase -from djangorestframework.compat import RequestFactory -from djangorestframework.parsers import MultiPartParser -from djangorestframework.views import BaseView -from StringIO import StringIO - -def encode_multipart_formdata(fields, files): - """For testing multipart parser. - fields is a sequence of (name, value) elements for regular form fields. - files is a sequence of (name, filename, value) elements for data to be uploaded as files - Return (content_type, body).""" - BOUNDARY = '----------ThIs_Is_tHe_bouNdaRY_$' - CRLF = '\r\n' - L = [] - for (key, value) in fields: - L.append('--' + BOUNDARY) - L.append('Content-Disposition: form-data; name="%s"' % key) - L.append('') - L.append(value) - for (key, filename, value) in files: - L.append('--' + BOUNDARY) - L.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename)) - L.append('Content-Type: %s' % get_content_type(filename)) - L.append('') - L.append(value) - L.append('--' + BOUNDARY + '--') - L.append('') - body = CRLF.join(L) - content_type = 'multipart/form-data; boundary=%s' % BOUNDARY - return content_type, body - -def get_content_type(filename): - return mimetypes.guess_type(filename)[0] or 'application/octet-stream' - -class TestMultiPartParser(TestCase): - def setUp(self): - self.req = RequestFactory() - self.content_type, self.body = encode_multipart_formdata([('key1', 'val1'), ('key1', 'val2')], - [('file1', 'pic.jpg', 'blablabla'), ('file1', 't.txt', 'blobloblo')]) - - def test_multipartparser(self): - """Ensure that MultiPartParser can parse multipart/form-data that contains a mix of several files and parameters.""" - post_req = RequestFactory().post('/', self.body, content_type=self.content_type) - view = BaseView() - view.request = post_req - (data, files) = MultiPartParser(view).parse(StringIO(self.body)) - self.assertEqual(data['key1'], 'val1') - self.assertEqual(files['file1'].read(), 'blablabla') +# """ +# .. +# >>> from djangorestframework.parsers import FormParser +# >>> from djangorestframework.compat import RequestFactory +# >>> from djangorestframework.views import BaseView +# >>> from StringIO import StringIO +# >>> from urllib import urlencode +# >>> req = RequestFactory().get('/') +# >>> some_view = BaseView() +# >>> some_view.request = req # Make as if this request had been dispatched +# +# FormParser +# ============ +# +# Data flatening +# ---------------- +# +# Here is some example data, which would eventually be sent along with a post request : +# +# >>> inpt = urlencode([ +# ... ('key1', 'bla1'), +# ... ('key2', 'blo1'), ('key2', 'blo2'), +# ... ]) +# +# Default behaviour for :class:`parsers.FormParser`, is to return a single value for each parameter : +# +# >>> (data, files) = FormParser(some_view).parse(StringIO(inpt)) +# >>> data == {'key1': 'bla1', 'key2': 'blo1'} +# True +# +# However, you can customize this behaviour by subclassing :class:`parsers.FormParser`, and overriding :meth:`parsers.FormParser.is_a_list` : +# +# >>> class MyFormParser(FormParser): +# ... +# ... def is_a_list(self, key, val_list): +# ... return len(val_list) > 1 +# +# This new parser only flattens the lists of parameters that contain a single value. +# +# >>> (data, files) = MyFormParser(some_view).parse(StringIO(inpt)) +# >>> data == {'key1': 'bla1', 'key2': ['blo1', 'blo2']} +# True +# +# .. note:: The same functionality is available for :class:`parsers.MultiPartParser`. +# +# Submitting an empty list +# -------------------------- +# +# When submitting an empty select multiple, like this one :: +# +# <select multiple="multiple" name="key2"></select> +# +# The browsers usually strip the parameter completely. A hack to avoid this, and therefore being able to submit an empty select multiple, is to submit a value that tells the server that the list is empty :: +# +# <select multiple="multiple" name="key2"><option value="_empty"></select> +# +# :class:`parsers.FormParser` provides the server-side implementation for this hack. Considering the following posted data : +# +# >>> inpt = urlencode([ +# ... ('key1', 'blo1'), ('key1', '_empty'), +# ... ('key2', '_empty'), +# ... ]) +# +# :class:`parsers.FormParser` strips the values ``_empty`` from all the lists. +# +# >>> (data, files) = MyFormParser(some_view).parse(StringIO(inpt)) +# >>> data == {'key1': 'blo1'} +# True +# +# Oh ... but wait a second, the parameter ``key2`` isn't even supposed to be a list, so the parser just stripped it. +# +# >>> class MyFormParser(FormParser): +# ... +# ... def is_a_list(self, key, val_list): +# ... return key == 'key2' +# ... +# >>> (data, files) = MyFormParser(some_view).parse(StringIO(inpt)) +# >>> data == {'key1': 'blo1', 'key2': []} +# True +# +# Better like that. Note that you can configure something else than ``_empty`` for the empty value by setting :attr:`parsers.FormParser.EMPTY_VALUE`. +# """ +# import httplib, mimetypes +# from tempfile import TemporaryFile +# from django.test import TestCase +# from djangorestframework.compat import RequestFactory +# from djangorestframework.parsers import MultiPartParser +# from djangorestframework.views import BaseView +# from StringIO import StringIO +# +# def encode_multipart_formdata(fields, files): +# """For testing multipart parser. +# fields is a sequence of (name, value) elements for regular form fields. +# files is a sequence of (name, filename, value) elements for data to be uploaded as files +# Return (content_type, body).""" +# BOUNDARY = '----------ThIs_Is_tHe_bouNdaRY_$' +# CRLF = '\r\n' +# L = [] +# for (key, value) in fields: +# L.append('--' + BOUNDARY) +# L.append('Content-Disposition: form-data; name="%s"' % key) +# L.append('') +# L.append(value) +# for (key, filename, value) in files: +# L.append('--' + BOUNDARY) +# L.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename)) +# L.append('Content-Type: %s' % get_content_type(filename)) +# L.append('') +# L.append(value) +# L.append('--' + BOUNDARY + '--') +# L.append('') +# body = CRLF.join(L) +# content_type = 'multipart/form-data; boundary=%s' % BOUNDARY +# return content_type, body +# +# def get_content_type(filename): +# return mimetypes.guess_type(filename)[0] or 'application/octet-stream' +# +#class TestMultiPartParser(TestCase): +# def setUp(self): +# self.req = RequestFactory() +# self.content_type, self.body = encode_multipart_formdata([('key1', 'val1'), ('key1', 'val2')], +# [('file1', 'pic.jpg', 'blablabla'), ('file1', 't.txt', 'blobloblo')]) +# +# def test_multipartparser(self): +# """Ensure that MultiPartParser can parse multipart/form-data that contains a mix of several files and parameters.""" +# post_req = RequestFactory().post('/', self.body, content_type=self.content_type) +# view = BaseView() +# view.request = post_req +# (data, files) = MultiPartParser(view).parse(StringIO(self.body)) +# self.assertEqual(data['key1'], 'val1') +# self.assertEqual(files['file1'].read(), 'blablabla') |
