o
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 . e Zd ZdgZe fddZdd Z ZS )TestUsageCount test.jsonc " t t }tjd|d d S Nrebuild_references_index)stdoutsuper
setUpTestDatar r call_commandclsoutput __class__ a/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/snippets/tests/test_usage.pyr
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
r c r )
TestUsedByr c r r r r r r r r $ r! 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) assertIsInstancetupler list)r+ r, usager r r test_snippet_used_by* s
z TestUsedBy.test_snippet_used_by)r. r/ r0 r1 r2 r r: r3 r r r r r5 ! r4 r5 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& create
save_revisionpublishr? clientr' r r r$ get_soupcontent
select_oner( get_text)r+ snippetresponsesoupsublabelr r r test_use_latest_draft_as_title: s
z3TestSnippetUsageView.test_use_latest_draft_as_titlec C s | j dd$ tjjdd}| tdtjjddd}|j|d W d n1 s-w Y | jt d d
gd }|
|d |
|d | j
|d
dd |
|d | j
|ddd | j |ddd |
|d |
|t d|j gd d d S )NTexecuter% r# Foobar Titler" titlesnippet_content_objectinstance"wagtailsnippets_tests_advert:usage1r@ !Welcome to the Wagtail test site! | Generic snippet page | htmlSnippet content objectField | If you confirm deletion | zwagtailadmin_pages:editz&#:w:contentpath=snippet_content_object)
captureOnCommitCallbacksr r& r' saver
r
add_childrF r assertContainsassertNotContainsid)r+ pagegfk_pagerL r r r
test_usageL s8 z TestSnippetUsageView.test_usagec C sr | j dddd}tjjddd}|j| | jddd | jtd d
gd }| |j
d | |td
d S )N
basicadminbasicadmin@example.compasswordusernameemailrl wagtailadmin access_admincontent_type__app_labelcodenamern rl rX rY r@ i. wagtailadmin_home) create_userr r& r' user_permissionsaddr< rF r r( status_codeassertRedirects)r+ r= admin_permissionrL r r r -test_usage_without_edit_permission_on_snippetk s zBTestSnippetUsageView.test_usage_without_edit_permission_on_snippetc C s | j dd tjjdd}| W d n1 s w Y | jdddd}tjjd d
d }tjjd d
d }|j| |j| | j ddd | j
t ddgd}| |j
d | |d | |d | j|ddd | j|ddd | j|ddd | j|ddd d S )NTrP r% r# rj rk rl rm rp rq rr tests
change_advertru rX rY r@ rZ z(Private page)
Page | r\ r_ r` zAdvert)ra r r& r' rb rw r rx ry r< rF r r( rz re rd )r+ rg r= r| advert_permissionrL r r r *test_usage_without_edit_permission_on_page s:
z?TestSnippetUsageView.test_usage_without_edit_permission_on_pagec C s | j dd tjjdd}| W d n1 s w Y | jtddgdd }| |d | j|d
dd | j|d dd | j|d
dd | |d | |d d S )NTrP r% r# rX rY r@ ?describe_on_delete=1rZ r r\ r_ r` r
z+: the advert placement will also be deleted) ra r r& r' rb rF r rd re r+ rg rL r r r *test_usage_with_describe_on_delete_cascade s
z?TestSnippetUsageView.test_usage_with_describe_on_delete_cascadec C s | j dd tj }| W d n1 sw Y | |j d | j
t d|jj gdd }|
||j | j
|ddd | j|d dd | j
|d
dd |
|d |
|d d S )
NTrP r" zwagtailimages:image_usager@ r zEvent page | r\ r_ r` z
Feed image: will unset the reference)ra r r& firstrb r(
feed_image get_usager* rF r' r rf rd rT re r r r r +test_usage_with_describe_on_delete_set_null s
z@TestSnippetUsageView.test_usage_with_describe_on_delete_set_nullc C s | j dd tjjdd}td|d}tjjddj|d W d n1 s)w Y | t |
d | j t dd gd
d }|
|d | |d | j|d
dd | j
|ddd | j|ddd | |d | |d d S )NTrP r" r# rR rS rV rX rY r@ r rZ r[ r\ r_ r` r^ r )ra r
r& r' r
r rc r( r r) r* rF r re rd )r+ r, rh rL r r r &test_usage_with_describe_on_delete_gfk s* z;TestSnippetUsageView.test_usage_with_describe_on_delete_gfkN) r. r/ r0 r1 r> rO ri r} r r r r r r r r r; 4 s " r; N)ior django.contrib.admin.utilsr django.contrib.auth.modelsr django.corer django.testr django.urlsr wagtail.modelsr r wagtail.test.testapp.modelsr
r r r
wagtail.test.utilsr r r5 r; r r r r s