- October 4, 2010
- 1 (after 1 ratings)
Sometimes you may want to check how many queries a specific set of operations is taking. This TestCase decorator allows you to do that.
1 2 3 4 5 6 7 8 9 10 11 12
def check_query_count(num_queries): def decorator(func): @wraps(func) def inner(self, *args, **kwargs): initial_queries = len(connection.queries) ret = func(self, *args, **kwargs) final_queries = len(connection.queries) if settings.DEBUG: self.assertEqual(final_queries - initial_queries, num_queries) return ret return inner return decorator