Skip to content

Commit e0790ae

Browse files
committed
Remove obsolete compat code
1 parent 3004b93 commit e0790ae

File tree

2 files changed

+15
-42
lines changed

2 files changed

+15
-42
lines changed

wagtail_ab_testing/test/tests/test_add_abtest.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
from django.contrib.contenttypes.models import ContentType
55
from django.test import TestCase
66
from django.urls import reverse
7-
from wagtail import VERSION as WAGTAIL_VERSION
87
from wagtail.models import Page
98
from wagtail.test.utils import WagtailTestUtils
109

@@ -259,14 +258,9 @@ def test_post_add_form_start(self):
259258
self.assertEqual(ab_test.status, AbTest.STATUS_RUNNING)
260259

261260
def test_post_add_form_start_without_publish_permission(self):
262-
if WAGTAIL_VERSION >= (5, 1):
263-
self.moderators_group.page_permissions.filter(
264-
permission__codename="publish_page"
265-
).delete()
266-
else:
267-
self.moderators_group.page_permissions.filter(
268-
permission_type="publish"
269-
).delete()
261+
self.moderators_group.page_permissions.filter(
262+
permission__codename="publish_page"
263+
).delete()
270264

271265
response = self.client.post(
272266
reverse("wagtail_ab_testing_admin:add_ab_test_form", args=[self.page.id]),

wagtail_ab_testing/test/tests/test_progress.py

Lines changed: 12 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from django.contrib.contenttypes.models import ContentType
33
from django.test import TestCase
44
from django.urls import reverse
5-
from wagtail import VERSION as WAGTAIL_VERSION
65
from wagtail.models import Page
76
from wagtail.test.utils import WagtailTestUtils
87

@@ -122,14 +121,9 @@ def test_post_end(self):
122121
self.assertEqual(self.ab_test.status, AbTest.STATUS_CANCELLED)
123122

124123
def test_post_start_without_publish_permission(self):
125-
if WAGTAIL_VERSION >= (5, 1):
126-
self.moderators_group.page_permissions.filter(
127-
permission__codename="publish_page"
128-
).delete()
129-
else:
130-
self.moderators_group.page_permissions.filter(
131-
permission_type="publish"
132-
).delete()
124+
self.moderators_group.page_permissions.filter(
125+
permission__codename="publish_page"
126+
).delete()
133127

134128
self.ab_test.status = AbTest.STATUS_DRAFT
135129
self.ab_test.save()
@@ -150,14 +144,9 @@ def test_post_start_without_publish_permission(self):
150144
self.assertEqual(self.ab_test.status, AbTest.STATUS_DRAFT)
151145

152146
def test_post_pause_without_publish_permission(self):
153-
if WAGTAIL_VERSION >= (5, 1):
154-
self.moderators_group.page_permissions.filter(
155-
permission__codename="publish_page"
156-
).delete()
157-
else:
158-
self.moderators_group.page_permissions.filter(
159-
permission_type="publish"
160-
).delete()
147+
self.moderators_group.page_permissions.filter(
148+
permission__codename="publish_page"
149+
).delete()
161150

162151
response = self.client.post(
163152
reverse("wagtailadmin_pages:edit", args=[self.page.id]),
@@ -175,14 +164,9 @@ def test_post_pause_without_publish_permission(self):
175164
self.assertEqual(self.ab_test.status, AbTest.STATUS_RUNNING)
176165

177166
def test_post_restart_without_publish_permission(self):
178-
if WAGTAIL_VERSION >= (5, 1):
179-
self.moderators_group.page_permissions.filter(
180-
permission__codename="publish_page"
181-
).delete()
182-
else:
183-
self.moderators_group.page_permissions.filter(
184-
permission_type="publish"
185-
).delete()
167+
self.moderators_group.page_permissions.filter(
168+
permission__codename="publish_page"
169+
).delete()
186170

187171
self.ab_test.status = AbTest.STATUS_PAUSED
188172
self.ab_test.save()
@@ -202,14 +186,9 @@ def test_post_restart_without_publish_permission(self):
202186
self.assertEqual(self.ab_test.status, AbTest.STATUS_PAUSED)
203187

204188
def test_post_end_without_publish_permission(self):
205-
if WAGTAIL_VERSION >= (5, 1):
206-
self.moderators_group.page_permissions.filter(
207-
permission__codename="publish_page"
208-
).delete()
209-
else:
210-
self.moderators_group.page_permissions.filter(
211-
permission_type="publish"
212-
).delete()
189+
self.moderators_group.page_permissions.filter(
190+
permission__codename="publish_page"
191+
).delete()
213192

214193
response = self.client.post(
215194
reverse("wagtailadmin_pages:edit", args=[self.page.id]),

0 commit comments

Comments
 (0)