mirror of
https://github.com/django/django.git
synced 2024-12-22 09:05:43 +00:00
Fixed #35011, Refs #28900 -- Added tests for QuerySet.union() with multiple models and DateTimeField annotations.
Ticket was resolved by 65ad4ade74
as part of #28900.
This commit is contained in:
parent
042b381e2e
commit
40bfd7b09a
@ -1,7 +1,9 @@
|
|||||||
import operator
|
import operator
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
from django.db import DatabaseError, NotSupportedError, connection
|
from django.db import DatabaseError, NotSupportedError, connection
|
||||||
from django.db.models import (
|
from django.db.models import (
|
||||||
|
DateTimeField,
|
||||||
Exists,
|
Exists,
|
||||||
F,
|
F,
|
||||||
IntegerField,
|
IntegerField,
|
||||||
@ -10,12 +12,13 @@ from django.db.models import (
|
|||||||
Transform,
|
Transform,
|
||||||
Value,
|
Value,
|
||||||
)
|
)
|
||||||
from django.db.models.functions import Mod
|
from django.db.models.functions import Cast, Mod
|
||||||
from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature
|
from django.test import TestCase, skipIfDBFeature, skipUnlessDBFeature
|
||||||
from django.test.utils import CaptureQueriesContext
|
from django.test.utils import CaptureQueriesContext
|
||||||
|
|
||||||
from .models import (
|
from .models import (
|
||||||
Annotation,
|
Annotation,
|
||||||
|
Article,
|
||||||
Author,
|
Author,
|
||||||
Celebrity,
|
Celebrity,
|
||||||
ExtraInfo,
|
ExtraInfo,
|
||||||
@ -440,6 +443,39 @@ class QuerySetSetOperationTests(TestCase):
|
|||||||
[("c1", -10, "cb"), ("rn1", 10, "rn")],
|
[("c1", -10, "cb"), ("rn1", 10, "rn")],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_union_multiple_models_with_values_list_and_datetime_annotations(self):
|
||||||
|
gen_x = datetime(1966, 6, 6)
|
||||||
|
Article.objects.create(name="Bellatrix", created=gen_x)
|
||||||
|
column_names = ["name", "created", "order"]
|
||||||
|
qs1 = Article.objects.annotate(order=Value(1)).values_list(*column_names)
|
||||||
|
|
||||||
|
gen_y = datetime(1991, 10, 10)
|
||||||
|
ReservedName.objects.create(name="Rigel", order=2)
|
||||||
|
qs2 = ReservedName.objects.annotate(
|
||||||
|
created=Cast(Value(gen_y), DateTimeField())
|
||||||
|
).values_list(*column_names)
|
||||||
|
|
||||||
|
expected_result = [("Bellatrix", gen_x, 1), ("Rigel", gen_y, 2)]
|
||||||
|
self.assertEqual(list(qs1.union(qs2).order_by("order")), expected_result)
|
||||||
|
|
||||||
|
def test_union_multiple_models_with_values_and_datetime_annotations(self):
|
||||||
|
gen_x = datetime(1966, 6, 6)
|
||||||
|
Article.objects.create(name="Bellatrix", created=gen_x)
|
||||||
|
column_names = ["name", "created", "order"]
|
||||||
|
qs1 = Article.objects.values(*column_names, order=Value(1))
|
||||||
|
|
||||||
|
gen_y = datetime(1991, 10, 10)
|
||||||
|
ReservedName.objects.create(name="Rigel", order=2)
|
||||||
|
qs2 = ReservedName.objects.values(
|
||||||
|
*column_names, created=Cast(Value(gen_y), DateTimeField())
|
||||||
|
)
|
||||||
|
|
||||||
|
expected_result = [
|
||||||
|
{"name": "Bellatrix", "created": gen_x, "order": 1},
|
||||||
|
{"name": "Rigel", "created": gen_y, "order": 2},
|
||||||
|
]
|
||||||
|
self.assertEqual(list(qs1.union(qs2).order_by("order")), expected_result)
|
||||||
|
|
||||||
def test_union_in_subquery(self):
|
def test_union_in_subquery(self):
|
||||||
ReservedName.objects.bulk_create(
|
ReservedName.objects.bulk_create(
|
||||||
[
|
[
|
||||||
|
Loading…
Reference in New Issue
Block a user