Ticket #21169: test_related_manager_fk_removal.diff

tests/custom_managers/tests.py

@@ -189 +189 @@
             ],
             lambda c: c.first_name
         )
+
+    def test_related_manager_fk_removal(self):
+        self.p1.favorite_book = self.b1
+        self.p1.save()
+        self.p2.favorite_book = self.b1
+        self.p2.save()
+
+        # This should not remove anything because p1 is not considered boring.
+        self.b1.favorite_books(manager='boring_people').remove(self.p1)
+        self.assertQuerysetEqual(
+            self.b1.favorite_books(manager='boring_people').all(), [
+                "Droopy",
+            ],
+            lambda c: c.first_name
+        )
+        self.assertQuerysetEqual(
+            self.b1.favorite_books(manager='fun_people').all(), [
+                "Bugs",
+            ],
+            lambda c: c.first_name
+        )
+
+        # This should only remove fun people, not boring peopel.
+        self.b1.favorite_books(manager='fun_people').clear()
+        self.assertQuerysetEqual(
+            self.b1.favorite_books(manager='boring_people').all(), [
+                "Droopy",
+            ],
+            lambda c: c.first_name
+        )
+        self.assertQuerysetEqual(
+            self.b1.favorite_books(manager='fun_people').all(), [
+            ],
+            lambda c: c.first_name
+        )
+
+        # This should succeed because p2 IS considered boring.
+        self.b1.favorite_books(manager='boring_people').remove(self.p2)
+        self.assertQuerysetEqual(
+            self.b1.favorite_books(manager='boring_people').all(), [
+            ],
+            lambda c: c.first_name
+        )
+        self.assertQuerysetEqual(
+            self.b1.favorite_books(manager='fun_people').all(), [
+            ],
+            lambda c: c.first_name
+        )