How come queries aren't being added to Django's db.connection.queries in tests?

Question:

I’m trying to capture the queries which my code submits to the database by examining the contents of django.db.connection.queries. For some reason though, after all the automatically produced setup queries are logged, no further queries are logged from my own code. The following test case demonstrates the behavior.

from django.test import TestCase
from django.db import reset_queries, connection
from django.contrib.auth.models import User
from django.conf import settings

class Test1(TestCase):
    def setUp(self):
        settings.DEBUG = True

    def test1(self):
        self.assert_(settings.DEBUG, 'DEBUG is False')
        reset_queries() #clears out all the setup queries
        User.objects.all()
        self.assert_(connection.queries, 'No queries')

And here are the results of running it:

Traceback (most recent call last):
  File "/Users/jacob/foo/bar_project/baz_application/tests.py", line 246, in test1
    self.assert_(connection.queries)
AssertionError: No queries

Would anyone be able to shed some light on this? Thanks.

Asked By: Jacob

||

Answers:

When you run tests DEBUG is set to False explicitly by Django test framework.

Answered By: Davor Lucic

You have to explicitly set DEBUG. For example, see the sample usage section for these tests in the django documentation:

# Set up.
# The test runner sets settings.DEBUG to False, but we want to gather queries
# so we'll set it to True here and reset it at the end of the test suite.
>>> from django.conf import settings
>>> settings.DEBUG = True

UPDATE: I may be missing something, but doing it in each test should definitely fix the issue. Take a look at the DjangoTestSuiteRunner — it seems that DEBUG is set False in the setup_test_environment, which is called in run_tests, which goes on to instantiate a DjangoTestRunner and call its run method. So you’ll need to undo that — based on a quick scan of the code, it might be sufficient to do this in your setup method.

Answered By: ars

You will not see any queries after executing User.objects.all(). This is only to be expected. The reason? Querysets are lazy. Unless you do something with the queryset NO query will be triggered. To verify this hypothesis, try the following and see if the test passes.

class Test1(TestCase):
    def setUp(self):
        settings.DEBUG = True

    def test1(self):
        self.assert_(settings.DEBUG, 'DEBUG is False')
        reset_queries() #clears out all the setup queries
        print User.objects.all() # <============= Printing the queryset.
        self.assert_(connection.queries, 'No queries')
Answered By: Manoj Govindan

In addition to the other answers:

Instead of overriding settings.DEBUG in your setUp(), you can modify DEBUG temporarily by using (Simple)TestCase.settings as a context manager, as follows:

class Test1(TestCase):
    def test1(self):
        with self.settings(DEBUG=True):
            # Evaluate queryset here
            ...
        print(connection.queryset)
Answered By: djvg
Categories: questions Tags: , , ,
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.