U
    {h                     @   sF   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 )    )TestCase)reverse)timezone)WagtailTestUtilsc                   @   s4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )TestBaseViewSetc                 C   s   |   | _d S )N)loginuser)self r
   v/var/www/dating/data/www/fatepal.com/env/lib/python3.8/site-packages/wagtail/admin/tests/viewsets/test_base_viewset.pysetUp	   s    zTestBaseViewSet.setUpc                 C   sF   | j td}| |jd | |d | |d | |d d S )NZwagtailadmin_home   ZMiscellaneouszThe CalendarzThe Greetings)clientgetr   assertEqualstatus_codeassertContains)r	   responser
   r
   r   test_menu_items   s
    zTestBaseViewSet.test_menu_itemsc                 C   s@   t d}| j|}t }| |d | ||j d d S )Nzcalendar:indexz/admin/calendar/	 calendar)r   r   r   r   nowr   r   yearr	   urlr   r   r
   r
   r   test_calendar_index_view   s
    z(TestBaseViewSet.test_calendar_index_viewc                 C   sH   t d}| j|}t }| |d | ||j d|j d d S )Nzcalendar:monthz/admin/calendar/month//r   )	r   r   r   r   r   r   r   r   monthr   r
   r
   r   test_calendar_month_view   s
    z(TestBaseViewSet.test_calendar_month_viewc                 C   sV   d| j _d| j _| j   td}| j|}| |d | |d | |d d S )NZGordonZFreemanzgreetings:indexz/admin/greetingz/Z	Greetingsz/Welcome to this greetings page, Gordon Freeman!)	r   Z
first_name	last_namesaver   r   r   r   r   )r	   r   r   r
   r
   r   test_greetings_view!   s    
z#TestBaseViewSet.test_greetings_viewN)__name__
__module____qualname__r   r   r   r   r    r
   r
   r
   r   r      s
   r   N)	Zdjango.testr   Zdjango.urlsr   Zdjango.utilsr   Z wagtail.test.utils.wagtail_testsr   r   r
   r
   r
   r   <module>   s   