o
    h#                     @   s   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ d dlmZmZmZ d dlmZ G d	d
 d
eeZdS )    )Group
Permission)ContentType)TestCase)reverse)get_document_model)DocumentsSummaryItem)
CollectionGroupCollectionPermissionSite)WagtailTestUtilsc                   @   sP   e Zd Zedd Zdd Zdd Zdd Zd	d
 Zdd Z	dd Z
dd ZdS )TestDocumentsSummaryc           	      C   sZ  t  }tj|}tjj|dd}tjj|dd}tjj|dd}t | _| jj	dd| _
tjjdd}tjj|| j|d tjjd	d}tjj|| j
|d tjjd
d}tjj|| j
|d | ddd| _| ddd| _| jj| | ddd| _| jj| |jjd| j| jd| _|jjd| j
| jd| _|jjd| j
| jd| _d S )Nadd_document)content_typecodenamechange_documentchoose_documentBirds)namezDocument changers)group
collection
permissionzBird adderszBird choosers	superuserzsuperuser@example.compasswordreportadderzreportadder@example.comreportchooserzreportchooser@example.comzreportadder's doc)titler   uploaded_by_userzreportadder's report)r   r   objectsget_for_modelr   getr	   get_first_root_noderoot_collection	add_childreports_collectionr   creater
   create_superuserr   create_userreport_addergroupsaddreport_chooserchanger_docchanger_reportadder_report)	selfDocumentdoc_content_typeadd_doc_permissionchange_doc_permissionchoose_doc_permissiondoc_changers_groupreport_adders_groupreport_choosers_group r8   i/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/documents/tests/test_site_summary.pysetUpTestData   sr   
z"TestDocumentsSummary.setUpTestDatac                 C   s   |  | j d S N)loginr   r/   r8   r8   r9   setUp_   s   zTestDocumentsSummary.setUpc                 C   s   | j tdjS )Nwagtailadmin_home)clientr    r   wsgi_requestr=   r8   r8   r9   get_requestb   s   z TestDocumentsSummary.get_requestc                 C   s    t |   }| || d S r;   )r   rB   render_htmlassertIn)r/   contentsummaryr8   r8   r9   assertSummaryContainse   s   z*TestDocumentsSummary.assertSummaryContainsc                 C   s,   |  tj d tj }| |j d S )N   )assertEqualr   r   countfirstrG   	site_name)r/   siter8   r8   r9   test_site_name_is_showni   s   
z,TestDocumentsSummary.test_site_name_is_shownc                 C   s   |  t|    d S r;   )
assertTruer   rB   is_shownr=   r8   r8   r9   )test_user_with_permissions_is_shown_paneln   s   z>TestDocumentsSummary.test_user_with_permissions_is_shown_panelc                 C   sF   d| j _| j jtjjddd | j   | t	| 
   d S )NFwagtailadminaccess_admin)content_type__app_labelr   )r   is_superuseruser_permissionsr*   r   r   r    saveassertFalser   rB   rP   r=   r8   r8   r9   0test_user_with_no_permissions_is_not_shown_panelq   s   
zETestDocumentsSummary.test_user_with_no_permissions_is_not_shown_panelc              	   C   sn   | j df| jdf| jdff}|D ]$\}}| j|d | | | | W d    n1 s/w   Y  qd S )Nz3 Documentsz2 Documents)user)r   r(   r+   subTestr<   rG   )r/   casesrZ   rE   r8   r8   r9   test_user_sees_proper_doc_count{   s   
z4TestDocumentsSummary.test_user_sees_proper_doc_countN)__name__
__module____qualname__classmethodr:   r>   rB   rG   rN   rQ   rY   r]   r8   r8   r8   r9   r      s    
Q
r   N)django.contrib.auth.modelsr   r   "django.contrib.contenttypes.modelsr   django.testr   django.urlsr   wagtail.documentsr   wagtail.documents.wagtail_hooksr   wagtail.modelsr	   r
   r   wagtail.test.utilsr   r   r8   r8   r8   r9   <module>   s    