U
{h7< @ s d dl mZ d dlmZ d dlmZmZ d dlmZm Z d dl
m Z m Z d dl
mZ d dlmZ d dlmZ d d lmZmZmZ d d
lmZ G d d d ee Zd
S ) )mock)
Permission) post_delete
pre_delete) HttpRequest HttpResponse)TestCaseoverride_settings)reverse)Page)page_unpublished)
SimplePage
StandardChild
StandardIndex)WagtailTestUtilsc @ s e Zd Zdd Zdd Zedddd Zed ded
d d d
Zed ded
d dd Zed ded
d dd Z dd Z
dd Z dd Z dd Z
dd Zd d Zd d Zd d! Zd"S )#TestPageDeletec C sz t jjdd| _tdddd| _| jj| jd tdd d
| _| jj| jd t d d d
| _
| jj| j
d | | _ d S )
N idz Hello world! hello-worldhellotitleslugcontentinstancez Hello indexz hello-index)r r z Hello Kittyz hello-kitty)
r objectsget root_pager
child_page add_childr child_indexr grandchild_pageloginuserself r( r/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/pages/test_delete_page.pysetUp s zTestPageDelete.setUpc C sH | j td| jjfd}| |jd | tj j
| jjd d S )Nwagtailadmin_pages:deleteargs r ) clientr r
r r assertEqual status_code
assertTruer
r filterexistsr' responser( r( r) test_page_delete# s
z TestPageDelete.test_page_delete
)Z'WAGTAILADMIN_UNSAFE_PAGE_DELETION_LIMITc C s t dddd}| jj|d t dddd}| jj|d | jtd| jjfd }| |jd
| |d |
t j j | jjd
| jtd| jjfd | t j j | jjd
d S )
Nchild 1child-1r r r child 2child-2r+ r, r. +3 pages.%Please type mysite to confirm.r= r )r
r r! r/ r r
r r0 r1 assertContainsr2 r r3 r4 r@ r( r( r) test_confirm_delete_scenario_2C s z-TestPageDelete.test_confirm_delete_scenario_2c C s t dddd}| jj|d t dddd}| jj|d | jjtd| jjfd d
d id }| |jd
dd |j d D }| t
|d | |d | |d | |d | t j
j| jjd d S )Nr9 r: r r r r; r< r+ r, confirm_site_namerandomdatar. c S s g | ]
}|j qS r( )message).0mr( r( r)
g s zATestPageDelete.test_confirm_delete_scenario_3..messages rF rG r= r )r
r r! r/ r> r
r r0 r1 contextlenrH r2 r r3 r4 )r' rA rB r6 rR r( r( r) test_confirm_delete_scenario_3W s z-TestPageDelete.test_confirm_delete_scenario_3c C s t dddd}| jj|d t dddd}| jj|d | jjtd| jjfd d
d id }| |td
| jjfd | t j
j | jjd | t j
j |jd | t j
j |jd d S )Nr9 r: r r r r; r< r+ r, rJ rE rL wagtailadmin_explorer )
r
r r! r/ r> r
r assertRedirectsr r? r r3 r4 r@ r( r( r) test_confirm_delete_scenario_4o s z-TestPageDelete.test_confirm_delete_scenario_4c C s> | j td| jjfd}| |jd | || j d S )Nr+ r, r. ) r/ r r
r r r0 r1 rH Zget_admin_display_titler5 r( r( r) %test_page_delete_specific_admin_title s
z4TestPageDelete.test_page_delete_specific_admin_titlec C st d| j _| j jtjjddd | j | jt d| j
j fd}| |j
d | tjj| j
j d d S ) NFZ wagtailadminZ access_admin)Zcontent_type__app_labelcodenamer+ r, i. r )r% Z is_superuserZuser_permissionsaddr r r saver/ r
r r r0 r1 r2 r
r3 r4 r5 r( r( r) test_page_delete_bad_permissions s
z/TestPageDelete.test_page_delete_bad_permissionsc C s t }t| z| jtd| jj fd}|
|td| j j fd | j t
t dd | tjj| j jdd d | |jd |jd d
}| |d | jj | |d | j | |d | jj W 5 t| X d S )
Nr+ r, rW $treebeard found consistency problemsmsgr Zpath__startswithr r rS r senderr )r MagicMockr connect
disconnectr/ r> r
r r rX r r? anyr
find_problemsr0 r r3 pathcount
call_countZ
mock_callsZspecific_classassertIsInstance)r' mock_handlerr6 Z mock_callr( r( r) test_page_delete_post s6
z$TestPageDelete.test_page_delete_postc C s d| j _| j t }t| z|| j t
d| j j fd}| |t
d| j
j fd | jtt dd | tjj| j
jdd d | |jd W 5 t| X d S )
NFr+ r, rW r_ r` r rb r )r liver] r rd r re rf r/ r> r
r rX r r? rg r rh r0 r r3 ri rj rk )r' rm r6 r( r( r) test_page_delete_notlive_post s2
z,TestPageDelete.test_page_delete_notlive_postc s g g g fdd}fdd} fdd}t | t| t| z6| jtd| jj fd}|
|td | j j fd | j t
t d
d | tjj| jj d | tjj| jj d | tjj| jj d | tjj| jj d | t| jj f | t| jj f | t| jj f | t| jj f | t| jj f | t| jj f W 5 t | t| t| X d S )
Nc s | |jf d S Nappendpkrc r kwargs)unpublish_signals_receivedr( r) page_unpublished_handler s zFTestPageDelete.test_subpage_deletion..page_unpublished_handlerc s | |jf d S rq rr ru )pre_delete_signals_receivedr( r) pre_delete_handler s z@TestPageDelete.test_subpage_deletion..pre_delete_handlerc s | |jf d S rq rr ru ) post_delete_signals_receivedr( r) post_delete_handler s zATestPageDelete.test_subpage_deletion..post_delete_handlerr+ r, rW r_ r` r )r re r r rf r/ r> r
r" r rX r r? rg r rh r r r3 r4 r r# ZassertIn)r' rx rz r| r6 r( )r{ ry rw r) test_subpage_deletion sp
z$TestPageDelete.test_subpage_deletionc s^ fdd} d| jtd jjfd}W 5 Q R X |jd |jd d S )Nc s& | t |j jj tdS Nz Overridden!rl r r0 r r r requestpager& r( r) hook_func> s z>TestPageDelete.test_before_delete_page_hook..hook_funcbefore_delete_pager+ r, r. Overridden!)
register_hookr/ r r
r r r0 r1 r r' r r6 r( r& r) test_before_delete_page_hook= s z+TestPageDelete.test_before_delete_page_hookc sz fdd} d| jtd jjfd}W 5 Q R X |jd |jd t
j j jjd
d S ) Nc s& | t |j jj tdS r~ r r r& r( r) r M s zCTestPageDelete.test_before_delete_page_hook_post..hook_funcr r+ r, r. r r )r r/ r> r
r r r0 r1 r r2 r r r3 r4 r r( r& r) !test_before_delete_page_hook_postL s z0TestPageDelete.test_before_delete_page_hook_postc sz fdd} d| jtd jjfd}W 5 Q R X |jd |jd t
j j jjd
d S ) Nc s& | t |j jj tdS r~ r r r& r( r) r _ s z=TestPageDelete.test_after_delete_page_hook..hook_funcZafter_delete_pager+ r, r. r r )r r/ r> r
r r r0 r1 r r? r r r3 r4 r r( r& r) test_after_delete_page_hook^ s z*TestPageDelete.test_after_delete_page_hookN)__name__
__module__ __qualname__r* r7 r rC rI rV rY rZ r^ rn rp r} r r r r( r( r( r) r s*
')Ir N)Zunittestr Zdjango.contrib.auth.modelsr Zdjango.db.models.signalsr r Z django.httpr r Z django.testr r Z django.urlsr
Zwagtail.modelsr Zwagtail.signalsr Zwagtail.test.testapp.modelsr
r r Zwagtail.test.utilsr r r( r( r( r) s