Disable instrumentation for disabled panels.

This eliminates the performance overhead when a panel is disabled.

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]