/usr/lib/python2.7/dist-packages/tests/panels/test_sql.py is in python-django-debug-toolbar 1:1.0.1-0+nmu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | # coding: utf-8
from __future__ import absolute_import, unicode_literals
from django.contrib.auth.models import User
from django.db import connection
from django.db.utils import DatabaseError
from django.utils import unittest
from ..base import BaseTestCase
class SQLPanelTestCase(BaseTestCase):
def setUp(self):
super(SQLPanelTestCase, self).setUp()
self.panel = self.toolbar.get_panel_by_id('SQLPanel')
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)
list(User.objects.all())
# ensure query was logged
self.assertEqual(len(self.panel._queries), 1)
query = self.panel._queries[0]
self.assertEqual(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 populated
self.assertTrue(len(query[1]['stacktrace']) > 0)
def test_non_ascii_query(self):
self.assertEqual(len(self.panel._queries), 0)
# non-ASCII text query
list(User.objects.extra(where=["username = 'apéro'"]))
self.assertEqual(len(self.panel._queries), 1)
# non-ASCII text parameters
list(User.objects.filter(username='thé'))
self.assertEqual(len(self.panel._queries), 2)
# non-ASCII bytes parameters
list(User.objects.filter(username='café'.encode('utf-8')))
self.assertEqual(len(self.panel._queries), 3)
self.panel.process_response(self.request, self.response)
# ensure the panel renders correctly
self.assertIn('café', self.panel.content)
@unittest.skipUnless(connection.vendor == 'postgresql',
'Test valid only on PostgreSQL')
def test_erroneous_query(self):
"""
Test that an error in the query isn't swallowed by the middleware.
"""
try:
connection.cursor().execute("erroneous query")
except DatabaseError as e:
self.assertTrue('erroneous query' in str(e))
def test_disable_stacktraces(self):
self.assertEqual(len(self.panel._queries), 0)
with self.settings(DEBUG_TOOLBAR_CONFIG={'ENABLE_STACKTRACES': False}):
list(User.objects.all())
# ensure query was logged
self.assertEqual(len(self.panel._queries), 1)
query = self.panel._queries[0]
self.assertEqual(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.assertEqual([], query[1]['stacktrace'])
|