django1/tests/regressiontests/model_inheritance_select_re.../tests.py

30 lines
968 B
Python

from operator import attrgetter
from django.test import TestCase
from models import Restaurant, Person
class ModelInheritanceSelectRelatedTests(TestCase):
def test_inherited_select_related(self):
# Regression test for #7246
r1 = Restaurant.objects.create(
name="Nobu", serves_sushi=True, serves_steak=False
)
r2 = Restaurant.objects.create(
name="Craft", serves_sushi=False, serves_steak=True
)
p1 = Person.objects.create(name="John", favorite_restaurant=r1)
p2 = Person.objects.create(name="Jane", favorite_restaurant=r2)
self.assertQuerysetEqual(
Person.objects.order_by("name").select_related(), [
"Jane",
"John",
],
attrgetter("name")
)
jane = Person.objects.order_by("name").select_related("favorite_restaurant")[0]
self.assertEqual(jane.favorite_restaurant.name, "Craft")