def test_middleware_response_insertion(self):
def get_response(request):
return regular_view(request, "İ")
response = DebugToolbarMiddleware(get_response)(self.request)
# check toolbar insertion before "</body>"
self.assertContains(response, "</div>\n</body>")
def test_middleware_no_injection_when_encoded(self):
def get_response(request):
response = HttpResponse("<html><body></body></html>")
response["Content-Encoding"] = "something"
return response
response = DebugToolbarMiddleware(get_response)(self.request)
self.assertEqual(response.content, b"<html><body></body></html>")
def test_tuple_urlconf(self):
request = rf.get('/')
urls = __import__('tests.urls').urls
urls.urlpatterns = tuple(urls.urlpatterns)
request.urlconf = urls
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True):
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, basestring))
def test_show_toolbar_TEST(self):
request = rf.get('/')
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], TEST=True, DEBUG=True):
self.assertFalse(middleware._show_toolbar(request))
with Settings(INTERNAL_IPS=['127.0.0.1'], TEST=False, DEBUG=True):
self.assertTrue(middleware._show_toolbar(request))
def test_show_toolbar_TEST(self):
request = self.request
middleware = DebugToolbarMiddleware()
with Settings(TEST=True, DEBUG=True):
self.assertFalse(middleware._show_toolbar(request))
with Settings(TEST=False, DEBUG=True):
self.assertTrue(middleware._show_toolbar(request))
def test_tuple_urlconf(self):
request = rf.get('/')
urls = __import__('tests.urls').urls
urls.urlpatterns = tuple(urls.urlpatterns)
request.urlconf = urls
middleware = DebugToolbarMiddleware()
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, six.string_types))
def test_show_toolbar_INTERNAL_IPS(self):
request = rf.get('/')
request.META = {'REMOTE_ADDR': '127.0.0.1'}
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True):
self.assertTrue(middleware._show_toolbar(request))
with Settings(INTERNAL_IPS=[], DEBUG=True):
self.assertFalse(middleware._show_toolbar(request))
def test_show_toolbar_TEST(self):
request = self.request
request.META = {'REMOTE_ADDR': '127.0.0.1'}
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], TEST=True, DEBUG=True):
self.assertFalse(middleware._show_toolbar(request))
with Settings(INTERNAL_IPS=['127.0.0.1'], TEST=False, DEBUG=True):
self.assertTrue(middleware._show_toolbar(request))
def test_request_urlconf_module(self):
request = rf.get('/')
request.urlconf = __import__('tests.urls').urls
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True):
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, string_types))
self.assertTrue(hasattr(request.urlconf.urlpatterns[1], '_callback_str'))
self.assertEqual(request.urlconf.urlpatterns[-1]._callback_str, 'tests.views.execute_sql')
def test_request_urlconf_module(self):
request = rf.get('/')
request.urlconf = __import__('tests.urls').urls
middleware = DebugToolbarMiddleware()
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, six.string_types))
patterns = request.urlconf.urlpatterns
self.assertTrue(hasattr(patterns[1], '_callback_str'))
self.assertEqual(patterns[-1]._callback_str, 'tests.views.execute_sql')
def test_request_urlconf_string(self):
request = rf.get('/')
request.urlconf = 'tests.urls'
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True):
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, basestring))
self.assertTrue(hasattr(request.urlconf.urlpatterns[0], '_callback_str'))
self.assertEquals(request.urlconf.urlpatterns[0]._callback_str, 'debug_toolbar.views.debug_media')
self.assertTrue(hasattr(request.urlconf.urlpatterns[1], '_callback_str'))
self.assertEquals(request.urlconf.urlpatterns[-1]._callback_str, 'tests.views.execute_sql')
def test_request_urlconf_string(self):
request = self.request
request.urlconf = 'debug_toolbar.tests.urls'
request.META = {'REMOTE_ADDR': '127.0.0.1'}
middleware = DebugToolbarMiddleware()
with Settings(DEBUG=True):
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, basestring))
self.assertTrue(hasattr(request.urlconf.urlpatterns[0], '_callback_str'))
self.assertEquals(request.urlconf.urlpatterns[0]._callback_str, 'debug_toolbar.views.debug_media')
self.assertEquals(request.urlconf.urlpatterns[-1].urlconf_name.__name__, 'debug_toolbar.tests.urls')
def test_request_urlconf_string_per_request(self):
request = rf.get('/')
request.urlconf = 'debug_toolbar.urls'
middleware = DebugToolbarMiddleware()
with self.settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True):
middleware.process_request(request)
request.urlconf = 'tests.urls'
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, six.string_types))
patterns = request.urlconf.urlpatterns
self.assertTrue(hasattr(patterns[1], '_callback_str'))
self.assertEqual(patterns[-1]._callback_str, 'tests.views.execute_sql')
def test_request_urlconf_module(self):
request = self.request
request.urlconf = __import__('tests.urls').urls
request.META = {'REMOTE_ADDR': '127.0.0.1'}
middleware = DebugToolbarMiddleware()
with Settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True):
middleware.process_request(request)
self.assertFalse(isinstance(request.urlconf, basestring))
self.assertTrue(
hasattr(request.urlconf.urlpatterns[0], '_callback_str'))
self.assertEquals(request.urlconf.urlpatterns[0]._callback_str,
'debug_toolbar.views.debug_media')
self.assertTrue(
hasattr(request.urlconf.urlpatterns[1], '_callback_str'))
self.assertEquals(request.urlconf.urlpatterns[-1]._callback_str,
'tests.views.execute_sql')
def test_middleware_response_insertion(self):
resp = regular_view(self.request, "İ")
DebugToolbarMiddleware().process_response(self.request, resp)
# check toolbar insertion before "</body>"
self.assertContains(resp, '</div>\n</body>')
def test_middleware_response_only(self):
DebugToolbarMiddleware().process_response(self.request, self.response)
def test_middleware_view_only(self):
DebugToolbarMiddleware().process_view(self.request, regular_view,
('title', ), {})
