diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/panels/test_sql.py | 5 | ||||
| -rw-r--r-- | tests/panels/test_template.py | 9 |
2 files changed, 12 insertions, 2 deletions
diff --git a/tests/panels/test_sql.py b/tests/panels/test_sql.py index a1de0ea..56b70dc 100644 --- a/tests/panels/test_sql.py +++ b/tests/panels/test_sql.py @@ -17,6 +17,11 @@ class SQLPanelTestCase(BaseTestCase): def setUp(self): super(SQLPanelTestCase, self).setUp() self.panel = self.toolbar.get_panel(SQLDebugPanel) + self.panel.enable_instrumentation() + + def tearDown(self): + self.panel.disable_instrumentation() + super(SQLPanelTestCase, self).tearDown() def test_recording(self): self.assertEqual(len(self.panel._queries), 0) diff --git a/tests/panels/test_template.py b/tests/panels/test_template.py index c98521e..545ad9b 100644 --- a/tests/panels/test_template.py +++ b/tests/panels/test_template.py @@ -18,6 +18,12 @@ class TemplateDebugPanelTestCase(BaseTestCase): def setUp(self): super(TemplateDebugPanelTestCase, self).setUp() self.panel = self.toolbar.get_panel(TemplateDebugPanel) + self.sql_panel = self.toolbar.get_panel(SQLDebugPanel) + self.sql_panel.enable_instrumentation() + + def tearDown(self): + self.sql_panel.disable_instrumentation() + super(TemplateDebugPanelTestCase, self).tearDown() def test_queryset_hook(self): t = Template("No context variables here!") @@ -30,8 +36,7 @@ class TemplateDebugPanelTestCase(BaseTestCase): t.render(c) # ensure the query was NOT logged - sql_panel = self.toolbar.get_panel(SQLDebugPanel) - self.assertEqual(len(sql_panel._queries), 0) + self.assertEqual(len(self.sql_panel._queries), 0) base_ctx_idx = 1 if django.VERSION[:2] >= (1, 5) else 0 ctx = self.panel.templates[0]['context'][base_ctx_idx] |