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)Image)ImagesSummaryItem)
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 )TestImagesSummaryc                 C   s`  t jt}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| tjjd| j| jddd| _tjjd| j
| jddd| _tjjd| j
| jddd| _d S )N	add_image)content_typecodenamechange_imagechoose_imageBirds)namezImage changers)group
collection
permissionzBird adderszBird choosers	superuserzsuperuser@example.compassword	birdadderzbirdadder@example.combirdchooserzbirdchooser@example.comzbirdadder's image   )titler   uploaded_by_userwidthheightzbirdadder's bird      )r   objectsget_for_modelr   r   getr	   get_first_root_noderoot_collection	add_childbirds_collectionr   creater
   create_superuserr   create_user
bird_addergroupsaddbird_chooserchanger_imagechanger_bird
adder_bird)selfimage_content_typeadd_image_permissionchange_image_permissionchoose_image_permissionimage_changers_groupbird_adders_groupbird_choosers_group r<   f/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/images/tests/test_site_summary.pysetUpTestData   s|   
		zTestImagesSummary.setUpTestDatac                 C   s   |  | j d S N)loginr   r4   r<   r<   r=   setUpc   s   zTestImagesSummary.setUpc                 C   s   | j tdjS )Nwagtailadmin_home)clientr%   r   wsgi_requestrA   r<   r<   r=   get_requestf   s   zTestImagesSummary.get_requestc                 C   s    t |   }| || d S r?   )r   rF   render_htmlassertIn)r4   contentsummaryr<   r<   r=   assertSummaryContainsi   s   z'TestImagesSummary.assertSummaryContainsc                 C   s,   |  tj d tj }| |j d S )Nr   )assertEqualr   r#   countfirstrK   	site_name)r4   siter<   r<   r=   test_site_name_is_shownm   s   
z)TestImagesSummary.test_site_name_is_shownc                 C   s   |  t|    d S r?   )
assertTruer   rF   is_shownrA   r<   r<   r=   )test_user_with_permissions_is_shown_panelr   s   z;TestImagesSummary.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   rF   rS   rA   r<   r<   r=   0test_user_with_no_permissions_is_not_shown_panelu   s   
zBTestImagesSummary.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 Imagesz2 Images)user)r   r-   r0   subTestr@   rK   )r4   casesr]   rI   r<   r<   r=   !test_user_sees_proper_image_count   s   
z3TestImagesSummary.test_user_sees_proper_image_countN)__name__
__module____qualname__classmethodr>   rB   rF   rK   rQ   rT   r\   r`   r<   r<   r<   r=   r      s    
U
r   N)django.contrib.auth.modelsr   r   "django.contrib.contenttypes.modelsr   django.testr   django.urlsr   wagtail.images.tests.utilsr   wagtail.images.wagtail_hooksr   wagtail.modelsr	   r
   r   wagtail.test.utilsr   r   r<   r<   r<   r=   <module>   s    