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

from django.test import TestCase
from django.test.client import RequestFactory
from rest_framework.negotiation import DefaultContentNegotiation

factory = RequestFactory()


class MockJSONRenderer(object):
    media_type = 'application/json'


class MockHTMLRenderer(object):
    media_type = 'text/html'


class TestAcceptedMediaType(TestCase):
    def setUp(self):
        self.renderers = [MockJSONRenderer(), MockHTMLRenderer()]
        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')