mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-07-18 18:55:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fixed #29851 -- Fixed crash of annotations with window expressions in Subquery.

Browse source
This commit is contained in:
Mariusz Felisiak 2018-12-27 20:21:57 +01:00 • committed by GitHub
parent 6fe9c45b72
commit dd8ed64113
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 39 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

31
tests/expressions_window/tests.py
View file

@ -4,7 +4,7 @@ from unittest import skipIf, skipUnless
from django.core.exceptions import FieldError
from django.db import NotSupportedError, connection
from django.db.models import (
F, RowRange, Value, ValueRange, Window, WindowFrame,
F, OuterRef, RowRange, Subquery, Value, ValueRange, Window, WindowFrame,
)
from django.db.models.aggregates import Avg, Max, Min, Sum
from django.db.models.functions import (
@ -584,6 +584,35 @@ class WindowFunctionTests(TestCase):
('Brown', 'Sales', 53000, datetime.date(2009, 9, 1), 148000)
], transform=lambda row: (row.name, row.department, row.salary, row.hire_date, row.sum))
def test_subquery_row_range_rank(self):
qs = Employee.objects.annotate(
highest_avg_salary_date=Subquery(
Employee.objects.filter(
department=OuterRef('department'),
).annotate(
avg_salary=Window(
expression=Avg('salary'),
order_by=[F('hire_date').asc()],
frame=RowRange(start=-1, end=1),
),
).order_by('-avg_salary', '-hire_date').values('hire_date')[:1],
),
).order_by('department', 'name')
self.assertQuerysetEqual(qs, [
('Adams', 'Accounting', datetime.date(2005, 11, 1)),
('Jenson', 'Accounting', datetime.date(2005, 11, 1)),
('Jones', 'Accounting', datetime.date(2005, 11, 1)),
('Williams', 'Accounting', datetime.date(2005, 11, 1)),
('Moore', 'IT', datetime.date(2013, 8, 1)),
('Wilkinson', 'IT', datetime.date(2013, 8, 1)),
('Johnson', 'Management', datetime.date(2005, 7, 1)),
('Miller', 'Management', datetime.date(2005, 7, 1)),
('Johnson', 'Marketing', datetime.date(2012, 3, 1)),
('Smith', 'Marketing', datetime.date(2012, 3, 1)),
('Brown', 'Sales', datetime.date(2009, 9, 1)),
('Smith', 'Sales', datetime.date(2009, 9, 1)),
], transform=lambda row: (row.name, row.department, row.highest_avg_salary_date))
def test_row_range_rank(self):
"""
A query with ROWS BETWEEN UNBOUNDED PRECEDING AND 3 FOLLOWING.