U
{h@ @ 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)Image)ImagesSummaryItem)
CollectionGroupCollectionPermissionSite)WagtailTestUtilsc @ sP e Zd Zedd Zdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dS )TestImagesSummaryc C s` t jt}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| tjj d| j| jddd| _tjj d| j
| jddd| _tjj d| j
| jddd| _d S )NZ add_image) content_typecodenameZ change_imageZ choose_imageZBirds)namezImage changers)group
collectionZ
permissionz Bird addersz
Bird choosers superuserzsuperuser@example.compasswordZ birdadderzbirdadder@example.comZ birdchooserzbirdchooser@example.comzbirdadder's image )titler Zuploaded_by_userwidthheightzbirdadder's bird )r objectsZ
get_for_modelr r getr Zget_first_root_nodeZroot_collectionZ add_childZbirds_collectionr creater
Zcreate_superuserr Z create_user
bird_addergroupsadd bird_chooserZ
changer_imageZ changer_birdZ
adder_bird)selfZimage_content_typeZadd_image_permissionZchange_image_permissionZchoose_image_permissionZimage_changers_groupZbird_adders_groupZbird_choosers_group r# n/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/images/tests/test_site_summary.py
setUpTestData
s
z TestImagesSummary.setUpTestDatac C s | | j d S N)loginr r" r# r# r$ setUpc s zTestImagesSummary.setUpc C s | j tdjS )NZwagtailadmin_home)clientr r Z wsgi_requestr( r# r# r$ get_requestf s z TestImagesSummary.get_requestc C s t | }| || d S r& )r r+ Z render_htmlZassertIn)r" contentsummaryr# r# r$ assertSummaryContainsi s z'TestImagesSummary.assertSummaryContainsc C s, | tj d tj }| |j d S )Nr )Z assertEqualr r countfirstr. Z site_name)r" siter# r# r$ test_site_name_is_shownm s
z)TestImagesSummary.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_panelr s z;TestImagesSummary.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+ r3 r( r# r# r$ 0test_user_with_no_permissions_is_not_shown_panelu s
zBTestImagesSummary.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 Imagesz2 Images)user)r r r! ZsubTestr' r. )r" Zcasesr7 r, r# r# r$ !test_user_sees_proper_image_count s
z3TestImagesSummary.test_user_sees_proper_image_countN) __name__
__module__ __qualname__ classmethodr% r) r+ r. r2 r4 r6 r8 r# r# r# r$ r
s
U
r
N)Zdjango.contrib.auth.modelsr r Z"django.contrib.contenttypes.modelsr Z django.testr Z django.urlsr Zwagtail.images.tests.utilsr Z wagtail.images.wagtail_hooksr Zwagtail.modelsr r
r Zwagtail.test.utilsr r
r# r# r# r$ s