a
    Dg                     @   sr   d dl 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 edgd	G d
d deeZdS )    N)ContentFile)TestCase)override_settings)reverse)models)WagtailTestUtilselasticsearch)Z _WAGTAILSEARCH_FORCE_AUTO_UPDATEc                   @   s<   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd ZdS )TestIssue613c                 C   s4   ddl m} ddlm} d|jvr,td|dS )Nr   )settings)get_search_backendr   zNo elasticsearch backend active)Zdjango.confr
   Zwagtail.search.backendsr   ZWAGTAILSEARCH_BACKENDSunittestZSkipTest)selfr
   r    r   b/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/documents/tests/test_search.pyget_elasticsearch_backend   s
    

z&TestIssue613.get_elasticsearch_backendc                 C   s   |   | _|   d S )N)r   search_backendlogin)r   r   r   r   setUp   s    
zTestIssue613.setUpc                 K   sn   t d}d|_d|d}|| | jtd|}| |td tjj	j
|d d}| |  | S )	N   A boring example documenttest.txtTest documenttitlefilezwagtaildocs:addwagtaildocs:indexr   r   )r   nameupdateclientpostr   assertRedirectsr   Documentobjectsfilter
assertTrueexistsfirst)r   params	fake_file	post_dataresponsedocr   r   r   add_document   s    
zTestIssue613.add_documentc                 K   s   t d}d|_tjjjd|d}t d}d|_d|d}|| | jt	d|j
fd|}| |t	d tjjj|d	 d
}| |  | S )Nr   r   r   r   zTest document changed!zwagtaildocs:edit)argsr   r   r   )r   r   r   r!   r"   creater   r   r   r   idr    r#   r$   r%   r&   )r   r'   r(   documentZanother_fake_filer)   r*   r+   r   r   r   edit_document1   s     
zTestIssue613.edit_documentc                 C   sf   | j   | j tj | jdd}| j   | j dtj}| t	|d | |d j
|j
 d S NZhello)tags   r   )r   reset_indexadd_typer   r!   r,   refresh_indexsearchassertEquallenr/   r   r0   resultsr   r   r   test_issue_613_on_addO   s    

z"TestIssue613.test_issue_613_on_addc                 C   sf   | j   | j tj | jdd}| j   | j dtj}| t	|d | |d j
|j
 d S r2   )r   r5   r6   r   r!   r1   r7   r8   r9   r:   r/   r;   r   r   r   test_issue_613_on_edit_   s    

z#TestIssue613.test_issue_613_on_editN)	__name__
__module____qualname__r   r   r,   r1   r=   r>   r   r   r   r   r	      s   
r	   )r   Zdjango.core.files.baser   Zdjango.testr   Zdjango.test.utilsr   Zdjango.urlsr   Zwagtail.documentsr   Zwagtail.test.utilsr   r	   r   r   r   r   <module>   s   
