Fixed #25040 -- Fixed migrations state crash with GenericForeignKey

1 files changed, 7 insertions, 0 deletions

diff --git a/tests/migrations/test_state.py b/tests/migrations/test_state.py
index 7a959c53eb..0189d35d82 100644
--- a/tests/migrations/test_state.py
+++ b/tests/migrations/test_state.py
@@ -1,4 +1,5 @@
 from django.apps.registry import Apps
+from django.contrib.contenttypes.fields import GenericForeignKey
 from django.db import models
 from django.db.migrations.exceptions import InvalidBasesError
 from django.db.migrations.operations import (
@@ -1049,6 +1050,12 @@ class RelatedModelsTests(SimpleTestCase):
         self.assertRelated(S, [A, B, T])
         self.assertRelated(T, [A, B, S])
 
+    def test_generic_fk(self):
+        A = self.create_model("A", foreign_keys=[models.ForeignKey('B'), GenericForeignKey()])
+        B = self.create_model("B", foreign_keys=[models.ForeignKey('C')])
+        self.assertRelated(A, [B])
+        self.assertRelated(B, [A])
+
     def test_abstract_base(self):
         A = self.create_model("A", abstract=True)
         B = self.create_model("B", bases=(A,))