Skip to content

Instantly share code, notes, and snippets.

@codysoyland

codysoyland/assert_num_queries.py

Created March 3, 2010 17:52
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save codysoyland/320810 to your computer and use it in GitHub Desktop.
Save codysoyland/320810 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
assert_num_queries.py
class BaseTestCase(TestCase):
@contextmanager
def assert_num_queries(self, n):
current_debug = settings.DEBUG
settings.DEBUG = True
current = len(connection.queries)
yield
self.assertEqual(current+n, len(connection.queries), connection.queries[current:])
settings.DEBUG = current_debug
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment