Add more tests to delete

Akash-Kumar-Sen · Akash-Kumar-Sen · commit ffdc28dc4b04 · 2024-04-14T19:07:33.000+05:30
diff --git a/tests/delete/models.py b/tests/delete/models.py
@@ -61,12 +61,16 @@ class A(models.Model):
     cascade_nullable = models.ForeignKey(
         R, models.CASCADE, null=True, related_name="cascade_nullable_set"
     )
+    db_cascade = models.ForeignKey(R, models.DB_CASCADE, null=True, related_name="db_cascade_set")
     protect = models.ForeignKey(
         R, models.PROTECT, null=True, related_name="protect_set"
     )
     restrict = models.ForeignKey(
         R, models.RESTRICT, null=True, related_name="restrict_set"
     )
+    db_restrict = models.ForeignKey(
+        R, models.DB_RESTRICT, null=True, related_name="db_restrict_set"
+    )
     donothing = models.ForeignKey(
         R, models.DO_NOTHING, null=True, related_name="donothing_set"
     )
@@ -77,6 +81,9 @@ class A(models.Model):
     cascade_p = models.ForeignKey(
         P, models.CASCADE, related_name="cascade_p_set", null=True
     )
+    db_cascade_p = models.ForeignKey(
+        P, models.DB_CASCADE, related_name="db_cascade_p_set", null=True
+    )
 
     # A OneToOneField is just a ForeignKey unique=True, so we don't duplicate
     # all the tests; just one smoke test to ensure on_delete works for it as
@@ -101,8 +108,10 @@ def create_a(name):
         "setdefault_none",
         "cascade",
         "cascade_nullable",
+        "db_cascade",
         "protect",
         "restrict",
+        "db_restrict",
         "donothing",
         "o2o_setnull",
     ):
diff --git a/tests/delete/tests.py b/tests/delete/tests.py
@@ -321,6 +321,27 @@ def test_restrict_gfk_no_fast_delete(self):
         self.assertFalse(GenericB2.objects.exists())
         self.assertFalse(GenericDeleteBottom.objects.exists())
 
+    def test_db_restrict_path_db_cascade_direct(self):
+        a = create_a("db_restrict")
+        a.db_restrict.p = P.objects.create()
+        a.db_restrict.save()
+        a.db_cascade_p = a.db_restrict.p
+        a.save()
+        a.db_restrict.p.delete()
+        self.assertFalse(A.objects.filter(name="db_restrict").exists())
+        self.assertFalse(R.objects.filter(pk=a.db_restrict_id).exists())
+
+    def test_db_restrict_path_cascade_direct(self):
+        a = create_a("db_restrict")
+        a.db_restrict.p = P.objects.create()
+        a.db_restrict.save()
+        a.cascade_p = a.db_restrict.p
+        a.save()
+        a.db_restrict.p.delete()
+        self.assertFalse(A.objects.filter(name="db_restrict").exists())
+        self.assertFalse(R.objects.filter(pk=a.db_restrict_id).exists())
+
+
 
 class DeletionTests(TestCase):
     def test_sliced_queryset(self):
