U
    {h                     @   sZ   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
 G dd de
eZdS )	    N)TestCase)widgets)Advert)AdvertChooserWidget)WagtailTestUtilsc                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
%TestChooserViewSetWithFilteredObjectsc                 C   s8   |   | _tjjdd tjjddd}|jd d S )N'Head On, apply directly to the forehead)texthttps://quiznos.comWe like the subs)urlr	   Zanimated)loginuserr   objectscreatetagsadd)selfZadvert2 r   y/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/viewsets/test_chooser_viewset.pysetUp   s    
 z+TestChooserViewSetWithFilteredObjects.setUpc                 C   s8   | j d}t|jd }| d| | d| d S )N/admin/animated_advert_chooser/htmlr   r   clientgetjsonloadscontentassertInZassertNotInr   responseZresponse_htmlr   r   r   test_get   s    z.TestChooserViewSetWithFilteredObjects.test_getc                 C   s`   | j dddi}t|jd }| d| | j dddi}t|jd }| d| d S )Nr   r   r
   r   r   zhttps://subway.comr   r    r   r   r   test_filter_by_url   s      z8TestChooserViewSetWithFilteredObjects.test_filter_by_urlc                 C   sj   t ddid}t |}| d|d  | d|d  | |d d | |d	 d
ddid d S )Nr   z#id_cool_url)Zlinked_fieldsz3<input id="__ID__" name="__NAME__" type="hidden" />r   ZChoose   Z__ID__   r   )ZmodalUrlZlinkedFields)r   r   ZBaseChooserAdapterjs_argsZassertInHTMLr   ZassertEqual)r   Zwidgetr&   r   r   r   $test_adapt_widget_with_linked_fields)   s     zJTestChooserViewSetWithFilteredObjects.test_adapt_widget_with_linked_fieldsN)__name__
__module____qualname__r   r"   r#   r'   r   r   r   r   r      s   
r   )r   Zdjango.testr   Zwagtail.adminr   Zwagtail.test.testapp.modelsr   Zwagtail.test.testapp.viewsr   Z wagtail.test.utils.wagtail_testsr   r   r   r   r   r   <module>   s   