o
    i                     @   s8   d Z ddlZddlmZ ddlmZ G dd deZdS )zTests for login redirects    N)
url_concat)NotebookTestBasec                   @   s$   e Zd Zdd Zdd Zdd ZdS )	LoginTestc                 C   sd   t |  d }|  t jt|  d d|id| j|jddd|jd}|  |jd S )	NloginnextF_xsrf )passwordr   )allow_redirectsdatacookiesLocation)	requestsgetbase_urlraise_for_statuspostr   tokenr   headers)selfr   firstresp r   Y/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/auth/tests/test_login.pyr   
   s   

zLoginTest.loginc                 C   sF   dd| j  d ddddfD ]}| j|d}| || j  q|s!J d S )	Nz//some-hostz//hosttreezhttps://google.comz/absolute/not/base_urlz///jupyter.orgz/\some-hostr   )
url_prefixr   assertEqual)r   bad_nexturlr   r   r   test_next_bad   s   zLoginTest.test_next_badc                 C   sN   d|   d ddfD ]}d|v r|}n| j| }| j|d}| || q
d S )Nztree/zhas/hostznotebooks/notebook.ipynbztree//somethingz://r   )r   r   r   r   )r   	next_pathexpectedactualr   r   r   test_next_ok)   s   

zLoginTest.test_next_okN)__name__
__module____qualname__r   r    r$   r   r   r   r   r   	   s    r   )__doc__r   tornado.httputilr   notebook.tests.launchnotebookr   r   r   r   r   r   <module>   s
    