U
|h @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
mZ d dlmZmZmZmZ d d lmZ G d
d d e ZG d d
d
e ZG dd dee ZdS ) )StringIO)quote)
Permission)
management)TestCase)reverse)PageReferenceIndex)AdvertDraftStateModel EventPageGenericSnippetPage)WagtailTestUtilsc s. e Zd ZdgZe fddZdd Z ZS )TestUsageCount test.jsonc s" t t }tjd|d d S NZrebuild_references_index)stdoutsuper
setUpTestDatar r Z call_commandclsoutput __class__ i/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/snippets/tests/test_usage.pyr s
z TestUsageCount.setUpTestDatac C s( t jjdd}| t| d d S )N pk )r
objectsget assertEqualr get_grouped_references_tocount)selfadvertr r r test_snippet_usage_count s z'TestUsageCount.test_snippet_usage_count)__name__
__module__ __qualname__fixtures classmethodr r(
__classcell__r r r r r s r c s. e Zd ZdgZe fddZdd Z ZS )
TestUsedByr c s" t t }tjd|d d S r r r r r r r $ s
zTestUsedBy.setUpTestDatac C sl t jjdd}t|}| |d t | |d d t | |d d t | |d d d t d S )Nr r r ) r
r! r" r r$ ZassertIsInstancetupler list)r& r' usager r r test_snippet_used_by* s
z TestUsedBy.test_snippet_used_by)r) r* r+ r, r- r r3 r. r r r r r/ ! s r/ c @ sR e Zd ZdgZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dd Z dS )TestSnippetUsageViewr c C s | | _d S )N)loginuser)r& r r r setUp7 s zTestSnippetUsageView.setUpc C sr t jjdd}| d|_| | jtdt |j
gd}| |j }|
d}| |jddd d S ) Nz Draft-enabled Foo, Published)textzDraft-enabled Bar, In Draftz+wagtailsnippets_tests_draftstatemodel:usageargsz.w-breadcrumbs__sublabelT)strip)r r! createZ
save_revisionpublishr8 clientr" r r r Zget_soupcontentZ
select_oner# Zget_text)r& ZsnippetresponseZsoupZsublabelr r r test_use_latest_draft_as_title: s
z3TestSnippetUsageView.test_use_latest_draft_as_titlec C s t jjdd}| tdtjjddd}|j|d | jtddgd }| |d
| |d | j |d d d
| |d | j |dd d
| j
|dd d
| |d d S )Nr r Foobar Titler titleZsnippet_content_objectinstance"wagtailsnippets_tests_advert:usage1r9 !Welcome to the Wagtail test site! | Generic snippet page | ThtmlSnippet content objectField | If you confirm deletion | ) r r! r" saver
r
add_childr> r assertContainsassertNotContains)r& pagegfk_pager@ r r r
test_usageL s( z TestSnippetUsageView.test_usagec C sr | j dddd}tjjddd}|j| | jddd | jtd d
gd }| |j
d | |td
d S )N
basicadminbasicadmin@example.compasswordusernameemailrY wagtailadmin access_adminZcontent_type__app_labelcodenamer[ rY rG rH r9 i. Zwagtailadmin_home) create_userr r! r" user_permissionsaddr5 r> r r# status_codeZassertRedirects)r& r6 admin_permissionr@ r r r -test_usage_without_edit_permission_on_snippete s& zBTestSnippetUsageView.test_usage_without_edit_permission_on_snippetc C s t jjdd}| | jdddd}tjjddd }tjjd
d d }|j| |j| | jddd | j t
d
dgd}| |j d |
|d | |d | j|ddd | j|ddd | j
|ddd | j|ddd d S )Nr r rW rX rY rZ r] r^ r_ ZtestsZ
change_advertra rG rH r9 rI z(Private page)
Page | TrK rN rO zAdvert)r r! r" rP rb r rc rd r5 r> r r# re rS rR )r& rT r6 rf Zadvert_permissionr@ r r r *test_usage_without_edit_permission_on_pagey s> z?TestSnippetUsageView.test_usage_without_edit_permission_on_pagec C s t jjdd}| | jtddgdd }| |d | j|dd d
| j|d d d
| j|d d d
| |d
| |d d S )Nr r rG rH r9 ?describe_on_delete=1rI ri TrK rN rO r
z+: the advert placement will also be deleted)r r! r" rP r> r rR rS r& rT r@ r r r *test_usage_with_describe_on_delete_cascade s z?TestSnippetUsageView.test_usage_with_describe_on_delete_cascadec C s t j }| | |j d | j t
d|jj gdd }| ||j
| j |ddd | j|ddd | j |d dd | |d
| |d d S ) Nr zwagtailimages:image_usager9 rk zEvent page | TrK rN rO z
Feed image: will unset the reference)r r! firstrP r# Z
feed_image get_usager% r> r" r idrR rD rS rl r r r +test_usage_with_describe_on_delete_set_null s
z@TestSnippetUsageView.test_usage_with_describe_on_delete_set_nullc C s t jjdd}td|d}tjjddj|d | t| d | j
t ddgdd }| |d
|
|d | j
|d d d
| j |dd d
| j
|dd d
|
|d |
|d d S )Nr r rB rC rE rG rH r9 rk rI rJ TrK rN rO rM rn )r
r! r" r
r rQ r# r r$ r% r> r rS rR )r& r' rU r@ r r r &test_usage_with_describe_on_delete_gfk s& z;TestSnippetUsageView.test_usage_with_describe_on_delete_gfkN) r) r* r+ r, r7 rA rV rg rj rm rr rs r r r r r4 4 s !r4 N)ior Zdjango.contrib.admin.utilsr Zdjango.contrib.auth.modelsr Z django.corer Z django.testr Z django.urlsr Zwagtail.modelsr r Zwagtail.test.testapp.modelsr
r r r
Zwagtail.test.utilsr r r/ r4 r r r r s