B
3¢\† ã @ s” d Z ddlZddlZddlmZ ejjZddlZddl Z ddl
m Z ddl m
Z
mZmZmZ ddlmZ ddlmZ dd lmZ G d
d „ d eƒZdS ) zTest the /files/ handler.é N)Ú normalize)Úwrite)Ú new_notebookÚnew_markdown_cellÚ
new_code_cellÚ
new_output)Ú
url_path_joiné )ÚNotebookTestBase)Ú py3compatc @ s4 e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
„ Zd S ) Ú FilesTestc
C s\ ddg}ddg}|| }| j }x‚|D ]z}t|| dtj¡ƒ}tj |¡sRt |¡ tt|dƒdƒ}| d¡ W d Q R X tt|dƒdƒ}| d¡ W d Q R X q$W xl|D ]d}t|| dtj¡ƒ}|
d t d
|dƒ¡}| ¡ |
|jd¡ |
d t d
|dƒ¡}|
|jd ¡ q¨W xT|D ]L}t|| dtj¡ƒ}x0d D ](} |
d t d
|| ƒ¡}|
|jd ¡ q4W qW d
| jj_zÚxv|D ]n}t|| dtj¡ƒ}|
d t d
|dƒ¡}| ¡ |
|jd¡ |
d t d
|dƒ¡}| ¡ |
|jd¡ qxW x\|D ]T}t|| dtj¡ƒ}x8d D ]0} |
d t d
|| ƒ¡}| ¡ |
|j| ¡ qW qðW W d d| jj_X d S )Nu Ã¥ bu
å b/ç. du .å bu
Ã¥ b/.ç dú/ÚfooÚwz.fooÚGETÚfilesi” )r z.fooTF)Ú notebook_dirÚpjoinÚreplaceÚosÚsepÚpathÚexistsÚmkdirÚopenr Úrequestr Zraise_for_statusÚ assertEqualÚtextÚ status_codeZnotebookZcontents_managerZ allow_hidden)
ÚselfZ
not_hiddenZhiddenÚdirsÚnbdirÚdr ÚfÚrr © r% ú8lib/python3.7/site-packages/notebook/tests/test_files.pyÚtest_hidden_files sV
zFilesTest.test_hidden_filesc C sŠ | j }ttdƒtdtdddgdgd}tjt|dƒd d
d }t||d d
W dQ R X t t|dƒd¡"}| dt
d¡ ¡ | ¡ W dQ R X t t|dƒd ¡}| d¡ | ¡ W dQ R X | dd¡}|
|jd¡ | d|j¡ t |j¡ | dd¡}|
|jd¡ |
|jd d¡ |
|jdd… d¡ |
t|jƒd ¡ | dd ¡}|
|jd¡ |
|jd d ¡ |
|jd¡ dS ) z@make sure ContentsManager returns right files (ipynb, bin, txt).u Created by test ³z
print(2*6)ÚstreamZ12)r )Zoutputs)Zcellsz testnb.ipynbr zutf-8)Úencodingé )ÚversionNztest.binÚwbó ÿé ztest.txtZfoobarr zfiles/testnb.ipynbéÈ zfiles/test.binz content-typezapplication/octet-streamr é zfiles/test.txtztext/plain; charset=UTF-8)r r r r r Úior r r r ÚurandomÚcloser r r ÚassertInr ÚjsonÚloadsÚheadersZcontentÚlen)r r! Znbr# r$ r% r% r& Útest_contents_managerO s6
z FilesTest.test_contents_managerc C sŒ | j }d}tt|dƒdƒ}| |¡ W d Q R X | dd¡}|j dd¡}| d|¡ | dd ¡}|j dd¡}| d|¡ | d
|¡ d S ) NZhelloztest.txtr r zfiles/test.txtzContent-DispositionÚ Z
attachmentzfiles/test.txt?download=1zfilename*=utf-8''test.txt) r r r r r r7 ÚgetZ assertNotInr4 )r r! r r# r$ Z dispositionr% r% r& Ú
test_downloady s zFilesTest.test_downloadc C sN | j }d}tt|dƒdƒ}| |¡ W d Q R X | dd¡}| |jd¡ d S )NzTest test
z test.htmlr r zview/test.htmlr/ )r r r r r r r )r r! Zhtmlr# r$ r% r% r& Útest_view_html‰ s zFilesTest.test_view_htmlc
C s | j }t t|dƒ¡ t t|ddƒ¡ xpdD ]f}tt||ddƒdƒ}| |d ¡ W dQ R X tt||ddƒdƒ}| |d ¡ W dQ R X tt||dƒdƒ}| |d
¡ W dQ R X tt||d ƒdƒ}| |d ¡ W dQ R X td
|ddƒ}| d|¡}| |j
d¡ | |j |d ¡ td
|ddƒ}| d|¡}| |j
d¡ | |j |d ¡ td
|dd ƒ}| d|¡}| |j
d¡ | |j |d ¡ q0W dS )z7pre-2.0 'files/' prefixed links are properly redirectedr Úsub)r: r> zf1.txtr z /files/f1Nzf2.txtz /files/f2z/f2zf3.txtz/f3Z notebooksr r/ ) r r r r Úmakedirsr r r r r r r )r r! Úprefixr# Zurlr$ r% r% r& Útest_old_files_redirect“ s0 z!FilesTest.test_old_files_redirectN)Ú__name__Ú
__module__Ú __qualname__r' r9 r< r= rA r% r% r% r& r s
7*
r )Ú__doc__r1 r Z unicodedatar r Újoinr Zrequestsr5 Znbformatr Z nbformat.v4r r r r Znotebook.utilsr Zlaunchnotebookr
Zipython_genutilsr r r% r% r% r& Ú s