Ticket #17485: 17485-only-select_related.diff

tests/regressiontests/defer_regress/models.py

@@ -52 +52 @@
 class Feature(models.Model):
     item = models.ForeignKey(SimpleItem)
 
+class SpecialFeature(models.Model):
+    feature = models.ForeignKey(Feature)
+
 class ItemAndSimpleItem(models.Model):
     item = models.ForeignKey(Item)
     simple = models.ForeignKey(SimpleItem)

tests/regressiontests/defer_regress/tests.py

@@ -9 +9 @@
 from django.test import TestCase
 
 from .models import (ResolveThis, Item, RelatedItem, Child, Leaf, Proxy,
-    SimpleItem, Feature, ItemAndSimpleItem)
+    SimpleItem, Feature, ItemAndSimpleItem, SpecialFeature)
 
 
 class DeferRegressionTest(TestCase):
@@ -115 +115 @@
                 RelatedItem,
                 ResolveThis,
                 SimpleItem,
+                SpecialFeature,
             ]
         )
 
@@ -152 +153 @@
                 "RelatedItem_Deferred_item_id",
                 "ResolveThis",
                 "SimpleItem",
+                "SpecialFeature",
             ]
         )
 
@@ -197 +199 @@
         self.assertEqual(obj.item, item2)
         self.assertEqual(obj.item_id, item2.id)
 
+    def test_only_with_select_related(self):
+        item = SimpleItem.objects.create(name='first', value=47)
+        feature = Feature.objects.create(item=item)
+        special_feature = SpecialFeature.objects.create(feature=feature)
+
+        qs = Feature.objects.only('item__name').select_related('item__name')
+        self.assertEqual(len(qs), 1)
+
+        qs = SpecialFeature.objects.only('feature__item__name').select_related('feature__item__name')
+        self.assertEqual(len(qs), 1)
+
     def test_deferred_class_factory(self):
         from django.db.models.query_utils import deferred_class_factory
         new_class = deferred_class_factory(Item,