Fixed #28908 -- Allowed ArrayField lookups on ArrayAgg annotations.

1 files changed, 12 insertions, 0 deletions

diff --git a/tests/postgres_tests/test_aggregates.py b/tests/postgres_tests/test_aggregates.py
index 056d08441b..d4a01ff027 100644
--- a/tests/postgres_tests/test_aggregates.py
+++ b/tests/postgres_tests/test_aggregates.py
@@ -46,6 +46,18 @@ class TestGeneralAggregate(PostgreSQLTestCase):
         values = AggregateTestModel.objects.aggregate(arrayagg=ArrayAgg('boolean_field'))
         self.assertEqual(values, {'arrayagg': []})
 
+    def test_array_agg_lookups(self):
+        aggr1 = AggregateTestModel.objects.create()
+        aggr2 = AggregateTestModel.objects.create()
+        StatTestModel.objects.create(related_field=aggr1, int1=1, int2=0)
+        StatTestModel.objects.create(related_field=aggr1, int1=2, int2=0)
+        StatTestModel.objects.create(related_field=aggr2, int1=3, int2=0)
+        StatTestModel.objects.create(related_field=aggr2, int1=4, int2=0)
+        qs = StatTestModel.objects.values('related_field').annotate(
+            array=ArrayAgg('int1')
+        ).filter(array__overlap=[2]).values_list('array', flat=True)
+        self.assertCountEqual(qs.get(), [1, 2])
+
     def test_bit_and_general(self):
         values = AggregateTestModel.objects.filter(
             integer_field__in=[0, 1]).aggregate(bitand=BitAnd('integer_field'))