Ticket #12822: ticket_12822_test.diff

new file tests/regressiontests/aggregation_regress/tests.py

@@ +1 @@
+from decimal import Decimal
+from datetime import date
+
+from django.test import TestCase
+from django.db.models import Max
+
+from regressiontests.aggregation_regress.models import *
+
+
+class AggregationTests(TestCase):
+
+    def test_aggregates_in_where_clause(self):
+        """
+        Regression test for #12822: DatabaseError: aggregates not allowed in
+        WHERE clause
+
+        Tests that the subselect works and returns results equivalent to a
+        query with the IDs listed.
+        """
+        qs = Book.objects.values('contact').annotate(Max('id'))
+        qs = qs.order_by('contact').values_list('id__max', flat=True)
+        # don't do anything with the queryset (qs) before including it as a
+        # subquery
+        books = Book.objects.order_by('id')
+        qs1 = books.filter(id__in=qs)
+        qs2 = books.filter(id__in=list(qs))
+        self.assertEqual(list(qs1), list(qs2))
+
+    def test_aggregates_in_where_clause_pre_eval(self):
+        """
+        Regression test for #12822: DatabaseError: aggregates not allowed in
+        WHERE clause
+
+        Same as the above test, but evaluates the queryset for the subquery
+        before it's used as a subquery.
+        """
+        qs = Book.objects.values('contact').annotate(Max('id'))
+        qs = qs.order_by('contact').values_list('id__max', flat=True)
+        # force the queryset (qs) for the subquery to be evaluated in its
+        # current state
+        list(qs)
+        books = Book.objects.order_by('id')
+        qs1 = books.filter(id__in=qs)
+        qs2 = books.filter(id__in=list(qs))
+        self.assertEqual(list(qs1), list(qs2))