a
    Dg2                     @   sf   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
mZ d dlmZ G dd deeZd	S )
    )AnonymousUser)engines)TestCase)get_dummy_request)PAGE_TEMPLATE_VARPageSite)WagtailTestUtilsc                   @   s8   e Zd Zdd ZdddZdddZd	d
 Zdd ZdS )TestCoreJinjac                 C   s0   t d | _| jdddd| _tjjdd| _d S )NZjinja2testztest@email.compassword)usernameemailr      )id)r   engineZcreate_superuseruserr   objectsgethomepage)self r   ^/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/site-packages/wagtail/admin/tests/test_jinja2.pysetUp   s
    
zTestCoreJinja.setUpNTc                 C   s"   |d u ri }| j |}||S )N)r   Zfrom_stringrender)r   stringcontextrequest_contexttemplater   r   r   r      s    zTestCoreJinja.renderc                 C   s(   t jjdd}t|d}|p t |_|S )NT)Zis_default_site)site)r   r   r   r   r   r   )r   r   r   requestr   r   r   dummy_request   s    
zTestCoreJinja.dummy_requestc              	   C   s.   |  dt| jd| | ji}| d| d S )N{{ wagtailuserbar() }}r    z$<!-- Wagtail user bar embed code -->)r   r   r   r!   r   ZassertInr   contentr   r   r   test_userbar    s    zTestCoreJinja.test_userbarc                 C   s*   |  dt| jd|  i}| |d d S )Nr"   r     )r   r   r   r!   assertEqualr#   r   r   r   test_userbar_anonymous_user*   s
    z)TestCoreJinja.test_userbar_anonymous_user)NT)N)__name__
__module____qualname__r   r   r!   r%   r(   r   r   r   r   r
   
   s
   


r
   N)Zdjango.contrib.auth.modelsr   Zdjango.templater   Zdjango.testr   Zwagtail.coreutilsr   Zwagtail.modelsr   r   r   Zwagtail.test.utilsr	   r
   r   r   r   r   <module>   s   