Ticket #16781: t16781-tests-modeltest-reverse_lookup-multi_level_test.diff

tests/regressiontests/queries/models.py

@@ -343 +343 @@
     def __unicode__(self):
         return "one2one " + self.new_name
     
+class User(models.Model):
+    name = models.CharField(max_length=50)
+
+    def __unicode__(self):
+        return self.name
+
+class Address(models.Model):
+    user = models.ForeignKey(User)
+    address = models.CharField(max_length=100)
+
+class Relationship(models.Model):
+    status = models.CharField(max_length=1)
+    from_user = models.ForeignKey(User, related_name='relationships_from')
+    to_user = models.ForeignKey(User, related_name='relationships_to')
+

tests/regressiontests/queries/tests.py

@@ -15 +15 @@
     DumbCategory, ExtraInfo, Fan, Item, LeafA, LoopX, LoopZ, ManagedModel,
     Member, NamedCategory, Note, Number, Plaything, PointerA, Ranking, Related,
     Report, ReservedName, Tag, TvChef, Valid, X, Food, Eaten, Node, ObjectA, ObjectB,
-    ObjectC, CategoryItem, SimpleCategory, SpecialCategory, OneToOneCategory)
+    ObjectC, CategoryItem, SimpleCategory, SpecialCategory, OneToOneCategory, User,
+    Address, Relationship)
 
 
 class BaseQuerysetTest(TestCase):
@@ -1276 +1277 @@
         )
 
 
+class Queries7Tests(TestCase):
+    def setUp(self):
+        john = User.objects.create(name="John Doe")
+        jim = User.objects.create(name="Jim Bo")
+        jerry = User.objects.create(name="Jerry Noh")
+        first_address = Address.objects.create(
+            address="First address",
+            user=john
+        )
+        second_address = Address.objects.create(
+            address="Second address",
+            user=jim
+        )
+        jerry_is_johns_friend = Relationship.objects.create(
+            status='f',
+            from_user=john,
+            to_user=jerry
+        )
+        john_is_jerrys_enemy = Relationship.objects.create(
+            status='e',
+            from_user=jerry,
+            to_user=john
+        )
+
+    def test_ticket16781_same_reverse_field(self):
+        u1 = User.objects.get(name__exact="Jerry Noh")
+        # Retrieve all polls for which the user is not the From User in a From-User relationship
+        address1 = Address.objects.exclude(user__relationships_from__from_user__id=u1.id)
+        self.assertEqual(address1.count(), 1)
+
+    def test_ticket16781_differing_reverse_field(self):
+        u1 = User.objects.get(name__exact="Jerry Noh")
+        # Retrieve all polls for which the user is not the From User in a To-User relationship
+        address1 = Address.objects.exclude(user__relationships_to__from_user__id=u1.id)
+        self.assertEqual(address1.count(), 1)
+
+
 class SelectRelatedTests(TestCase):
     def test_tickets_3045_3288(self):
         # Once upon a time, select_related() with circular relations would loop