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

149 lines
4.7 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
2014-05-16 03:20:40 +04:00
import sys
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
2014-05-16 03:20:40 +04:00
if sys.version_info[:2] >= (3, 4):
JSON_ERROR = 'JSON parse error - Expecting value:'
else:
JSON_ERROR = 'JSON parse error - No JSON object could be decoded'
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
class ErrorView(APIView):
def get(self, request, *args, **kwargs):
raise Exception
@api_view(['GET'])
def error_view(request):
raise Exception
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)
2014-05-16 03:20:40 +04:00
chop = len(JSON_ERROR)
2012-10-11 14:21:50 +04:00
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 = {
2014-05-16 03:20:40 +04:00
'detail': JSON_ERROR
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 = {
2014-05-16 03:20:40 +04:00
'detail': JSON_ERROR
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 = {
2014-05-16 03:20:40 +04:00
'detail': JSON_ERROR
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 = {
2014-05-16 03:20:40 +04:00
'detail': JSON_ERROR
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)
class TestCustomExceptionHandler(TestCase):
def setUp(self):
self.DEFAULT_HANDLER = api_settings.EXCEPTION_HANDLER
def exception_handler(exc):
return Response('Error!', status=status.HTTP_400_BAD_REQUEST)
api_settings.EXCEPTION_HANDLER = exception_handler
def tearDown(self):
api_settings.EXCEPTION_HANDLER = self.DEFAULT_HANDLER
def test_class_based_view_exception_handler(self):
view = ErrorView.as_view()
request = factory.get('/', content_type='application/json')
response = view(request)
expected = 'Error!'
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
self.assertEqual(response.data, expected)
def test_function_based_view_exception_handler(self):
view = error_view
request = factory.get('/', content_type='application/json')
response = view(request)
expected = 'Error!'
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
self.assertEqual(response.data, expected)