Fix flaky test for GFK

model-bakers · Aug 26, 2024 · c984eea · c984eea
1 parent f9f441c
commit c984eea
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/generic/models.py b/tests/generic/models.py
@@ -287,7 +287,9 @@ class UnsupportedModel(models.Model):
 
     class DummyGenericForeignKeyModel(models.Model):
         content_type = models.ForeignKey(
-            contenttypes.ContentType, on_delete=models.CASCADE
+            contenttypes.ContentType,
+            on_delete=models.CASCADE,
+            limit_choices_to={"model__in": ["person", "dog"]},
         )
         object_id = models.PositiveIntegerField()
         content_object = GenericForeignKey("content_type", "object_id")
@@ -297,7 +299,11 @@ class DummyGenericRelationModel(models.Model):
 
     class GenericForeignKeyModelWithOptionalData(models.Model):
         content_type = models.ForeignKey(
-            contenttypes.ContentType, on_delete=models.CASCADE, blank=True, null=True
+            contenttypes.ContentType,
+            on_delete=models.CASCADE,
+            blank=True,
+            null=True,
+            limit_choices_to={"model__in": ["person", "dog"]},
         )
         object_id = models.PositiveIntegerField(blank=True, null=True)
         content_object = GenericForeignKey("content_type", "object_id")