66import pytest
77from wagtail .blocks import StructBlock , StructValue
88
9- from cms .dynamic_content .blocks import PageLink , SourceLinkBlock , check_permissions
9+ from cms .dynamic_content .blocks import PageLink , SourceLinkBlock
1010
1111
1212class TestSourceLinkBlockClean :
@@ -182,14 +182,14 @@ def test_public_page_is_always_authorised(self):
182182 assert result ["title" ] == "Test title"
183183 assert result ["sub_title" ] == "Test subtitle"
184184
185- @mock .patch ("cms.dynamic_content.blocks.check_permissions " )
186- def test_non_public_page_permission_denied (self , mock_check_permissions ):
185+ @mock .patch ("cms.dynamic_content.blocks.check_page_permissions " )
186+ def test_non_public_page_permission_denied (self , mock_check_page_permissions ):
187187 """
188188 Given a non-public page and permissions are denied
189189 When get_api_representation() is called
190190 Then the response is unauthorised and fields are blanked.
191191 """
192- mock_check_permissions .return_value = False
192+ mock_check_page_permissions .return_value = False
193193
194194 block = PageLink ()
195195
@@ -222,16 +222,16 @@ def test_non_public_page_permission_denied(self, mock_check_permissions):
222222 assert result ["sub_title" ] == ""
223223 assert result ["page" ] == ""
224224
225- mock_check_permissions .assert_called_once ()
225+ mock_check_page_permissions .assert_called_once ()
226226
227- @mock .patch ("cms.dynamic_content.blocks.check_permissions " )
228- def test_non_public_page_permission_granted (self , mock_check_permissions ):
227+ @mock .patch ("cms.dynamic_content.blocks.check_page_permissions " )
228+ def test_non_public_page_permission_granted (self , mock_check_page_permissions ):
229229 """
230230 Given a non-public page and permissions are granted
231231 When get_api_representation() is called
232232 Then the response is authorised and fields are preserved.
233233 """
234- mock_check_permissions .return_value = True
234+ mock_check_page_permissions .return_value = True
235235
236236 block = PageLink ()
237237
@@ -265,16 +265,16 @@ def test_non_public_page_permission_granted(self, mock_check_permissions):
265265 assert result ["sub_title" ] == "Test subtitle"
266266 assert result ["page" ] == "https://test-page-url"
267267
268- mock_check_permissions .assert_called_once ()
268+ mock_check_page_permissions .assert_called_once ()
269269
270- @mock .patch ("cms.dynamic_content.blocks.check_permissions " )
271- def test_non_public_page_missing_request (self , mock_check_permissions ):
270+ @mock .patch ("cms.dynamic_content.blocks.check_page_permissions " )
271+ def test_non_public_page_missing_request (self , mock_check_page_permissions ):
272272 """
273273 Given a non-public page and no request in context
274274 When get_api_representation() is called
275275 Then the response is unauthorised and fields are blanked.
276276 """
277- mock_check_permissions .return_value = False
277+ mock_check_page_permissions .return_value = False
278278
279279 block = PageLink ()
280280
@@ -298,4 +298,4 @@ def test_non_public_page_missing_request(self, mock_check_permissions):
298298 assert result ["sub_title" ] == ""
299299 assert result ["page" ] == ""
300300
301- mock_check_permissions .assert_called_once ()
301+ mock_check_page_permissions .assert_called_once ()
0 commit comments