diff options
| author | Iuri de Silvio <iurisilvio@gmail.com> | 2021-04-08 16:31:45 -0300 |
|---|---|---|
| committer | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2021-04-12 21:11:40 +0200 |
| commit | 9760e262f85ae57df39abe2799eff48a82b14474 (patch) | |
| tree | f460abf7708b37a6bfc91357dea702138beeea26 /tests | |
| parent | 823a9e6bac38d38f7b0347497b833eec732bd384 (diff) | |
Fixed #32627 -- Fixed QuerySet.values()/values_list() crash on combined querysets ordered by unannotated columns.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/queries/test_qs_combinators.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/tests/queries/test_qs_combinators.py b/tests/queries/test_qs_combinators.py index 3b8204645d..351a401902 100644 --- a/tests/queries/test_qs_combinators.py +++ b/tests/queries/test_qs_combinators.py @@ -5,7 +5,7 @@ from django.db.models import Exists, F, IntegerField, OuterRef, Value from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature from django.test.utils import CaptureQueriesContext -from .models import Number, ReservedName +from .models import Celebrity, Number, ReservedName @skipUnlessDBFeature('supports_select_union') @@ -234,6 +234,24 @@ class QuerySetSetOperationTests(TestCase): operator.itemgetter('num'), ) + def test_union_multiple_models_with_values_list_and_order(self): + reserved_name = ReservedName.objects.create(name='rn1', order=0) + qs1 = Celebrity.objects.all() + qs2 = ReservedName.objects.all() + self.assertSequenceEqual( + qs1.union(qs2).order_by('name').values_list('pk', flat=True), + [reserved_name.pk], + ) + + def test_union_multiple_models_with_values_list_and_order_by_extra_select(self): + reserved_name = ReservedName.objects.create(name='rn1', order=0) + qs1 = Celebrity.objects.extra(select={'extra_name': 'name'}) + qs2 = ReservedName.objects.extra(select={'extra_name': 'name'}) + self.assertSequenceEqual( + qs1.union(qs2).order_by('extra_name').values_list('pk', flat=True), + [reserved_name.pk], + ) + def test_count_union(self): qs1 = Number.objects.filter(num__lte=1).values('num') qs2 = Number.objects.filter(num__gte=2, num__lte=3).values('num') |
