Ticket #13839: test13839.patch

tests/regressiontests/select_related_onetoone/models.py

@@ -52 +52 @@
 class Product(models.Model):
     name = models.CharField(max_length=100)
     image = models.OneToOneField(Image, null=True)
+
+
+class UserChild(User):
+    pass

tests/regressiontests/select_related_onetoone/tests.py

@@ -3 +3 @@
 from django.test import TestCase
 
 from models import (User, UserProfile, UserStat, UserStatResult, StatDetails,
-    AdvancedUserStat, Image, Product)
+    AdvancedUserStat, Image, Product, UserChild)
 
 class ReverseSelectRelatedTestCase(TestCase):
     def setUp(self):
@@ -92 +92 @@
         p2 = Product.objects.create(name="Talking Django Plushie")
 
         self.assertEqual(len(Product.objects.select_related("image")), 2)
+
+    def test_select_related_with_inheritance(self):
+        """
+        select_related should not pre-fill the SingleRelatedObjectDescriptor
+        cache with None when the related object does not exist.
+
+        This tests both the inheritance and the standard OneToOneField usage
+        case.
+        """
+        User.objects.create()
+        user = User.objects.all()[0]
+
+        self.assertRaises(UserChild.DoesNotExist, lambda: user.userchild)
+        self.assertRaises(UserProfile.DoesNotExist, lambda: user.userprofile)
+
+        user = User.objects.all().select_related('userchild', 'userprofile')[0]
+        self.assertRaises(UserChild.DoesNotExist, lambda: user.userchild)
+        self.assertRaises(UserProfile.DoesNotExist, lambda: user.userprofile)