1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
|
from django.test import TestCase
from polymorphic.tests.models import Bottom, Middle, Top
class RegressionTests(TestCase):
def test_for_query_result_incomplete_with_inheritance(self):
"""https://github.com/bconstantin/django_polymorphic/issues/15"""
top = Top()
top.save()
middle = Middle()
middle.save()
bottom = Bottom()
bottom.save()
expected_queryset = [top, middle, bottom]
self.assertQuerySetEqual(
Top.objects.order_by("pk"),
[repr(r) for r in expected_queryset],
transform=repr,
)
expected_queryset = [middle, bottom]
self.assertQuerySetEqual(
Middle.objects.order_by("pk"),
[repr(r) for r in expected_queryset],
transform=repr,
)
expected_queryset = [bottom]
self.assertQuerySetEqual(
Bottom.objects.order_by("pk"),
[repr(r) for r in expected_queryset],
transform=repr,
)
|