
    Ib6                     b    d Z 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	e      Zy)
zTest the /tree handlers    N)url_path_join)write)new_notebook)urlparse)NotebookTestBasec                       e Zd Zd Zd Zy)TreeTestc                    | j                   }t        j                  j                  |d      }t        j                  |       t        t        j                  j                  |d      dd      5 }t               }t        ||d       d d d        t        t        j                  j                  |d      dd      5 }|j                  d	       d d d        | j                          y # 1 sw Y   `xY w# 1 sw Y   &xY w)
Nfooz	bar.ipynbwzutf-8)encoding   )versionzbaz.txtflamingo)	notebook_dirospathjoinmkdiropenr   r   base_url)selfnbdirdfnbs        G/usr/lib/python3/dist-packages/notebook/tree/tests/test_tree_handler.pysetUpzTreeTest.setUp   s    !!GGLL&
"'',,q+.gF 	$!B"a#	$ "'',,q),cGD 	 GGJ	  		$ 	$	  	 s   .C&;C2&C/2C;c           	      v   | j                  dd      }| j                  |j                  | j                         dz          | j                  ddd      }| j                  |j                  d       | j                  |j
                  d   t        t        | j                         d	            j                         y )
NGETztree/foo/bar.ipynbznotebooks/foo/bar.ipynbztree/foo/baz.txtF)allow_redirectsi.  Locationzfiles/foo/baz.txt)	requestassertEqualurlr   status_codeheadersr   r   r   )r   rs     r   test_redirectzTreeTest.test_redirect   s    LL 452K KLLL 2ELJ,:.mDMMO5HIJOO	Q    N)__name__
__module____qualname__r   r)    r*   r   r	   r	   
   s    Qr*   r	   )__doc__r   notebook.utilsr   nbformatr   nbformat.v4r   urllib.parser   notebook.tests.launchnotebookr   r	   r.   r*   r   <module>r5      s)     	 (  $ ! :Q Qr*   