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

32 lines
1000 B
Python

from __future__ import unicode_literals
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
)
Person.objects.create(name="John", favorite_restaurant=r1)
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")