U {h%@sddlmZddlmZddlmZmZddlmZddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZGd d d eeZGd ddeeZdS))mock) Permission) HttpRequest HttpResponse)TestCase)reverse) gettext_lazy)Page)page_unpublished) SimplePage)WagtailTestUtilsc@sLeZdZddZddZddZddZd d Zd d Zd dZ ddZ dS)TestPageUnpublishcCs@||_tjjdd|_tddddd|_|jj|jddS) Nid Hello world! hello-worldhelloT)titleslugcontentliveinstance) loginuserr objectsget root_pager page add_childselfr#u/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/pages/test_unpublish_page.pysetUps zTestPageUnpublish.setUpcCs8|jtd|jjfd}||jd||ddS)z\ This tests that the unpublish view responds with an unpublish confirm page wagtailadmin_pages:unpublishargs)wagtailadmin/pages/confirm_unpublish.htmlN)clientrrrr assertEqual status_codeassertTemplateUsedr"responser#r#r$test_unpublish_views z%TestPageUnpublish.test_unpublish_viewcCs&|jtddd}||jddS)z_ This tests that the unpublish view returns an error if the page id is invalid r&)i90r'iN)r+rrr,r-r/r#r#r$#test_unpublish_view_invalid_page_id*s z5TestPageUnpublish.test_unpublish_view_invalid_page_idcCsXd|j_|jjtjjddd|j|jt d|j j fd}| |j ddS) zf This tests that the unpublish view doesn't allow users without unpublish permissions FZ wagtailadminZ access_admin)Zcontent_type__app_labelcodenamer&r'i.N)rZ is_superuserZuser_permissionsaddrrrsaver+rrrr,r-r/r#r#r$#test_unpublish_view_bad_permissions6s z5TestPageUnpublish.test_unpublish_view_bad_permissionsc Cst}t|z|jtd|jj fd}| |td|j j fd| t jj|jj dj||jd|jdd}||d|jj||d |j||d |jjW5t|Xd S) z[ This posts to the unpublish view and checks that the page was unpublished r&r'wagtailadmin_explorerrrZsenderrN)rZ MagicMockr connectZ disconnectr+postrrrassertRedirectsr assertFalser rrrr,Z call_countZ mock_callsZspecific_classassertIsInstance)r"Z mock_handlerr0Z mock_callr#r#r$test_unpublish_view_postKs" z*TestPageUnpublish.test_unpublish_view_postc sfdd}d|&i}jtdjjfd|}W5QRX|jd|jdj jj t ddS) Ncs&|t|jjjtdSNz Overridden!r=rr,rrrrequestrr!r#r$ hook_funcls z>TestPageUnpublish.test_after_unpublish_page..hook_funcZafter_unpublish_pager&r'r) Overridden!Zdraft register_hookr+r:rrrr,r-rZrefresh_from_dbZ status_string_r"rCZ post_datar0r#r!r$test_after_unpublish_pageks  z+TestPageUnpublish.test_after_unpublish_pagec sfdd}d|&i}jtdjjfd|}W5QRX|jd|jdj jj t ddS) Ncs&|t|jjjtdSr?r@rAr!r#r$rCs z?TestPageUnpublish.test_before_unpublish_page..hook_funcZbefore_unpublish_pager&r'r)rDrrErHr#r!r$test_before_unpublish_page~s  z,TestPageUnpublish.test_before_unpublish_pagecCsD|jtd|jjfd}||jd||d||ddS)z This tests that the unpublish view responds with an unpublish confirm page that does not contain the form field 'include_descendants' r&r'r)r*name="include_descendants"N) r+rrrrr,r-r.ZassertNotContainsr/r#r#r$test_unpublish_descendants_views z1TestPageUnpublish.test_unpublish_descendants_viewN) __name__ __module__ __qualname__r%r1r2r6r>rIrJrLr#r#r#r$r s    r c@s,eZdZddZddZddZddZd S) %TestPageUnpublishIncludingDescendantscCsx||_tjjdd|_|jjtddddddd |_|jjtd d ddddd |_ |jjtd d ddddd |_ dS)NrrrrrTF)rrrrZhas_unpublished_changesrz Child pagez child-pagezAnother Child pagezanother-child-page) rrr rrrr r test_pagetest_child_pagetest_another_child_pager!r#r#r$r%s:   z+TestPageUnpublishIncludingDescendants.setUpcCsD|jtd|jjfd}||jd||d||ddS)z This tests that the unpublish view responds with an unpublish confirm page that contains the form field 'include_descendants' r&r'r)r*rKN) r+rrrQrr,r-r.ZassertContainsr/r#r#r$rLs zETestPageUnpublishIncludingDescendants.test_unpublish_descendants_viewcCs|jtd|jjfdddi}||td|jjfd|tj j |jjdj |tj j |j jdj |tj j |j jdj dS)zp This posts to the unpublish view and checks that the page and its descendants were unpublished r&r'Zinclude_descendantsonr7rN)r+r:rrQrr;rr<r rrrrRrSr/r#r#r$)test_unpublish_include_children_view_postszOTestPageUnpublishIncludingDescendants.test_unpublish_include_children_view_postcCs|jtd|jjfdi}||td|jjfd|tj j |jjdj | tj j |j jdj | tj j |jjdj dS)zx This posts to the unpublish view and checks that the page was unpublished but its descendants were not r&r'r7rN)r+r:rrQrr;rr<r rrrZ assertTruerRrSr/r#r#r$-test_unpublish_not_include_children_view_postszSTestPageUnpublishIncludingDescendants.test_unpublish_not_include_children_view_postN)rMrNrOr%rLrUrVr#r#r#r$rPs%rPN)ZunittestrZdjango.contrib.auth.modelsrZ django.httprrZ django.testrZ django.urlsrZdjango.utils.translationrrGZwagtail.modelsr Zwagtail.signalsr Zwagtail.test.testapp.modelsr Zwagtail.test.utilsr r rPr#r#r#r$s