summaryrefslogtreecommitdiff
path: root/tests/model_inheritance_select_related/tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/model_inheritance_select_related/tests.py')
-rw-r--r--tests/model_inheritance_select_related/tests.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/tests/model_inheritance_select_related/tests.py b/tests/model_inheritance_select_related/tests.py
new file mode 100644
index 0000000000..078b466d0e
--- /dev/null
+++ b/tests/model_inheritance_select_related/tests.py
@@ -0,0 +1,31 @@
+from __future__ import absolute_import
+
+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")