summaryrefslogtreecommitdiff
path: root/tests/test_utils/test_transactiontestcase.py
diff options
context:
space:
mode:
authorreficul31 <reficul31@gmail.com>2017-02-28 06:30:09 +0530
committerTim Graham <timograham@gmail.com>2017-06-16 21:06:13 -0400
commit92e286498acd4f4562be11c21879856de8f47252 (patch)
tree305ed0a4eeb31d07966974fab641a5fe7cce6ed9 /tests/test_utils/test_transactiontestcase.py
parent6f44f714c92d2966dca390ebd3054e5fb0bb0c80 (diff)
Fixed #27539 -- Made TransactionTestCase._pre_setup() clear the queries_log so it's less likely to overflow.
TransactionTestCase.assertNumQueries() fails in an overflow situation.
Diffstat (limited to 'tests/test_utils/test_transactiontestcase.py')
-rw-r--r--tests/test_utils/test_transactiontestcase.py18
1 files changed, 17 insertions, 1 deletions
diff --git a/tests/test_utils/test_transactiontestcase.py b/tests/test_utils/test_transactiontestcase.py
index 593c06f7e3..40c9b7576f 100644
--- a/tests/test_utils/test_transactiontestcase.py
+++ b/tests/test_utils/test_transactiontestcase.py
@@ -1,6 +1,7 @@
from unittest import mock
-from django.test import TransactionTestCase
+from django.db import connections
+from django.test import TestCase, TransactionTestCase, override_settings
class TestSerializedRollbackInhibitsPostMigrate(TransactionTestCase):
@@ -28,3 +29,18 @@ class TestSerializedRollbackInhibitsPostMigrate(TransactionTestCase):
reset_sequences=False, inhibit_post_migrate=True,
database='default', verbosity=0,
)
+
+
+@override_settings(DEBUG=True) # Enable query logging for test_queries_cleared
+class TransactionTestCaseMultiDbTests(TestCase):
+ available_apps = []
+ multi_db = True
+
+ def test_queries_cleared(self):
+ """
+ TransactionTestCase._pre_setup() clears the connections' queries_log
+ so that it's less likely to overflow. An overflow causes
+ assertNumQueries() to fail.
+ """
+ for alias in connections:
+ self.assertEqual(len(connections[alias].queries_log), 0, 'Failed for alias %s' % alias)