summaryrefslogtreecommitdiff
path: root/tests/aggregation_regress
diff options
context:
space:
mode:
authorAnssi Kääriäinen <akaariai@gmail.com>2015-05-05 14:44:33 +0300
committerTim Graham <timograham@gmail.com>2015-05-11 11:52:14 -0400
commitd5ce2dd7bc5b797d2e338c4bc6e6f3e339b748e5 (patch)
tree584dfffad7ce06e14634a1ed9c3b6613fa222527 /tests/aggregation_regress
parent056a91dbfae9e741cd4bf033d3062d3d35ff4d90 (diff)
[1.8.x] Fixed #24748 -- Fixed incorrect GROUP BY on MySQL in some queries
When the query's model had a self-referential foreign key, the compiler.get_group_by() code incorrectly used the self-referential foreign key's column (for example parent_id) as GROUP BY clause when it should have used the model's primary key column (id). Backport of adc57632bc26cc8fe42bdb6aff463f883214980a from master
Diffstat (limited to 'tests/aggregation_regress')
-rw-r--r--tests/aggregation_regress/models.py5
-rw-r--r--tests/aggregation_regress/tests.py35
2 files changed, 15 insertions, 25 deletions
diff --git a/tests/aggregation_regress/models.py b/tests/aggregation_regress/models.py
index 2cad0a5486..4d675952c2 100644
--- a/tests/aggregation_regress/models.py
+++ b/tests/aggregation_regress/models.py
@@ -104,3 +104,8 @@ class Bravo(models.Model):
class Charlie(models.Model):
alfa = models.ForeignKey(Alfa, null=True)
bravo = models.ForeignKey(Bravo, null=True)
+
+
+class SelfRefFK(models.Model):
+ name = models.CharField(max_length=50)
+ parent = models.ForeignKey('self', null=True, blank=True, related_name='children')
diff --git a/tests/aggregation_regress/tests.py b/tests/aggregation_regress/tests.py
index bc6a55875d..12bb1ae3f4 100644
--- a/tests/aggregation_regress/tests.py
+++ b/tests/aggregation_regress/tests.py
@@ -16,7 +16,7 @@ from django.utils import six
from .models import (
Alfa, Author, Book, Bravo, Charlie, Clues, Entries, HardbackBook, ItemTag,
- Publisher, WithManualPK,
+ Publisher, SelfRefFK, WithManualPK,
)
@@ -1174,28 +1174,13 @@ class JoinPromotionTests(TestCase):
self.assertIn(' INNER JOIN ', str(qs.query))
-class AggregationOnRelationTest(TestCase):
- def setUp(self):
- self.a = Author.objects.create(name='Anssi', age=33)
- self.p = Publisher.objects.create(name='Manning', num_awards=3)
- Book.objects.create(isbn='asdf', name='Foo', pages=10, rating=0.1, price="0.0",
- contact=self.a, publisher=self.p, pubdate=datetime.date.today())
-
- def test_annotate_on_relation(self):
- qs = Book.objects.annotate(avg_price=Avg('price'), publisher_name=F('publisher__name'))
- self.assertEqual(qs[0].avg_price, 0.0)
- self.assertEqual(qs[0].publisher_name, "Manning")
-
- def test_aggregate_on_relation(self):
- # A query with an existing annotation aggregation on a relation should
- # succeed.
- qs = Book.objects.annotate(avg_price=Avg('price')).aggregate(
- publisher_awards=Sum('publisher__num_awards')
- )
- self.assertEqual(qs['publisher_awards'], 3)
- Book.objects.create(isbn='asdf', name='Foo', pages=10, rating=0.1, price="0.0",
- contact=self.a, publisher=self.p, pubdate=datetime.date.today())
- qs = Book.objects.annotate(avg_price=Avg('price')).aggregate(
- publisher_awards=Sum('publisher__num_awards')
+class SelfReferentialFKTests(TestCase):
+ def test_ticket_24748(self):
+ t1 = SelfRefFK.objects.create(name='t1')
+ SelfRefFK.objects.create(name='t2', parent=t1)
+ SelfRefFK.objects.create(name='t3', parent=t1)
+ self.assertQuerysetEqual(
+ SelfRefFK.objects.annotate(num_children=Count('children')).order_by('name'),
+ [('t1', 2), ('t2', 0), ('t3', 0)],
+ lambda x: (x.name, x.num_children)
)
- self.assertEqual(qs['publisher_awards'], 6)