test_files.py
136 lines
| 4.7 KiB
| text/x-python
|
PythonLexer
|
r13175 | # coding: utf-8 | ||
"""Test the /files/ handler.""" | ||||
import io | ||||
import os | ||||
from unicodedata import normalize | ||||
pjoin = os.path.join | ||||
import requests | ||||
|
r18152 | import json | ||
|
r18163 | import binascii | ||
|
r18152 | |||
from IPython.nbformat.current import (new_notebook, write, new_worksheet, | ||||
new_heading_cell, new_code_cell, | ||||
new_output) | ||||
|
r13175 | |||
from IPython.html.utils import url_path_join | ||||
from .launchnotebook import NotebookTestBase | ||||
from IPython.utils import py3compat | ||||
|
r18152 | |||
|
r13175 | class FilesTest(NotebookTestBase): | ||
def test_hidden_files(self): | ||||
not_hidden = [ | ||||
u'Ã¥ b', | ||||
|
r13564 | u'å b/ç. d', | ||
|
r13175 | ] | ||
hidden = [ | ||||
u'.Ã¥ b', | ||||
|
r13564 | u'å b/.ç d', | ||
|
r13175 | ] | ||
dirs = not_hidden + hidden | ||||
nbdir = self.notebook_dir.name | ||||
for d in dirs: | ||||
|
r13182 | path = pjoin(nbdir, d.replace('/', os.sep)) | ||
|
r13175 | if not os.path.exists(path): | ||
os.mkdir(path) | ||||
|
r13178 | with open(pjoin(path, 'foo'), 'w') as f: | ||
f.write('foo') | ||||
with open(pjoin(path, '.foo'), 'w') as f: | ||||
f.write('.foo') | ||||
|
r13175 | url = self.base_url() | ||
for d in not_hidden: | ||||
|
r13182 | path = pjoin(nbdir, d.replace('/', os.sep)) | ||
|
r13175 | r = requests.get(url_path_join(url, 'files', d, 'foo')) | ||
r.raise_for_status() | ||||
|
r13565 | self.assertEqual(r.text, 'foo') | ||
|
r13175 | r = requests.get(url_path_join(url, 'files', d, '.foo')) | ||
|
r15103 | self.assertEqual(r.status_code, 404) | ||
|
r13175 | |||
for d in hidden: | ||||
|
r13182 | path = pjoin(nbdir, d.replace('/', os.sep)) | ||
|
r13175 | for foo in ('foo', '.foo'): | ||
r = requests.get(url_path_join(url, 'files', d, foo)) | ||||
|
r15103 | self.assertEqual(r.status_code, 404) | ||
|
r13564 | |||
|
r18152 | def test_contents_manager(self): | ||
"make sure ContentsManager returns right files (ipynb, bin, txt)." | ||||
nbdir = self.notebook_dir.name | ||||
base = self.base_url() | ||||
nb = new_notebook(name='testnb') | ||||
ws = new_worksheet() | ||||
nb.worksheets = [ws] | ||||
ws.cells.append(new_heading_cell(u'Created by test ³')) | ||||
cc1 = new_code_cell(input=u'print(2*6)') | ||||
cc1.outputs.append(new_output(output_text=u'12', output_type='stream')) | ||||
ws.cells.append(cc1) | ||||
with io.open(pjoin(nbdir, 'testnb.ipynb'), 'w', | ||||
encoding='utf-8') as f: | ||||
write(nb, f, format='ipynb') | ||||
with io.open(pjoin(nbdir, 'test.bin'), 'wb') as f: | ||||
|
r18161 | f.write(b'\xff' + os.urandom(5)) | ||
|
r18152 | f.close() | ||
with io.open(pjoin(nbdir, 'test.txt'), 'w') as f: | ||||
f.write(u'foobar') | ||||
f.close() | ||||
r = requests.get(url_path_join(base, 'files', 'testnb.ipynb')) | ||||
self.assertEqual(r.status_code, 200) | ||||
self.assertIn('print(2*6)', r.text) | ||||
json.loads(r.text) | ||||
r = requests.get(url_path_join(base, 'files', 'test.bin')) | ||||
self.assertEqual(r.status_code, 200) | ||||
self.assertEqual(r.headers['content-type'], 'application/octet-stream') | ||||
|
r18165 | self.assertTrue(r.content[0] == 255 or r.content[0] == b'\xff') | ||
|
r18161 | self.assertEqual(len(r.content), 6) | ||
|
r18152 | |||
r = requests.get(url_path_join(base, 'files', 'test.txt')) | ||||
self.assertEqual(r.status_code, 200) | ||||
self.assertEqual(r.headers['content-type'], 'text/plain') | ||||
self.assertEqual(r.text, 'foobar') | ||||
|
r13564 | def test_old_files_redirect(self): | ||
"""pre-2.0 'files/' prefixed links are properly redirected""" | ||||
nbdir = self.notebook_dir.name | ||||
base = self.base_url() | ||||
os.mkdir(pjoin(nbdir, 'files')) | ||||
os.makedirs(pjoin(nbdir, 'sub', 'files')) | ||||
for prefix in ('', 'sub'): | ||||
with open(pjoin(nbdir, prefix, 'files', 'f1.txt'), 'w') as f: | ||||
f.write(prefix + '/files/f1') | ||||
with open(pjoin(nbdir, prefix, 'files', 'f2.txt'), 'w') as f: | ||||
f.write(prefix + '/files/f2') | ||||
with open(pjoin(nbdir, prefix, 'f2.txt'), 'w') as f: | ||||
f.write(prefix + '/f2') | ||||
with open(pjoin(nbdir, prefix, 'f3.txt'), 'w') as f: | ||||
f.write(prefix + '/f3') | ||||
url = url_path_join(base, 'notebooks', prefix, 'files', 'f1.txt') | ||||
r = requests.get(url) | ||||
self.assertEqual(r.status_code, 200) | ||||
|
r13565 | self.assertEqual(r.text, prefix + '/files/f1') | ||
|
r13564 | |||
url = url_path_join(base, 'notebooks', prefix, 'files', 'f2.txt') | ||||
r = requests.get(url) | ||||
self.assertEqual(r.status_code, 200) | ||||
|
r13565 | self.assertEqual(r.text, prefix + '/files/f2') | ||
|
r13564 | |||
url = url_path_join(base, 'notebooks', prefix, 'files', 'f3.txt') | ||||
r = requests.get(url) | ||||
self.assertEqual(r.status_code, 200) | ||||
|
r13565 | self.assertEqual(r.text, prefix + '/f3') | ||
|
r13564 | |||