aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework/tests/negotiation.py
diff options
context:
space:
mode:
authorTom Christie2012-10-05 10:23:47 +0100
committerTom Christie2012-10-05 10:23:47 +0100
commitd07dc77e91c1f99b47915b3cef30b565f2618e82 (patch)
tree9f68ca55f685090f1672a557ce985931ccee47b3 /rest_framework/tests/negotiation.py
parentad5e6eb16f4db928e1fc8d0a6af4f9f4584f7b08 (diff)
downloaddjango-rest-framework-d07dc77e91c1f99b47915b3cef30b565f2618e82.tar.bz2
Accepted media type uses most specific of client/renderer media types.
Diffstat (limited to 'rest_framework/tests/negotiation.py')
-rw-r--r--rest_framework/tests/negotiation.py58
1 files changed, 58 insertions, 0 deletions
diff --git a/rest_framework/tests/negotiation.py b/rest_framework/tests/negotiation.py
new file mode 100644
index 00000000..dd9f6a76
--- /dev/null
+++ b/rest_framework/tests/negotiation.py
@@ -0,0 +1,58 @@
+from django.test import TestCase
+from django.test.client import RequestFactory
+from rest_framework.decorators import api_view, renderer_classes
+from rest_framework.negotiation import DefaultContentNegotiation
+from rest_framework.response import Response
+
+factory = RequestFactory()
+
+
+class MockJSONRenderer(object):
+ media_type = 'application/json'
+
+ def __init__(self, view):
+ pass
+
+
+class MockHTMLRenderer(object):
+ media_type = 'text/html'
+
+ def __init__(self, view):
+ pass
+
+
+@api_view(('GET',))
+@renderer_classes((MockJSONRenderer, MockHTMLRenderer))
+def example(request):
+ return Response()
+
+
+class TestAcceptedMediaType(TestCase):
+ def setUp(self):
+ self.renderers = [MockJSONRenderer(None), MockHTMLRenderer(None)]
+ self.negotiator = DefaultContentNegotiation()
+
+ def negotiate(self, request):
+ return self.negotiator.negotiate(request, self.renderers)
+
+ def test_client_without_accept_use_renderer(self):
+ request = factory.get('/')
+ accepted_renderer, accepted_media_type = self.negotiate(request)
+ self.assertEquals(accepted_media_type, 'application/json')
+
+ def test_client_underspecifies_accept_use_renderer(self):
+ request = factory.get('/', HTTP_ACCEPT='*/*')
+ accepted_renderer, accepted_media_type = self.negotiate(request)
+ self.assertEquals(accepted_media_type, 'application/json')
+
+ def test_client_overspecifies_accept_use_client(self):
+ request = factory.get('/', HTTP_ACCEPT='application/json; indent=8')
+ accepted_renderer, accepted_media_type = self.negotiate(request)
+ self.assertEquals(accepted_media_type, 'application/json; indent=8')
+
+
+class IntegrationTests(TestCase):
+ def test_accepted_negotiation_set_on_request(self):
+ request = factory.get('/', HTTP_ACCEPT='*/*')
+ response = example(request)
+ self.assertEquals(response.accepted_media_type, 'application/json')