Add (failing) tests for the profiling panel.

Refs #466.

1 files changed, 38 insertions, 1 deletions

diff --git a/tests/panels/test_profiling.py b/tests/panels/test_profiling.py
index fc76a39..4e01073 100644
--- a/tests/panels/test_profiling.py
+++ b/tests/panels/test_profiling.py
@@ -4,6 +4,44 @@ from django.contrib.auth.models import User
 from django.db import IntegrityError, transaction
 from django.test import TestCase
 from django.test.utils import override_settings
+from django.utils import unittest
+
+try:
+    import line_profiler
+except ImportError:
+    line_profiler = None
+
+from debug_toolbar.panels import profiling
+
+from ..base import BaseTestCase
+from ..views import regular_view
+
+
+@override_settings(DEBUG_TOOLBAR_PANELS=['debug_toolbar.panels.profiling.ProfilingDebugPanel'])
+class ProfilingPanelTestCase(BaseTestCase):
+
+    def setUp(self):
+        super(ProfilingPanelTestCase, self).setUp()
+        self.panel = self.toolbar.get_panel_by_id('ProfilingDebugPanel')
+
+    def _test_render_with_or_without_line_profiler(self):
+        self.panel.process_view(self.request, regular_view, ('profiling',), {})
+        self.panel.process_response(self.request, self.response)
+        self.assertIn('func_list', self.panel.get_stats())
+        self.assertIn('regular_view', self.panel.content())
+
+    @unittest.skipIf(line_profiler is None, "line_profiler isn't available")
+    def test_render_with_line_profiler(self):
+        self._test_render_with_or_without_line_profiler()
+
+    def test_without_line_profiler(self):
+        _use_line_profiler = profiling.DJ_PROFILE_USE_LINE_PROFILER
+        profiling.DJ_PROFILE_USE_LINE_PROFILER = False
+        try:
+            self._test_render_with_or_without_line_profiler()
+        finally:
+            profiling.DJ_PROFILE_USE_LINE_PROFILER = _use_line_profiler
+
 
 
 @override_settings(DEBUG=True,
@@ -11,7 +49,6 @@ from django.test.utils import override_settings
 class ProfilingPanelIntegrationTestCase(TestCase):
 
     def test_view_executed_once(self):
-
         self.assertEqual(User.objects.count(), 0)
 
         response = self.client.get('/new_user/')