1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78
|
from __future__ import absolute_import, unicode_literals
import copy
from django.conf import settings
from django.http import HttpResponse
from django.test.utils import override_settings
from ..base import BaseTestCase
@override_settings(DEBUG_TOOLBAR_CONFIG={'INTERCEPT_REDIRECTS': True})
class RedirectsPanelTestCase(BaseTestCase):
def setUp(self):
super(RedirectsPanelTestCase, self).setUp()
self.panel = self.toolbar.get_panel_by_id('RedirectsPanel')
def test_regular_response(self):
response = self.panel.process_response(self.request, self.response)
self.assertTrue(response is self.response)
def test_not_a_redirect(self):
redirect = HttpResponse(status=304) # not modified
response = self.panel.process_response(self.request, redirect)
self.assertTrue(response is redirect)
def test_redirect(self):
redirect = HttpResponse(status=302)
redirect['Location'] = 'http://somewhere/else/'
response = self.panel.process_response(self.request, redirect)
self.assertFalse(response is redirect)
try:
self.assertContains(response, '302 Found')
except AssertionError: # Django < 1.9
self.assertContains(response, '302 FOUND')
self.assertContains(response, 'http://somewhere/else/')
def test_redirect_with_broken_context_processor(self):
TEMPLATES = copy.deepcopy(settings.TEMPLATES)
TEMPLATES[0]['OPTIONS']['context_processors'] = ['tests.context_processors.broken']
with self.settings(TEMPLATES=TEMPLATES):
redirect = HttpResponse(status=302)
redirect['Location'] = 'http://somewhere/else/'
response = self.panel.process_response(self.request, redirect)
self.assertFalse(response is redirect)
try:
self.assertContains(response, '302 Found')
except AssertionError: # Django < 1.9
self.assertContains(response, '302 FOUND')
self.assertContains(response, 'http://somewhere/else/')
def test_unknown_status_code(self):
redirect = HttpResponse(status=369)
redirect['Location'] = 'http://somewhere/else/'
response = self.panel.process_response(self.request, redirect)
try:
self.assertContains(response, '369 Unknown Status Code')
except AssertionError: # Django < 1.9
self.assertContains(response, '369 UNKNOWN STATUS CODE')
def test_unknown_status_code_with_reason(self):
redirect = HttpResponse(status=369, reason='Look Ma!')
redirect['Location'] = 'http://somewhere/else/'
response = self.panel.process_response(self.request, redirect)
self.assertContains(response, '369 Look Ma!')
def test_insert_content(self):
"""
Test that the panel only inserts content after generate_stats and
not the process_response.
"""
redirect = HttpResponse(status=304)
response = self.panel.process_response(self.request, redirect)
self.assertIsNotNone(response)
response = self.panel.generate_stats(self.request, redirect)
self.assertIsNone(response)
|