Выберите связанные запросы на выполнение - PullRequest
Новая
       28

Выберите связанные запросы на выполнение

0 голосов
/ 04 февраля 2019

Я не могу заставить select_related работать со следующей конфигурацией:

Модели: 

class Text(models.Model):
    author = models.ForeignKey('authors.Author', 
                               on_delete=models.SET_NULL,
                               blank=True,
                               null=True)

Авторская модель: 

class Author(models.Model):
    name = models.CharField(max_length=200)
    def get_absolute_url(self):
        kwargs = {'pk': self.pk}
        return reverse('author-detail', kwargs=kwargs)

Просмотр

В представлении я использую функцию select_related, чтобы избежать попадания в БД при запросе авторов текста, например: mytext.author: 

class TextsViewTest(TestCase):
    def text_view(request,
                 pk,                            
                 template_name='texts/detail.html'):

        source_text = Text.objects.select_related('author').get(pk=pk)
        return render(request, template_name,
                 {
                     'source': source_text,
                 })

Тест

Согласно select_related он не должен попадать в базу данных при доступе к отношению Text.author, но при тестировании с: 

def test_layout_content_header__uses_prefetched_relationships(self):
    author = Author.objects.create(name="foobar")
    source_text = Text.objects.create(author=author)
    context = {'source': source_text}
    with self.assertNumQueries(0):
        from django.template.loader import render_to_string
        rendered = render_to_string("text/_content_header.html", context)

template

text/content_header.html: 

{% if source.author %} by <em><a href="{{source.author.get_absolute_url}}">{{source.author.name}}</a></em>{% endif %}

Вывод

./manage test texts.test_views показывает хит: 

Creating test database for alias 'default'...
System check identified no issues (0 silenced).
F
======================================================================
FAIL: test_layout_content_header__uses_prefetched_relationships (author.tests.test_views.TextsViewTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/.../text/tests/test_views.py", line 1035, in test_layout_content_header__uses_prefetched_relationships
    source_text.author
  File "/.../lib/python3.6/site-packages/django/test/testcases.py", line 80, in __exit__
    '%d. %s' % (i, query['sql']) for i, query in enumerate(self.captured_queries, start=1)
AssertionError: 1 != 0 : 1 queries executed, 0 expected
Captured queries were:
1. SELECT "authors_author"."id", "authors_author"."name", FROM "authors_author" WHERE "authors_author"."id" = 1

----------------------------------------------------------------------
Ran 1 test in 0.489s

FAILED (failures=1)
Destroying test database for alias 'default'...

Есть идеи?

1 Ответ

0 голосов
/ 04 февраля 2019

Похоже, вы не используете код своего представления внутри теста.Попробуйте скопировать тот же запрос в ваш тест, например: 

context = {'source': Text.objects.select_related('author').get(pk=source_text.pk)}
with self.assertNumQueries(0):
    from django.template.loader import render_to_string
    rendered = render_to_string("text/_content_header.html", context)

Или повторно использовать код представления (кажется, он объявлен в тестовом примере, верно?) 

with self.assertNumQueries(1):
    self.text_view(MagicMock(), source_text.pk)

Хотявам может потребоваться указать более продвинутый макет запроса, например, используя RequestFactory

...