django-rest-framework/rest_framework/tests/test_views.py

102 lines
3.5 KiB
Python
Raw Normal View History

2012-11-22 03:20:49 +04:00
from __future__ import unicode_literals
2013-05-19 17:06:24 +04:00
import copy
2012-10-08 20:10:50 +04:00
from django.test import TestCase
from rest_framework import status
from rest_framework.decorators import api_view
from rest_framework.response import Response
from rest_framework.settings import api_settings
2013-06-28 20:17:39 +04:00
from rest_framework.test import APIRequestFactory
2012-10-08 20:10:50 +04:00
from rest_framework.views import APIView
2013-05-19 17:06:24 +04:00
2013-06-28 20:17:39 +04:00
factory = APIRequestFactory()
2012-10-08 20:10:50 +04:00
class BasicView(APIView):
def get(self, request, *args, **kwargs):
return Response({'method': 'GET'})
def post(self, request, *args, **kwargs):
return Response({'method': 'POST', 'data': request.DATA})
@api_view(['GET', 'POST', 'PUT', 'PATCH'])
2012-10-08 20:10:50 +04:00
def basic_view(request):
if request.method == 'GET':
return {'method': 'GET'}
elif request.method == 'POST':
return {'method': 'POST', 'data': request.DATA}
elif request.method == 'PUT':
return {'method': 'PUT', 'data': request.DATA}
elif request.method == 'PATCH':
return {'method': 'PATCH', 'data': request.DATA}
2012-10-08 20:10:50 +04:00
2012-10-11 14:21:50 +04:00
def sanitise_json_error(error_dict):
"""
Exact contents of JSON error messages depend on the installed version
of json.
"""
ret = copy.copy(error_dict)
chop = len('JSON parse error - No JSON object could be decoded')
ret['detail'] = ret['detail'][:chop]
return ret
2012-10-08 20:10:50 +04:00
class ClassBasedViewIntegrationTests(TestCase):
def setUp(self):
self.view = BasicView.as_view()
def test_400_parse_error(self):
request = factory.post('/', 'f00bar', content_type='application/json')
response = self.view(request)
expected = {
2012-11-22 03:20:49 +04:00
'detail': 'JSON parse error - No JSON object could be decoded'
2012-10-08 20:10:50 +04:00
}
2013-02-28 01:15:00 +04:00
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
self.assertEqual(sanitise_json_error(response.data), expected)
2012-10-08 20:10:50 +04:00
def test_400_parse_error_tunneled_content(self):
content = 'f00bar'
content_type = 'application/json'
form_data = {
api_settings.FORM_CONTENT_OVERRIDE: content,
api_settings.FORM_CONTENTTYPE_OVERRIDE: content_type
}
request = factory.post('/', form_data)
response = self.view(request)
expected = {
2012-11-22 03:20:49 +04:00
'detail': 'JSON parse error - No JSON object could be decoded'
2012-10-08 20:10:50 +04:00
}
2013-02-28 01:15:00 +04:00
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
self.assertEqual(sanitise_json_error(response.data), expected)
2012-10-08 20:10:50 +04:00
class FunctionBasedViewIntegrationTests(TestCase):
def setUp(self):
self.view = basic_view
def test_400_parse_error(self):
request = factory.post('/', 'f00bar', content_type='application/json')
response = self.view(request)
expected = {
2012-11-22 03:20:49 +04:00
'detail': 'JSON parse error - No JSON object could be decoded'
2012-10-08 20:10:50 +04:00
}
2013-02-28 01:15:00 +04:00
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
self.assertEqual(sanitise_json_error(response.data), expected)
2012-10-08 20:10:50 +04:00
def test_400_parse_error_tunneled_content(self):
content = 'f00bar'
content_type = 'application/json'
form_data = {
api_settings.FORM_CONTENT_OVERRIDE: content,
api_settings.FORM_CONTENTTYPE_OVERRIDE: content_type
}
request = factory.post('/', form_data)
response = self.view(request)
expected = {
2012-11-22 03:20:49 +04:00
'detail': 'JSON parse error - No JSON object could be decoded'
2012-10-08 20:10:50 +04:00
}
2013-02-28 01:15:00 +04:00
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
self.assertEqual(sanitise_json_error(response.data), expected)