Ticket #5416: assert_number_of_queries.diff

test/testcases.py

@@ -427 +427 @@
             (u"Template '%s' was used unexpectedly in rendering the"
              u" response") % template_name)
 
+    def assertNumberofQueries(self, num, callable, *args, **kwargs):
+        """
+        Asserts that exactly num queries were used when callable was invoked.
+        """
+        DEBUG = settings.DEBUG
+        settings.DEBUG = True
+
+        num_queries_before = len(connection.queries)
+        callable(*args, **kwargs)
+        num_queries_after = len(connection.queries)
+        actual = (num_queries_after - num_queries_before)
+
+        settings.DEBUG = DEBUG
+        self.failUnlessEqual(num, actual,
+                             (u"Callable did not execute %s queries as expected."
+                              u" Actual number of queries: %s.") % (num, actual))
+
 class TestCase(TransactionTestCase):
     """
     Does basically the same as TransactionTestCase, but surrounds every test