From 295f6f8f9a269cc547054f7094473b4388d58b16 Mon Sep 17 00:00:00 2001
From: Colin Howe
Date: Fri, 7 Oct 2011 08:09:31 +0100
Subject: Make stacktraces configurable

---
 tests/tests.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'tests')

diff --git a/tests/tests.py b/tests/tests.py
index 154615a..f0ed971 100644
--- a/tests/tests.py
+++ b/tests/tests.py
@@ -204,6 +204,28 @@ class SQLPanelTestCase(BaseTestCase):
         self.assertTrue('duration' in query[1])
         self.assertTrue('stacktrace' in query[1])
 
+        # ensure the stacktrace is populated
+        self.assertTrue(len(query[1]['stacktrace']) > 0)
+
+    def test_disable_stacktraces(self):
+        panel = self.toolbar.get_panel(SQLDebugPanel)
+        self.assertEquals(len(panel._queries), 0)
+       
+        with Settings(DEBUG_TOOLBAR_CONFIG={ 'ENABLE_STACKTRACES' : False }):
+            list(User.objects.all())
+        
+        # ensure query was logged
+        self.assertEquals(len(panel._queries), 1)
+        query = panel._queries[0]
+        self.assertEquals(query[0], 'default')
+        self.assertTrue('sql' in query[1])
+        self.assertTrue('duration' in query[1])
+        self.assertTrue('stacktrace' in query[1])
+
+        # ensure the stacktrace is empty
+        self.assertEquals([], query[1]['stacktrace'])
+
+
 class TemplatePanelTestCase(BaseTestCase):
     def test_queryset_hook(self):
         template_panel = self.toolbar.get_panel(TemplateDebugPanel)
-- 
cgit v1.2.3