U
    {®Þh»  ã                   @   s‚   d dl 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ƒZG d	d
„ d
e
eƒZG dd„ de
eƒZdZdZdS )é    N)Úreverse)ÚTestDocumentDetailÚTestDocumentListingÚTestDocumentListingSearch)ÚDocumenté   )ÚAdminAPITestCasec                   @   s2   e Zd ZdgZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )ÚTestAdminDocumentListingúdemosite.jsonc                 K   s   | j  tdƒ|¡S ©Nz"wagtailadmin_api:documents:listing©ÚclientÚgetr   ©ÚselfÚparams© r   ún/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/api/test_documents.pyÚget_response   s    z%TestAdminDocumentListing.get_responsec                 C   s   dd„ |d D ƒS )Nc                 S   s   g | ]}|d  ‘qS ©Úidr   ©Ú.0Údocumentr   r   r   Ú
<listcomp>   s     zATestAdminDocumentListing.get_document_id_list.<locals>.<listcomp>Úitemsr   ©r   Úcontentr   r   r   Úget_document_id_list   s    z-TestAdminDocumentListing.get_document_id_listc                 C   sL  |   ¡ }|  |jd¡ |  |d d¡ t |j d¡¡}|  d|¡ |  |d t	¡ |  d|d ¡ |  |d d t
¡ |  |d d tj ¡ ¡ |  d|¡ |  |d t¡ |d D ]}|  d|¡ |  |d t	¡ |  t|d  ¡ ƒdd	d
dh¡ |  |d d d¡ |  |d d	 d|d  ¡ |  |d d
  d|d  ¡¡ q¶d S )NéÈ   úContent-typeúapplication/jsonúUTF-8ÚmetaZtotal_countr   ÚtypeÚ
detail_urlÚdownload_urlÚtagsúwagtaildocs.Documentz-http://localhost/admin/api/main/documents/%d/r   zhttp://localhost/documents/%d/)r   ÚassertEqualÚstatus_codeÚjsonÚloadsr   ÚdecodeÚassertInÚassertIsInstanceÚdictÚintr   ÚobjectsÚcountÚlistÚsetÚkeysZ
assertTrueÚ
startswith©r   Úresponser   r   r   r   r   Ú
test_basic   s8    
þ

þ
ÿÿz#TestAdminDocumentListing.test_basicc                 C   sf   |   ¡ }t |j d¡¡}|d D ]>}|  t| ¡ ƒdddh¡ |  t|d  ¡ ƒdddd	h¡ q"d S )
Nr"   r   r   r#   Útitler$   r%   r&   r'   )r   r+   r,   r   r-   r)   r5   r6   r8   r   r   r   Útest_fields_defaultK   s    
þz,TestAdminDocumentListing.test_fields_defaultN)Ú__name__Ú
__module__Ú__qualname__Úfixturesr   r   r:   r<   r   r   r   r   r	      s
   1r	   c                   @   s"   e Zd ZdgZdd„ Zdd„ ZdS )ÚTestAdminDocumentListingSearchr
   c                 K   s   | j  tdƒ|¡S r   r   r   r   r   r   r   Z   s    z+TestAdminDocumentListingSearch.get_responsec                 C   s   dd„ |d D ƒS )Nc                 S   s   g | ]}|d  ‘qS r   r   r   r   r   r   r   ^   s     zGTestAdminDocumentListingSearch.get_document_id_list.<locals>.<listcomp>r   r   r   r   r   r   r   ]   s    z3TestAdminDocumentListingSearch.get_document_id_listN)r=   r>   r?   r@   r   r   r   r   r   r   rA   W   s   rA   c                   @   s"   e Zd ZdgZdd„ Zdd„ ZdS )ÚTestAdminDocumentDetailr
   c                 K   s   | j  td|fd|¡S )Nz!wagtailadmin_api:documents:detail)Úargsr   )r   Zimage_idr   r   r   r   r   d   s     ÿz$TestAdminDocumentDetail.get_responsec                 C   s"  |   d¡}|  |jd¡ |  |d d¡ t |j d¡¡}|  d|¡ |  |d d¡ |  d|¡ |  |d t	¡ |  d|d ¡ |  |d d d	¡ |  d
|d ¡ |  |d d
 d¡ |  d|d ¡ |  |d d d¡ |  d|¡ |  |d d¡ |  d|d ¡ |  |d d g ¡ d S )Nr   r   r    r!   r"   r   r#   r$   r(   r%   z,http://localhost/admin/api/main/documents/1/r&   z6http://localhost/documents/1/wagtail_by_markyharky.jpgr;   zWagtail by mark Harkinr'   )
r   r)   r*   r+   r,   r   r-   r.   r/   r0   )r   r9   r   r   r   r   r:   i   s0    

þ
þz"TestAdminDocumentDetail.test_basicN)r=   r>   r?   r@   r   r:   r   r   r   r   rB   a   s   rB   )r+   Zdjango.urlsr   Z#wagtail.api.v2.tests.test_documentsr   r   r   Zwagtail.documents.modelsr   Úutilsr   r	   rA   rB   r   r   r   r   Ú<module>   s   H
5