Display the number of queries executed by the fixtures

when --querycount is used with --setup-show

Author: renanivo

pytest_django/plugin.py

@@ -323,6 +323,32 @@ def pytest_runtest_setup(item):
         _disable_class_methods(cls)
 
 
+@pytest.hookimpl(hookwrapper=True)
+def pytest_fixture_setup(fixturedef, request):
+    config = request.config
+
+    if config.option.querycount is None or not config.option.setupshow:
+        yield
+        return
+
+    from django.test.utils import CaptureQueriesContext
+    from django.db import connection
+
+    with CaptureQueriesContext(connection) as context:
+        yield
+
+    querycount = len(context.captured_queries)
+
+    if querycount:
+        capman = config.pluginmanager.getplugin('capturemanager')
+        capman.suspendcapture()
+
+        tw = config.get_terminal_writer()
+        tw.write(' (# queries executed: {})'.format(querycount))
+
+        capman.resumecapture()
+
+
 @pytest.hookimpl(hookwrapper=True)
 def pytest_runtest_call(item):
     count_parameter = item.config.option.querycount

tests/test_report.py

@@ -88,3 +88,25 @@ def test_two_queries():
         )
         assert 'test_two_queries' in lines[0]
         assert 'test_one_query' in lines[1]
+
+    def test_should_report_fixture_queries(self, django_testdir):
+        django_testdir.create_test_module('''
+            import pytest
+            from django.db import connection
+
+            @pytest.fixture
+            def one_query():
+                with connection.cursor() as cursor:
+                    cursor.execute('SELECT 1')
+
+            @pytest.mark.django_db
+            def test_without_queries(one_query):
+                pass
+        ''')
+
+        result = django_testdir.runpytest_subprocess(
+            '--setup-show',
+            '--querycount=5'
+        )
+
+        assert '(# queries executed: 1)' in result.stdout.str()