U
    {®Þh1  ã                   @   sh   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 d dlmZ G dd„ deeeƒZd	S )
é    )ÚTestCase)Úreverse)ÚPage)ÚFormPageWithRedirectÚPageChooserModelÚ
SimplePage)ÚWagtailTestUtils)ÚAdminTemplateTestUtilsc                   @   s8   e Zd Zg Zdd„ Zdd„ Zdd„ Zdd„ Zd	d
„ ZdS )ÚTestPageUsagec                 C   sX   |   ¡ | _tjjdd| _tdddd}| jj|d | ¡  	¡  tjj|j
d| _d S )Né   )ÚidúHello world!zhello-worldZhello)ÚtitleÚslugÚcontent©Úinstance)ÚloginÚuserr   ÚobjectsÚgetÚ	root_pager   Ú	add_childZsave_revisionÚpublishr   Úpage)Úselfr   © r   úq/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/pages/test_page_usage.pyÚsetUp   s    
ýzTestPageUsage.setUpc           	      C   s  t d| jjfd}| j |¡}|  |jd¡ |  |d¡ |  |d¡ |  |d¡ t dƒdd	œt d
| j	jfddd	œt d
| jjfddd	œddddœg}|  
||j¡ t d| jjfd}|  |j¡}| d|› d¡}|  t|ƒd¡ |d }|j d¡}|  d|¡ |  d|¡ d S )Núwagtailadmin_pages:usage©ÚargséÈ   ú!wagtailadmin/generic/listing.htmlÚUsager   Zwagtailadmin_explore_rootZRoot)ÚurlÚlabelZwagtailadmin_explorez!Welcome to your new Wagtail site!zHello world! (simple page)Ú )r%   r&   Zsublabelúwagtailadmin_pages:editza[href='z']é   r   Úclasszw-header-buttonZbutton)r   r   r   Úclientr   ÚassertEqualÚstatus_codeÚassertTemplateUsedÚassertContainsr   ZassertBreadcrumbsItemsRenderedr   Zget_soupÚselectÚlenÚattrsZassertIn)	r   Ú	usage_urlÚresponseÚitemsZedit_urlZsoupZ
edit_linksZ	edit_linkÚclassesr   r   r   Útest_simple   s:    þþþýózTestPageUsage.test_simplec                 C   s€   t jj| jd td| jjfd}| j |¡}|  |j	d¡ |  
|d¡ |  |d¡ |  |d¡ |  |d¡ | j|d	d
d d S )N©r   r   r    r"   r#   r$   r   z(Private page chooser model)z<td>Page chooser model</td>T©Úhtml)r   r   Úcreater   r   r   r+   r   r,   r-   r.   r/   )r   r3   r4   r   r   r   Útest_has_private_usageG   s    z$TestPageUsage.test_has_private_usagec                 C   s¸   t ddddd| jd}| jj|d}td| jjfd	}| j |¡}|  |j	d
¡ |  
|d¡ |  |d¡ |  |d¡ |  |d¡ |  |td|jfd	¡ |  |d¡ | j|ddd d S )Nz
Contact usz
contact-uszto@email.comzfrom@email.comzThe subject)r   r   Z
to_addressZfrom_addressÚsubjectZthank_you_redirect_pager   r   r    r"   r#   r$   r   r(   zThank you redirect pagez <td>Form page with redirect</td>Tr9   )r   r   r   r   r   r   r+   r   r,   r-   r.   r/   )r   Z	form_pager3   r4   r   r   r   Útest_has_editable_usageT   s,    ú	 ÿz%TestPageUsage.test_has_editable_usagec                 C   s   t dƒD ]}tjj| jd qtd| jjfd}| j |› d¡}|  	|j
d¡ |  |d¡ |  |d¡ |  ||› d	¡ |  ||› d
¡ d S )Né2   r8   r   r    z?p=2r"   r#   zPage 2 of 3.z?p=1z?p=3)Úranger   r   r;   r   r   r   r+   r   r,   r-   r.   r/   )r   Ú_r3   r4   r   r   r   Útest_paginationo   s    zTestPageUsage.test_paginationN)	Ú__name__Ú
__module__Ú__qualname__Zbase_breadcrumb_itemsr   r7   r<   r>   rB   r   r   r   r   r
      s   (r
   N)Zdjango.testr   Zdjango.urlsr   Zwagtail.modelsr   Zwagtail.test.testapp.modelsr   r   r   Zwagtail.test.utilsr   Z!wagtail.test.utils.template_testsr	   r
   r   r   r   r   Ú<module>   s   