Merge pull request #206 from lincolnloop/stats_decouple

Decouple statistics from presentation

1 files changed, 9 insertions, 1 deletions

diff --git a/tests/tests.py b/tests/tests.py
index 60620bb..b03e167 100644
--- a/tests/tests.py
+++ b/tests/tests.py
@@ -38,18 +38,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):
@@ -143,6 +147,7 @@ class DebugToolbarTestCase(BaseTestCase):
 
     def test_with_process_view(self):
         request = self.request
+        response = self.response
         
         def _test_view(request):
             return HttpResponse('')
@@ -151,15 +156,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)