From 06cc1040475377071c4e7b2fdf10a8d5235423a4 Mon Sep 17 00:00:00 2001
From: Daniel Kirkham <d.kirkham@kirkham.id.au>
Date: Sun, 18 Feb 2024 22:24:56 +1100
Subject: [PATCH] Extend tests to Wagtail 6.0

---
 tox.ini                            | 3 +++
 wagtailmenus/tests/test_backend.py | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/tox.ini b/tox.ini
index 19652b84..9e355c65 100644
--- a/tox.ini
+++ b/tox.ini
@@ -8,6 +8,8 @@ envlist =
     wt52-dj41-py{38,39,310,311}
     wt52-dj42-py{38,39,310,311,312}
     wt52-dj50-py{310,311,312}
+    wt60-dj42-py{38,39,310,311,312}
+    wt60-dj50-py{310,311,312}
 
 [gh-actions]
 python =
@@ -30,3 +32,4 @@ deps =
     dj42: Django>=4.2,<4.3
     dj50: Django>=5.0,<5.1
     wt52: wagtail>=5.2,<5.3
+    wt60: wagtail>=6.0,<6.1
diff --git a/wagtailmenus/tests/test_backend.py b/wagtailmenus/tests/test_backend.py
index f4f8b5bd..13847821 100644
--- a/wagtailmenus/tests/test_backend.py
+++ b/wagtailmenus/tests/test_backend.py
@@ -46,7 +46,8 @@ def test_copy_footer_menu(self):
         response = form.submit().follow()
 
         assert len(response.context['object_list']) == 4
-        assert '<div class="filterable__filters">' in response
+        # Check for Wagtail 5.2 or >=6.0 response
+        assert '<div class="filterable__filters">' in response or '<div id="filters-drilldown"' in response
 
         # Let's just compare the two menu with the old one
         site_two_footer_menu = FlatMenu.get_for_site('footer', site_two)