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
|
More like this
- Template tag - list punctuation for a list of items by shapiromatron 10 months, 4 weeks ago
- JSONRequestMiddleware adds a .json() method to your HttpRequests by cdcarter 11 months ago
- Serializer factory with Django Rest Framework by julio 1 year, 5 months ago
- Image compression before saving the new model / work with JPG, PNG by Schleidens 1 year, 6 months ago
- Help text hyperlinks by sa2812 1 year, 7 months ago
Comments
Please login first before commenting.