django-rest-framework/djangorestframework/tests/content.py

80 lines
3.7 KiB
Python
Raw Normal View History

2011-04-11 16:22:54 +04:00
"""
Tests for content parsing, and form-overloaded content parsing.
"""
2011-04-11 14:21:41 +04:00
from django.test import TestCase
from djangorestframework.compat import RequestFactory
from djangorestframework.mixins import RequestMixin
2011-05-10 13:49:28 +04:00
from djangorestframework.parsers import FormParser, MultiPartParser, PlainTextParser
2011-04-11 16:13:25 +04:00
class TestContentParsing(TestCase):
2011-04-11 14:21:41 +04:00
def setUp(self):
self.req = RequestFactory()
def ensure_determines_no_content_GET(self, view):
2011-04-11 14:47:22 +04:00
"""Ensure view.RAW_CONTENT returns None for GET request with no content."""
2011-04-11 14:21:41 +04:00
view.request = self.req.get('/')
self.assertEqual(view.RAW_CONTENT, None)
2011-04-11 14:47:22 +04:00
def ensure_determines_form_content_POST(self, view):
2011-04-11 14:54:26 +04:00
"""Ensure view.RAW_CONTENT returns content for POST request with form content."""
2011-04-11 14:47:22 +04:00
form_data = {'qwerty': 'uiop'}
2011-05-10 13:49:28 +04:00
view.parsers = (FormParser, MultiPartParser)
2011-04-11 14:47:22 +04:00
view.request = self.req.post('/', data=form_data)
self.assertEqual(view.RAW_CONTENT, form_data)
2011-04-11 14:56:07 +04:00
def ensure_determines_non_form_content_POST(self, view):
2011-04-11 14:54:26 +04:00
"""Ensure view.RAW_CONTENT returns content for POST request with non-form content."""
content = 'qwerty'
content_type = 'text/plain'
view.parsers = (PlainTextParser,)
view.request = self.req.post('/', content, content_type=content_type)
2011-04-11 14:56:07 +04:00
self.assertEqual(view.RAW_CONTENT, content)
2011-04-11 14:54:26 +04:00
2011-04-11 14:56:07 +04:00
def ensure_determines_form_content_PUT(self, view):
2011-04-11 14:54:26 +04:00
"""Ensure view.RAW_CONTENT returns content for PUT request with form content."""
form_data = {'qwerty': 'uiop'}
2011-05-10 13:49:28 +04:00
view.parsers = (FormParser, MultiPartParser)
2011-04-11 14:54:26 +04:00
view.request = self.req.put('/', data=form_data)
self.assertEqual(view.RAW_CONTENT, form_data)
2011-04-11 14:56:07 +04:00
def ensure_determines_non_form_content_PUT(self, view):
2011-04-11 14:54:26 +04:00
"""Ensure view.RAW_CONTENT returns content for PUT request with non-form content."""
content = 'qwerty'
content_type = 'text/plain'
view.parsers = (PlainTextParser,)
view.request = self.req.post('/', content, content_type=content_type)
2011-04-11 14:56:07 +04:00
self.assertEqual(view.RAW_CONTENT, content)
2011-04-11 14:54:26 +04:00
2011-04-11 14:21:41 +04:00
def test_standard_behaviour_determines_no_content_GET(self):
2011-04-11 15:19:28 +04:00
"""Ensure view.RAW_CONTENT returns None for GET request with no content."""
2011-04-11 14:21:41 +04:00
self.ensure_determines_no_content_GET(RequestMixin())
2011-04-11 14:47:22 +04:00
def test_standard_behaviour_determines_form_content_POST(self):
2011-04-11 15:19:28 +04:00
"""Ensure view.RAW_CONTENT returns content for POST request with form content."""
2011-04-11 14:47:22 +04:00
self.ensure_determines_form_content_POST(RequestMixin())
2011-04-11 14:54:26 +04:00
def test_standard_behaviour_determines_non_form_content_POST(self):
2011-04-11 15:19:28 +04:00
"""Ensure view.RAW_CONTENT returns content for POST request with non-form content."""
2011-04-11 14:54:26 +04:00
self.ensure_determines_non_form_content_POST(RequestMixin())
def test_standard_behaviour_determines_form_content_PUT(self):
2011-04-11 15:19:28 +04:00
"""Ensure view.RAW_CONTENT returns content for PUT request with form content."""
2011-04-11 14:54:26 +04:00
self.ensure_determines_form_content_PUT(RequestMixin())
2011-04-11 15:19:28 +04:00
def test_standard_behaviour_determines_non_form_content_PUT(self):
"""Ensure view.RAW_CONTENT returns content for PUT request with non-form content."""
self.ensure_determines_non_form_content_PUT(RequestMixin())
def test_overloaded_behaviour_allows_content_tunnelling(self):
"""Ensure request.RAW_CONTENT returns content for overloaded POST request"""
content = 'qwerty'
content_type = 'text/plain'
view = RequestMixin()
form_data = {view.CONTENT_PARAM: content,
view.CONTENTTYPE_PARAM: content_type}
view.request = self.req.post('/', form_data)
view.parsers = (PlainTextParser,)
view.perform_form_overloading()
self.assertEqual(view.RAW_CONTENT, content)