Initial commit

env/lib/python3.10/site-packages/treebeard/tests/test_migrations.py

@@ -0,0 +1,54 @@
+"""
+Check that all changes to Treebeard models have had migrations created in our test app. If there
+are outstanding model changes that need migrations, fail the tests.
+
+This module is taken from https://github.com/wagtail/wagtail/blob/master/wagtail/core/tests/test_migrations.py.
+"""
+
+from django.apps import apps
+from django.db.migrations.autodetector import MigrationAutodetector
+from django.db.migrations.loader import MigrationLoader
+from django.db.migrations.questioner import MigrationQuestioner
+from django.db.migrations.state import ProjectState
+from django.test import TestCase
+
+
+class TestForMigrations(TestCase):
+    def test__migrations(self):
+        app_labels = set(app.label for app in apps.get_app_configs() if app.name.startswith('tests.'))
+        for app_label in app_labels:
+            apps.get_app_config(app_label.split('.')[-1])
+        loader = MigrationLoader(None, ignore_no_migrations=True)
+
+        conflicts = dict(
+            (app_label, conflict)
+            for app_label, conflict in loader.detect_conflicts().items()
+            if app_label in app_labels
+        )
+
+        if conflicts:
+            name_str = "; ".join("%s in %s" % (", ".join(names), app) for app, names in conflicts.items())
+            self.fail("Conflicting migrations detected (%s)." % name_str)
+
+        autodetector = MigrationAutodetector(
+            loader.project_state(),
+            ProjectState.from_apps(apps),
+            MigrationQuestioner(specified_apps=app_labels, dry_run=True),
+        )
+
+        changes = autodetector.changes(
+            graph=loader.graph,
+            trim_to_apps=app_labels or None,
+            convert_apps=app_labels or None,
+        )
+
+        if changes:
+            migrations = '\n'.join((
+                '  {migration}\n{changes}'.format(
+                    migration=migration,
+                    changes='\n'.join('    {0}'.format(operation.describe())
+                                      for operation in migration.operations))
+                for (_, migrations) in changes.items()
+                for migration in migrations))
+
+            self.fail('Model changes with no migrations detected:\n%s' % migrations)