diff options
| author | Dražen Odobašić <dodobas@quaternion.co> | 2017-11-19 10:13:10 -0500 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2017-11-21 10:45:02 -0500 |
| commit | 7f4e17451135a8aee597e24aac4670a6d8860047 (patch) | |
| tree | 448522e58d6091715beb41c1ecf42f3a415e0080 /tests/backends/test_postgresql.py | |
| parent | 90c120c2b73534bd84166f5e67a01cddfeb4cede (diff) | |
[1.11.x] Fixed #28817 -- Made QuerySet.iterator() use server-side cursors after values() and values_list().
Backport of d97f026a7ab5212192426e45121f7a52751a2044 from master
Diffstat (limited to 'tests/backends/test_postgresql.py')
| -rw-r--r-- | tests/backends/test_postgresql.py | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/tests/backends/test_postgresql.py b/tests/backends/test_postgresql.py index 8ea8638306..2efec5eef2 100644 --- a/tests/backends/test_postgresql.py +++ b/tests/backends/test_postgresql.py @@ -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__count1=expr).values_list(expr, 'id__count1').iterator()) + def test_server_side_cursor_many_cursors(self): persons = Person.objects.iterator() persons2 = Person.objects.iterator() |
