diff options
Diffstat (limited to 'tests/aggregation/tests.py')
-rw-r--r-- | tests/aggregation/tests.py | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/tests/aggregation/tests.py b/tests/aggregation/tests.py index 0f799c4bc3..5ba2e180e0 100644 --- a/tests/aggregation/tests.py +++ b/tests/aggregation/tests.py @@ -1170,3 +1170,23 @@ class AggregateTestCase(TestCase): Exists(long_books_qs), ).annotate(total=Count('*')) self.assertEqual(dict(has_long_books_breakdown), {True: 2, False: 3}) + + def test_aggregation_subquery_annotation_related_field(self): + publisher = Publisher.objects.create(name=self.a9.name, num_awards=2) + book = Book.objects.create( + isbn='159059999', name='Test book.', pages=819, rating=2.5, + price=Decimal('14.44'), contact=self.a9, publisher=publisher, + pubdate=datetime.date(2019, 12, 6), + ) + book.authors.add(self.a5, self.a6, self.a7) + books_qs = Book.objects.annotate( + contact_publisher=Subquery( + Publisher.objects.filter( + pk=OuterRef('publisher'), + name=OuterRef('contact__name'), + ).values('name')[:1], + ) + ).filter( + contact_publisher__isnull=False, + ).annotate(count=Count('authors')) + self.assertSequenceEqual(books_qs, [book]) |