diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/tests.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tests/tests.py b/tests/tests.py index c074c23..154615a 100644 --- a/tests/tests.py +++ b/tests/tests.py @@ -40,18 +40,22 @@ class Settings(object): class BaseTestCase(TestCase): def setUp(self): request = Dingus('request') + response = Dingus('response') toolbar = DebugToolbar(request) DebugToolbarMiddleware.debug_toolbars[thread.get_ident()] = toolbar self.request = request + self.response = response self.toolbar = toolbar + self.toolbar.stats = {} class DebugToolbarTestCase(BaseTestCase): urls = 'tests.urls' def test_middleware(self): - resp = self.client.get('/execute_sql/') + with Settings(INTERNAL_IPS=['127.0.0.1'], DEBUG=True): + resp = self.client.get('/execute_sql/') self.assertEquals(resp.status_code, 200) def test_show_toolbar_DEBUG(self): @@ -145,6 +149,7 @@ class DebugToolbarTestCase(BaseTestCase): def test_with_process_view(self): request = self.request + response = self.response def _test_view(request): return HttpResponse('') @@ -153,15 +158,18 @@ class DebugToolbarTestCase(BaseTestCase): panel = self.toolbar.get_panel(RequestVarsDebugPanel) panel.process_request(request) panel.process_view(request, _test_view, [], {}) + panel.process_response(request, response) content = panel.content() self.assertTrue('tests.tests._test_view' in content, content) def test_without_process_view(self): request = self.request + response = self.response with Settings(DEBUG=True): panel = self.toolbar.get_panel(RequestVarsDebugPanel) panel.process_request(request) + panel.process_response(request, response) content = panel.content() self.assertTrue('<no view>' in content, content) |