
    Ib                     >    d Z ddlZddlmZ ddlmZ  G d de      Zy)zTests for login redirects    N)
url_concat)NotebookTestBasec                       e Zd Zd Zd Zd Zy)	LoginTestc           	      z   t        j                  | j                         dz         }|j                          t        j                  t        | j                         dz   d|i      d| j                  |j                  j                  dd      d|j                        }|j                          |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resps       @/usr/lib/python3/dist-packages/notebook/auth/tests/test_login.pyr   zLoginTest.login
   s    T]]_w67 }}') " JJ**7B7 MM
 	||J''    c                     dd| j                   z   dz   ddfD ]0  }| j                  |      }| j                  || j                          2 sJ y )Nz//some-hostz//hosttreezhttps://google.comz/absolute/not/base_urlr	   
url_prefixr   assertEqual)r   bad_nexturls      r   test_next_badzLoginTest.test_next_bad   s\    t&/ $	
 	3H **(*+CS$//2	3 
sr   c                     dd| j                   z   dz   ddfD ]5  }| j                   |z   }| j                  |      }| j                  ||       7 y )Nztree/z//r   znotebooks/notebook.ipynbztree//somethingr   r    )r   	next_pathexpectedactuals       r   test_next_okzLoginTest.test_next_ok'   s]    4??"V+&	
 	/I 2HZZXZ.FVX.	/r   N)__name__
__module____qualname__r   r%   r*    r   r   r   r   	   s    ($		/r   r   )__doc__r   tornado.httputilr   notebook.tests.launchnotebookr   r   r.   r   r   <module>r2      s      ' :'/  '/r   