Tests for #22325

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/migrations/test_executor.py b/tests/migrations/test_executor.py
index e8dcacd80e..be3909d719 100644
--- a/tests/migrations/test_executor.py
+++ b/tests/migrations/test_executor.py
@@ -167,3 +167,25 @@ class ExecutorTests(MigrationTestBase):
         executor.migrate([("migrations", None)])
         self.assertTableNotExists("migrations_author")
         self.assertTableNotExists("migrations_tribble")
+
+    @override_settings(
+        MIGRATION_MODULES={"migrations": "migrations.test_migrations_custom_user"},
+        AUTH_USER_MODEL="migrations.Author",
+    )
+    def test_custom_user(self):
+        """
+        Regression test for #22325 - references to a custom user model defined in the
+        same app are not resolved correctly.
+        """
+        executor = MigrationExecutor(connection)
+        self.assertTableNotExists("migrations_author")
+        self.assertTableNotExists("migrations_tribble")
+        # Migrate forwards
+        executor.migrate([("migrations", "0001_initial")])
+        self.assertTableExists("migrations_author")
+        self.assertTableExists("migrations_tribble")
+        # And migrate back to clean up the database
+        executor.loader.build_graph()
+        executor.migrate([("migrations", None)])
+        self.assertTableNotExists("migrations_author")
+        self.assertTableNotExists("migrations_tribble")