U
{hJ @ s d dl mZmZ d dlmZ d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZmZmZ d dlmZ G d d
d
eeZd S ) )Group
Permission) ContentType)TestCase)reverse)get_document_model)DocumentsSummaryItem)
CollectionGroupCollectionPermissionSite)WagtailTestUtilsc @ sP e Zd Zedd Zdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dS )TestDocumentsSummaryc C sZ t }tj|}tjj|dd}tjj|dd}tjj|dd}t | _| jj dd| _
t jj dd}t
jj || j|d t jj d d}t
jj || j
|d t jj d
d}t
jj || j
|d | d d d
| _| ddd
| _| jj| | ddd
| _| jj| |jj d| j| jd| _|jj d| j
| jd| _|jj d| j
| jd| _d S )NZ add_document) content_typecodenameZchange_documentZchoose_documentZBirds)namezDocument changers)group
collectionZ
permissionz Bird addersz
Bird choosers superuserzsuperuser@example.compasswordZ reportadderzreportadder@example.comZ
reportchooserzreportchooser@example.comzreportadder's doc)titler Zuploaded_by_userzreportadder's report)r r objectsZ
get_for_modelr getr Zget_first_root_nodeZroot_collectionZ add_childZreports_collectionr creater
Zcreate_superuserr Z create_user report_addergroupsaddreport_chooserZ changer_docZchanger_reportZ adder_report) selfZDocumentZdoc_content_typeZadd_doc_permissionZchange_doc_permissionZchoose_doc_permissionZdoc_changers_groupZreport_adders_groupZreport_choosers_group r q/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/documents/tests/test_site_summary.py
setUpTestData
s
z"TestDocumentsSummary.setUpTestDatac C s | | j d S N)loginr r r r r setUp_ s zTestDocumentsSummary.setUpc C s | j tdjS )NZwagtailadmin_home)clientr r Z wsgi_requestr# r r r get_requestb s z TestDocumentsSummary.get_requestc C s t | }| || d S r! )r r& Z render_htmlZassertIn)r contentsummaryr r r assertSummaryContainse s z*TestDocumentsSummary.assertSummaryContainsc C s, | tj d tj }| |j d S )N )Z assertEqualr r countfirstr) Z site_name)r siter r r test_site_name_is_showni s
z,TestDocumentsSummary.test_site_name_is_shownc C s | t| d S r! )Z
assertTruer r& is_shownr# r r r )test_user_with_permissions_is_shown_paneln s z>TestDocumentsSummary.test_user_with_permissions_is_shown_panelc C sF d| j _| j jtjjddd | j | t |
d S )NFZ wagtailadminZ access_admin)Zcontent_type__app_labelr ) r Z is_superuserZuser_permissionsr r r r saveZ assertFalser r& r/ r# r r r 0test_user_with_no_permissions_is_not_shown_panelq s
zETestDocumentsSummary.test_user_with_no_permissions_is_not_shown_panelc
C sZ | j df| jdf| jdff}|D ]4\}}| j|d | | | | W 5 Q R X q d S )Nz3 Documentsz2 Documents)user)r r r ZsubTestr" r) )r Zcasesr3 r' r r r test_user_sees_proper_doc_count{ s
z4TestDocumentsSummary.test_user_sees_proper_doc_countN) __name__
__module__ __qualname__ classmethodr r$ r& r) r. r0 r2 r4 r r r r r
s
Q
r
N)Zdjango.contrib.auth.modelsr r Z"django.contrib.contenttypes.modelsr Z django.testr Z django.urlsr Zwagtail.documentsr Z wagtail.documents.wagtail_hooksr Zwagtail.modelsr r
r Zwagtail.test.utilsr r
r r r r s