U
    |h;	                     @   s   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	m
Z
 G dd dZej D ].Zede  Zeeeefd	eie e< qXG d
d deZdS )    )settings)TestCase)Page)get_search_backend)BaseSearchQueryCompilerBaseSearchResultsc                   @   sB   e Zd ZdgZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dS )PageSearchTestsz	test.jsonc                 C   s<   t | j| _|   tj D ]}| j| q|   d S N)	r   backend_namebackendreset_indexr   objectsalladdrefresh_index)selfpage r   m/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/search/tests/test_page_search.pysetUp   s
    zPageSearchTests.setUpc                 C   s>   | j jr:| j t}| j |}| }|t |  d S r	   )r   Zrebuilder_classget_index_for_modelr   startZ	add_modelfinish)r   indexZ	rebuilderr   r   r   r      s    
zPageSearchTests.reset_indexc                 C   s   | j t}|r|  d S r	   )r   r   r   refresh)r   r   r   r   r   r      s    zPageSearchTests.refresh_indexc                 C   s"   t tjdjdd| jd d S )NtitleZblahF)Zorder_by_relevancer   )listr   r   Zorder_bysearchr
   r   r   r   r   test_order_by_title$   s      z#PageSearchTests.test_order_by_titlec                 C   s   t tj jd| jd d S )Nbread)r   r   r   r   Zspecificr   r
   r   r   r   r   test_search_specific_queryset+   s    z-PageSearchTests.test_search_specific_querysetc                 C   s"   t tj jddg| jd d S )Nr    r   )fieldsr   r!   r   r   r   r   )test_search_specific_queryset_with_fields.   s    
  z9PageSearchTests.test_search_specific_queryset_with_fieldsN)
__name__
__module____qualname__Zfixturesr   r   r   r   r"   r$   r   r   r   r   r   	   s   r   zTest%sBackendr
   c                   @   s   e Zd Zdd ZdS )TestBaseSearchResultsc                 C   s6   t dt}|dd }| |jd | |jd d S )NZBackendIrrelevantr   )r   r   ZassertEqualr   stop)r   Zbase_search_resultsobjr   r   r   test_get_item_no_resultsC   s     z.TestBaseSearchResults.test_get_item_no_resultsN)r%   r&   r'   r+   r   r   r   r   r(   B   s   r(   N)Zdjango.confr   Zdjango.testr   Zwagtail.modelsr   Zwagtail.search.backendsr   Zwagtail.search.backends.baser   r   r   ZWAGTAILSEARCH_BACKENDSkeysr
   strr   Z	test_nametypeglobalsr(   r   r   r   r   <module>   s   -
