Fix test for Wagtail 4.1

torchbox · Feb 5, 2024 · 9201673 · 9201673
1 parent 5260b09
commit 9201673
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/tests/test_admin.py b/tests/test_admin.py
@@ -26,10 +26,19 @@ def test_media_field_in_admin(self):
                 args=("wagtailmedia_tests", "eventpage", self.root_page.id),
             )
         )
-        self.assertContains(
-            response,
-            '<script>createMediaChooser("id_related_media-__prefix__-link_media");</script>',
-        )
+
+        from wagtail import VERSION as WAGTAIL_VERSION
+
+        if WAGTAIL_VERSION >= (4, 2):
+            self.assertContains(
+                response,
+                '<script>createMediaChooser("id_related_media-__prefix__-link_media");</script>',
+            )
+        else:
+            self.assertContains(
+                response,
+                '<script>createMediaChooser("id_related_media-__prefix__-link_media");<-/script>',
+            )
         self.assertContains(response, "media-chooser.js")
 
     def test_media_block_in_admin(self):