2
2
from django .contrib .contenttypes .models import ContentType
3
3
from django .test import TestCase
4
4
from django .urls import reverse
5
- from wagtail import VERSION as WAGTAIL_VERSION
6
5
from wagtail .models import Page
7
6
from wagtail .test .utils import WagtailTestUtils
8
7
@@ -122,14 +121,9 @@ def test_post_end(self):
122
121
self .assertEqual (self .ab_test .status , AbTest .STATUS_CANCELLED )
123
122
124
123
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 ()
133
127
134
128
self .ab_test .status = AbTest .STATUS_DRAFT
135
129
self .ab_test .save ()
@@ -150,14 +144,9 @@ def test_post_start_without_publish_permission(self):
150
144
self .assertEqual (self .ab_test .status , AbTest .STATUS_DRAFT )
151
145
152
146
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 ()
161
150
162
151
response = self .client .post (
163
152
reverse ("wagtailadmin_pages:edit" , args = [self .page .id ]),
@@ -175,14 +164,9 @@ def test_post_pause_without_publish_permission(self):
175
164
self .assertEqual (self .ab_test .status , AbTest .STATUS_RUNNING )
176
165
177
166
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 ()
186
170
187
171
self .ab_test .status = AbTest .STATUS_PAUSED
188
172
self .ab_test .save ()
@@ -202,14 +186,9 @@ def test_post_restart_without_publish_permission(self):
202
186
self .assertEqual (self .ab_test .status , AbTest .STATUS_PAUSED )
203
187
204
188
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 ()
213
192
214
193
response = self .client .post (
215
194
reverse ("wagtailadmin_pages:edit" , args = [self .page .id ]),
0 commit comments