diff options
Diffstat (limited to 'rest_framework/tests/htmlrenderer.py')
| -rw-r--r-- | rest_framework/tests/htmlrenderer.py | 9 | 
1 files changed, 5 insertions, 4 deletions
| diff --git a/rest_framework/tests/htmlrenderer.py b/rest_framework/tests/htmlrenderer.py index 702e8024..fd4ee2fd 100644 --- a/rest_framework/tests/htmlrenderer.py +++ b/rest_framework/tests/htmlrenderer.py @@ -4,6 +4,7 @@ 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 @@ -69,13 +70,13 @@ class TemplateHTMLRendererTests(TestCase):      def test_not_found_html_view(self):          response = self.client.get('/not_found') -        self.assertEquals(response.status_code, 404) +        self.assertEquals(response.status_code, status.HTTP_404_NOT_FOUND)          self.assertEquals(response.content, six.b("404 Not Found"))          self.assertEquals(response['Content-Type'], 'text/html')      def test_permission_denied_html_view(self):          response = self.client.get('/permission_denied') -        self.assertEquals(response.status_code, 403) +        self.assertEquals(response.status_code, status.HTTP_403_FORBIDDEN)          self.assertEquals(response.content, six.b("403 Forbidden"))          self.assertEquals(response['Content-Type'], 'text/html') @@ -106,12 +107,12 @@ class TemplateHTMLRendererExceptionTests(TestCase):      def test_not_found_html_view_with_template(self):          response = self.client.get('/not_found') -        self.assertEquals(response.status_code, 404) +        self.assertEquals(response.status_code, status.HTTP_404_NOT_FOUND)          self.assertEquals(response.content, six.b("404: Not found"))          self.assertEquals(response['Content-Type'], 'text/html')      def test_permission_denied_html_view_with_template(self):          response = self.client.get('/permission_denied') -        self.assertEquals(response.status_code, 403) +        self.assertEquals(response.status_code, status.HTTP_403_FORBIDDEN)          self.assertEquals(response.content, six.b("403: Permission denied"))          self.assertEquals(response['Content-Type'], 'text/html') | 
