Refs #21961 -- Added DatabaseFeatures.supports_on_delete_db_(cascade/null) feature flags.

Needed on MongoDB.

Co-Authored-By: Mariusz Felisiak <felisiak.mariusz@gmail.com>

1 files changed, 13 insertions, 13 deletions

diff --git a/tests/delete/models.py b/tests/delete/models.py
index bd9caf42a7..d8f81827be 100644
--- a/tests/delete/models.py
+++ b/tests/delete/models.py
@@ -49,17 +49,26 @@ class RelatedDbOptionParent(models.Model):
     p = models.ForeignKey(RelatedDbOptionGrandParent, models.DB_CASCADE, null=True)
 
 
-class RelatedDbOption(models.Model):
+class CascadeDbModel(models.Model):
     name = models.CharField(max_length=30)
+    db_cascade = models.ForeignKey(
+        RelatedDbOptionParent, models.DB_CASCADE, related_name="db_cascade_set"
+    )
+
+    class Meta:
+        required_db_features = {"supports_on_delete_db_cascade"}
+
+
+class SetNullDbModel(models.Model):
     db_setnull = models.ForeignKey(
         RelatedDbOptionParent,
         models.DB_SET_NULL,
         null=True,
         related_name="db_setnull_set",
     )
-    db_cascade = models.ForeignKey(
-        RelatedDbOptionParent, models.DB_CASCADE, related_name="db_cascade_set"
-    )
+
+    class Meta:
+        required_db_features = {"supports_on_delete_db_null"}
 
 
 class SetDefaultDbModel(models.Model):
@@ -159,15 +168,6 @@ def create_a(name):
     return a
 
 
-def create_related_db_option(name):
-    a = RelatedDbOption(name=name)
-    for name in ["db_setnull", "db_cascade"]:
-        r = RelatedDbOptionParent.objects.create()
-        setattr(a, name, r)
-    a.save()
-    return a
-
-
 class M(models.Model):
     m2m = models.ManyToManyField(R, related_name="m_set")
     m2m_through = models.ManyToManyField(R, through="MR", related_name="m_through_set")