mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-07-24 05:36:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Fixed #28817 -- Made QuerySet.iterator() use server-side cursors after values() and values_list().

Browse source
This commit is contained in:
Dražen Odobašić 2017-11-19 10:13:10 -05:00 • committed by Tim Graham
parent 6cb6382639
commit d97f026a7a
4 changed files with 27 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/backends/postgresql/test_server_side_cursors.py
View file

@ -3,7 +3,7 @@ import unittest
from collections import namedtuple
from contextlib import contextmanager
from django.db import connection
from django.db import connection, models
from django.test import TestCase
from ..models import Person
@ -53,6 +53,19 @@ class ServerSideCursorsPostgres(TestCase):
def test_server_side_cursor(self):
self.assertUsesCursor(Person.objects.iterator())
def test_values(self):
self.assertUsesCursor(Person.objects.values('first_name').iterator())
def test_values_list(self):
self.assertUsesCursor(Person.objects.values_list('first_name').iterator())
def test_values_list_flat(self):
self.assertUsesCursor(Person.objects.values_list('first_name', flat=True).iterator())
def test_values_list_fields_not_equal_to_names(self):
expr = models.Count('id')
self.assertUsesCursor(Person.objects.annotate(id__count=expr).values_list(expr, 'id__count').iterator())
def test_server_side_cursor_many_cursors(self):
persons = Person.objects.iterator()
persons2 = Person.objects.iterator()