32 lines
1010 B
Python
32 lines
1010 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
|
|
)
|
|
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")
|