|
@@ -0,0 +1,29 @@
|
|
|
+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")
|