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
- Generate and render HTML Table by LLyaudet 5 days, 9 hours ago
- My firs Snippets by GutemaG 1 week, 1 day ago
- FileField having auto upload_to path by junaidmgithub 1 month, 2 weeks ago
- LazyPrimaryKeyRelatedField by LLyaudet 1 month, 3 weeks ago
- CacheInDictManager by LLyaudet 1 month, 3 weeks ago
Comments
Please login first before commenting.