diff options
| author | James Rutherford | 2015-03-11 10:38:03 +0000 | 
|---|---|---|
| committer | James Rutherford | 2015-03-11 10:38:03 +0000 | 
| commit | 4a2d27975ab5249269aebafd803be87a2107092b (patch) | |
| tree | 55b524c93b02eef404304f734be98871bbb1324f /tests/test_negotiation.py | |
| parent | 856dc855c952746f566a6a8de263afe951362dfb (diff) | |
| parent | dc56e5a0f41fdd6350e91a5749023d086bd1640f (diff) | |
| download | django-rest-framework-4a2d27975ab5249269aebafd803be87a2107092b.tar.bz2 | |
Merge pull request #1 from tomchristie/master
Merge in from upstream
Diffstat (limited to 'tests/test_negotiation.py')
| -rw-r--r-- | tests/test_negotiation.py | 45 | 
1 files changed, 45 insertions, 0 deletions
| diff --git a/tests/test_negotiation.py b/tests/test_negotiation.py new file mode 100644 index 00000000..04b89eb6 --- /dev/null +++ b/tests/test_negotiation.py @@ -0,0 +1,45 @@ +from __future__ import unicode_literals +from django.test import TestCase +from rest_framework.negotiation import DefaultContentNegotiation +from rest_framework.request import Request +from rest_framework.renderers import BaseRenderer +from rest_framework.test import APIRequestFactory + + +factory = APIRequestFactory() + + +class MockJSONRenderer(BaseRenderer): +    media_type = 'application/json' + + +class MockHTMLRenderer(BaseRenderer): +    media_type = 'text/html' + + +class NoCharsetSpecifiedRenderer(BaseRenderer): +    media_type = 'my/media' + + +class TestAcceptedMediaType(TestCase): +    def setUp(self): +        self.renderers = [MockJSONRenderer(), MockHTMLRenderer()] +        self.negotiator = DefaultContentNegotiation() + +    def select_renderer(self, request): +        return self.negotiator.select_renderer(request, self.renderers) + +    def test_client_without_accept_use_renderer(self): +        request = Request(factory.get('/')) +        accepted_renderer, accepted_media_type = self.select_renderer(request) +        self.assertEqual(accepted_media_type, 'application/json') + +    def test_client_underspecifies_accept_use_renderer(self): +        request = Request(factory.get('/', HTTP_ACCEPT='*/*')) +        accepted_renderer, accepted_media_type = self.select_renderer(request) +        self.assertEqual(accepted_media_type, 'application/json') + +    def test_client_overspecifies_accept_use_client(self): +        request = Request(factory.get('/', HTTP_ACCEPT='application/json; indent=8')) +        accepted_renderer, accepted_media_type = self.select_renderer(request) +        self.assertEqual(accepted_media_type, 'application/json; indent=8') | 
