p31829507
/
django
mirror of https://github.com/django/django.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed #31651 -- Made ExpressionWrapper use grouping columns from wrapped expression.

This commit is contained in:
Thodoris Sotiropoulos 2020-06-03 00:14:01 +03:00 committed by Mariusz Felisiak
parent 22a59c01c0
commit df32fd42b8
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
django/db/models/expressions.py
View File

@ -863,6 +863,9 @@ class ExpressionWrapper(Expression):
def get_source_expressions(self):
return [self.expression]
def get_group_by_cols(self, alias=None):
return self.expression.get_group_by_cols(alias=alias)
def as_sql(self, compiler, connection):
return self.expression.as_sql(compiler, connection)

10
tests/expressions/tests.py
View File

@ -1828,3 +1828,13 @@ class CombinableTests(SimpleTestCase):
def test_reversed_or(self):
with self.assertRaisesMessage(NotImplementedError, self.bitwise_msg):
object() | Combinable()
class ExpressionWrapperTests(SimpleTestCase):
def test_empty_group_by(self):
expr = ExpressionWrapper(Value(3), output_field=IntegerField())
self.assertEqual(expr.get_group_by_cols(alias=None), [])
def test_non_empty_group_by(self):
expr = ExpressionWrapper(Lower(Value('f')), output_field=IntegerField())
self.assertEqual(expr.get_group_by_cols(alias=None), [expr.expression])