o
    ËÉ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   ó   | j  tdƒ|¡S ©Nz"wagtailadmin_api:documents:listing©ÚclientÚgetr   ©ÚselfÚparams© r   úf/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/admin/tests/api/test_documents.pyÚget_response   ó   z%TestAdminDocumentListing.get_responsec                 C   ó   dd„ |d D ƒS )Nc                 S   ó   g | ]}|d  ‘qS ©Úidr   ©Ú.0Údocumentr   r   r   Ú
<listcomp>   ó    zATestAdminDocumentListing.get_document_id_list.<locals>.<listcomp>Úitemsr   ©r   Úcontentr   r   r   Úget_document_id_list   r   z-TestAdminDocumentListing.get_document_id_listc                 C   sH  |   ¡ }|  |jd¡ |  |d d¡ t |j d¡¡}|  d|¡ |  |d t	¡ |  d|d ¡ |  |d d t
¡ |  |d d tj ¡ ¡ |  d|¡ |  |d t¡ |d D ]F}|  d|¡ |  |d t	¡ |  t|d  ¡ ƒh d£¡ |  |d d	 d
¡ |  |d d d|d  ¡ |  |d d  d|d  ¡¡ q[d S )NéÈ   úContent-typeúapplication/jsonúUTF-8ÚmetaÚtotal_countr    >   ÚtagsÚtypeÚ
detail_urlÚdownload_urlr+   úwagtaildocs.Documentr,   z-http://localhost/admin/api/main/documents/%d/r   r-   zhttp://localhost/documents/%d/)r   ÚassertEqualÚstatus_codeÚjsonÚloadsr"   ÚdecodeÚassertInÚassertIsInstanceÚdictÚintr   ÚobjectsÚcountÚlistÚsetÚkeysÚ
assertTrueÚ
startswith©r   Úresponser"   r   r   r   r   Ú
test_basic   s:   þ

þ
ÿÿîz#TestAdminDocumentListing.test_basicc                 C   s`   |   ¡ }t |j d¡¡}|d D ]}|  t| ¡ ƒh d£¡ |  t|d  ¡ ƒh d£¡ qd S )Nr'   r    >   r   r(   Útitler(   >   r*   r+   r,   r-   )r   r1   r2   r"   r3   r/   r;   r<   r?   r   r   r   Útest_fields_defaultK   s   þþz,TestAdminDocumentListing.test_fields_defaultN)Ú__name__Ú
__module__Ú__qualname__Úfixturesr   r#   rA   rC   r   r   r   r   r	      s    1r	   c                   @   ó"   e Zd ZdgZdd„ Zdd„ ZdS )ÚTestAdminDocumentListingSearchr
   c                 K   r   r   r   r   r   r   r   r   Z   r   z+TestAdminDocumentListingSearch.get_responsec                 C   r   )Nc                 S   r   r   r   r   r   r   r   r   ^   r   zGTestAdminDocumentListingSearch.get_document_id_list.<locals>.<listcomp>r    r   r!   r   r   r   r#   ]   r   z3TestAdminDocumentListingSearch.get_document_id_listN)rD   rE   rF   rG   r   r#   r   r   r   r   rI   W   s    rI   c                   @   rH   )ÚTestAdminDocumentDetailr
   c                 K   s   | j  td|fd|¡S )Nz!wagtailadmin_api:documents:detail)Úargsr   )r   Ú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.jpgrB   zWagtail by mark Harkinr*   )
r   r/   r0   r1   r2   r"   r3   r4   r5   r6   )r   r@   r"   r   r   r   rA   i   s0   

þ
þz"TestAdminDocumentDetail.test_basicN)rD   rE   rF   rG   r   rA   r   r   r   r   rJ   a   s    rJ   )r1   Údjango.urlsr   Ú#wagtail.api.v2.tests.test_documentsr   r   r   Úwagtail.documents.modelsr   Úutilsr   r	   rI   rJ   r   r   r   r   Ú<module>   s    H
5