Fix tests

2022-11-18 16:44:42 +00:00
parent 5642e117f1
commit e31793e748
8 changed files with 41 additions and 24 deletions
--- a/tests/unit/test_wagtail_hooks.py
+++ b/tests/unit/test_wagtail_hooks.py
@ -1,6 +1,11 @@
 import pytest
 from django.http import Http404
-from wagtail.core.models import Page
+from wagtail import VERSION as WAGTAIL_VERSION
+
+if WAGTAIL_VERSION >= (3, 0):
+    from wagtail.models import Page
+else:
+    from wagtail.core.models import Page

 from tests.factories.page import ContentPageFactory
 from tests.factories.segment import SegmentFactory