Fix collections import for latest Wagtail main (#240)

torchbox · Apr 3, 2024 · 4b2b18e · 4b2b18e
1 parent 327267e
commit 4b2b18e
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/testapp/migrations/0001_initial.py b/tests/testapp/migrations/0001_initial.py
@@ -6,13 +6,18 @@
 import taggit.managers
 import wagtail.blocks
 import wagtail.fields
-import wagtail.models.collections
 import wagtail.search.index
 import wagtailmedia.blocks
 
 from django.conf import settings
 from django.db import migrations, models
 
+
+try:
+    import wagtail.models.media as collections
+except ImportError:
+    import wagtail.models.collections as collections
+
 import tests.testapp.models
 
 
@@ -169,7 +174,7 @@ class Migration(migrations.Migration):
                 (
                     "collection",
                     models.ForeignKey(
-                        default=wagtail.models.collections.get_root_collection_id,
+                        default=collections.get_root_collection_id,
                         on_delete=django.db.models.deletion.CASCADE,
                         related_name="+",
                         to="wagtailcore.collection",