diff options
| author | Andreas Pelme | 2014-03-02 12:40:30 +0100 | 
|---|---|---|
| committer | Andreas Pelme | 2014-03-02 12:40:30 +0100 | 
| commit | 971578ca345c3d3bae7fd93b87c41d43483b6f05 (patch) | |
| tree | 72a6d3d0ced5750a8ba01bedc3a6f8a936e05e2f /tests/test_htmlrenderer.py | |
| parent | 62786a7ad64918022f11f1b95ce84adb8d798830 (diff) | |
| download | django-rest-framework-971578ca345c3d3bae7fd93b87c41d43483b6f05.tar.bz2 | |
Support for running the test suite with py.test
 * Get rid of runtests.py
 * Moved test code  from rest_framework/tests and rest_framework/runtests to tests
 * Invoke py.test from setup.py
 * Invoke py.test from Travis
 * Invoke py.test from tox
 * Changed setUpClass to be just plain setUp in test_permissions.py
 * Updated contribution guideline to show how to invoke py.test
Diffstat (limited to 'tests/test_htmlrenderer.py')
| -rw-r--r-- | tests/test_htmlrenderer.py | 118 | 
1 files changed, 118 insertions, 0 deletions
| diff --git a/tests/test_htmlrenderer.py b/tests/test_htmlrenderer.py new file mode 100644 index 00000000..c748fbdb --- /dev/null +++ b/tests/test_htmlrenderer.py @@ -0,0 +1,118 @@ +from __future__ import unicode_literals +from django.core.exceptions import PermissionDenied +from django.http import Http404 +from django.test import TestCase +from django.template import TemplateDoesNotExist, Template +import django.template.loader +from rest_framework import status +from rest_framework.compat import patterns, url +from rest_framework.decorators import api_view, renderer_classes +from rest_framework.renderers import TemplateHTMLRenderer +from rest_framework.response import Response +from rest_framework.compat import six + + +@api_view(('GET',)) +@renderer_classes((TemplateHTMLRenderer,)) +def example(request): +    """ +    A view that can returns an HTML representation. +    """ +    data = {'object': 'foobar'} +    return Response(data, template_name='example.html') + + +@api_view(('GET',)) +@renderer_classes((TemplateHTMLRenderer,)) +def permission_denied(request): +    raise PermissionDenied() + + +@api_view(('GET',)) +@renderer_classes((TemplateHTMLRenderer,)) +def not_found(request): +    raise Http404() + + +urlpatterns = patterns('', +    url(r'^$', example), +    url(r'^permission_denied$', permission_denied), +    url(r'^not_found$', not_found), +) + + +class TemplateHTMLRendererTests(TestCase): +    urls = 'tests.test_htmlrenderer' + +    def setUp(self): +        """ +        Monkeypatch get_template +        """ +        self.get_template = django.template.loader.get_template + +        def get_template(template_name): +            if template_name == 'example.html': +                return Template("example: {{ object }}") +            raise TemplateDoesNotExist(template_name) + +        django.template.loader.get_template = get_template + +    def tearDown(self): +        """ +        Revert monkeypatching +        """ +        django.template.loader.get_template = self.get_template + +    def test_simple_html_view(self): +        response = self.client.get('/') +        self.assertContains(response, "example: foobar") +        self.assertEqual(response['Content-Type'], 'text/html; charset=utf-8') + +    def test_not_found_html_view(self): +        response = self.client.get('/not_found') +        self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND) +        self.assertEqual(response.content, six.b("404 Not Found")) +        self.assertEqual(response['Content-Type'], 'text/html; charset=utf-8') + +    def test_permission_denied_html_view(self): +        response = self.client.get('/permission_denied') +        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN) +        self.assertEqual(response.content, six.b("403 Forbidden")) +        self.assertEqual(response['Content-Type'], 'text/html; charset=utf-8') + + +class TemplateHTMLRendererExceptionTests(TestCase): +    urls = 'tests.test_htmlrenderer' + +    def setUp(self): +        """ +        Monkeypatch get_template +        """ +        self.get_template = django.template.loader.get_template + +        def get_template(template_name): +            if template_name == '404.html': +                return Template("404: {{ detail }}") +            if template_name == '403.html': +                return Template("403: {{ detail }}") +            raise TemplateDoesNotExist(template_name) + +        django.template.loader.get_template = get_template + +    def tearDown(self): +        """ +        Revert monkeypatching +        """ +        django.template.loader.get_template = self.get_template + +    def test_not_found_html_view_with_template(self): +        response = self.client.get('/not_found') +        self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND) +        self.assertEqual(response.content, six.b("404: Not found")) +        self.assertEqual(response['Content-Type'], 'text/html; charset=utf-8') + +    def test_permission_denied_html_view_with_template(self): +        response = self.client.get('/permission_denied') +        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN) +        self.assertEqual(response.content, six.b("403: Permission denied")) +        self.assertEqual(response['Content-Type'], 'text/html; charset=utf-8') | 
