Add django_debug_sql ini option

TODO:

- [ ] doc
- [ ] use a command line option instead?

I've thought about having a fixture also, but it would require to set
"force_debug_cursor" on the connections then, and it probably not
useful; typically you want to use this for a short time only - therefore
a command line option might be better suited also (but you can use `-o
django_debug_sql = 1`).  And on the other hand, it will only show up on
failures, and is therefore maybe good to set it in general.

Author: blueyed

pytest_django/fixtures.py

@@ -2,6 +2,7 @@
 
 from __future__ import with_statement
 
+import logging
 import os
 import warnings
 from contextlib import contextmanager
@@ -80,6 +81,44 @@ def django_db_createdb(request):
     return request.config.getvalue("create_db")
 
 
+def _setup_logging():
+    logger = logging.getLogger('django.db.backends')
+
+    oldlevel = logger.level
+    logger.setLevel(logging.DEBUG)
+
+    logger.setLevel(oldlevel)
+
+
+def _setup_sql_debug_logging():
+    # Simulate what Django's DebugSQLTextTestResult does.
+    logger = logging.getLogger('django.db.backends')
+    oldlevel = logger.level
+    logger.setLevel(logging.DEBUG)
+    return oldlevel
+
+
+def _restore_sql_debug_logging(request, oldlevel):
+    logger = logging.getLogger('django.db.backends')
+    if logger.level != logging.DEBUG:
+        request.node.warn(pytest.PytestWarning(
+            "Debug logging level of django.db.backends was changed (to {}). "
+            "SQL queries might be missing.  "
+            "This might be caused by calling django.setup() too late/unnecessarily.".format(
+                logger.level
+            )
+        ))
+
+    logger.setLevel(oldlevel)
+
+
+@pytest.fixture
+def django_debug_sql(request):
+    oldlevel = _setup_sql_debug_logging()
+    yield
+    _restore_sql_debug_logging(request, oldlevel)
+
+
 @pytest.fixture(scope="session")
 def django_db_setup(
     request,
@@ -95,6 +134,11 @@ def django_db_setup(
 
     setup_databases_args = {}
 
+    debug_sql = request.config.getini("django_debug_sql")
+    if debug_sql:
+        setup_databases_args["debug_sql"] = True
+        old_loglevel = _setup_sql_debug_logging()
+
     if not django_db_use_migrations:
         _disable_native_migrations()
 
@@ -108,7 +152,9 @@ def django_db_setup(
             **setup_databases_args
         )
 
-    def teardown_database():
+    yield
+
+    if not django_db_keepdb:
         with django_db_blocker.unblock():
             try:
                 teardown_databases(db_cfg, verbosity=request.config.option.verbose)
@@ -119,8 +165,8 @@ def teardown_database():
                     )
                 )
 
-    if not django_db_keepdb:
-        request.addfinalizer(teardown_database)
+    if debug_sql:
+        _restore_sql_debug_logging(request, old_loglevel)
 
 
 def _django_db_fixture_helper(

pytest_django/plugin.py

@@ -124,6 +124,12 @@ def pytest_addoption(parser):
         type="bool",
         default=True,
     )
+    parser.addini(
+        "django_debug_sql",
+        "Enable logging of SQL statements, displayed with failed tests.",
+        type="bool",
+        default=False,
+    )
     group._addoption(
         "--fail-on-template-vars",
         action="store_true",