mirror of
https://github.com/django/django.git
synced 2025-10-24 06:06:09 +00:00
Fixed #24485 -- Allowed combined expressions to set output_field
This commit is contained in:
@@ -5,12 +5,15 @@ from decimal import Decimal
|
||||
|
||||
from django.core.exceptions import FieldDoesNotExist, FieldError
|
||||
from django.db.models import (
|
||||
F, BooleanField, CharField, Count, Func, IntegerField, Sum, Value,
|
||||
F, BooleanField, CharField, Count, DateTimeField, ExpressionWrapper, Func,
|
||||
IntegerField, Sum, Value,
|
||||
)
|
||||
from django.test import TestCase
|
||||
from django.utils import six
|
||||
|
||||
from .models import Author, Book, Company, DepartmentStore, Employee, Store
|
||||
from .models import (
|
||||
Author, Book, Company, DepartmentStore, Employee, Publisher, Store, Ticket,
|
||||
)
|
||||
|
||||
|
||||
def cxOracle_513_py3_bug(func):
|
||||
@@ -52,6 +55,24 @@ class NonAggregateAnnotationTestCase(TestCase):
|
||||
for book in books:
|
||||
self.assertEqual(book.num_awards, book.publisher.num_awards)
|
||||
|
||||
def test_mixed_type_annotation_date_interval(self):
|
||||
active = datetime.datetime(2015, 3, 20, 14, 0, 0)
|
||||
duration = datetime.timedelta(hours=1)
|
||||
expires = datetime.datetime(2015, 3, 20, 14, 0, 0) + duration
|
||||
Ticket.objects.create(active_at=active, duration=duration)
|
||||
t = Ticket.objects.annotate(
|
||||
expires=ExpressionWrapper(F('active_at') + F('duration'), output_field=DateTimeField())
|
||||
).first()
|
||||
self.assertEqual(t.expires, expires)
|
||||
|
||||
def test_mixed_type_annotation_numbers(self):
|
||||
test = self.b1
|
||||
b = Book.objects.annotate(
|
||||
combined=ExpressionWrapper(F('pages') + F('rating'), output_field=IntegerField())
|
||||
).get(isbn=test.isbn)
|
||||
combined = int(test.pages + test.rating)
|
||||
self.assertEqual(b.combined, combined)
|
||||
|
||||
def test_annotate_with_aggregation(self):
|
||||
books = Book.objects.annotate(
|
||||
is_book=Value(1, output_field=IntegerField()),
|
||||
|
||||
Reference in New Issue
Block a user