Uploaded Test files

venv/Lib/site-packages/notebook/tree/tests/test_tree_handler.py

@@ -0,0 +1,35 @@
+"""Test the /tree handlers"""
+import os
+import io
+from notebook.utils import url_path_join
+from nbformat import write
+from nbformat.v4 import new_notebook
+from urllib.parse import urlparse
+
+import requests
+
+from notebook.tests.launchnotebook import NotebookTestBase
+
+class TreeTest(NotebookTestBase):
+    def setUp(self):
+        nbdir = self.notebook_dir
+        d = os.path.join(nbdir, 'foo')
+        os.mkdir(d)
+
+        with io.open(os.path.join(d, 'bar.ipynb'), 'w', encoding='utf-8') as f:
+            nb = new_notebook()
+            write(nb, f, version=4)
+
+        with io.open(os.path.join(d, 'baz.txt'), 'w', encoding='utf-8') as f:
+            f.write(u'flamingo')
+
+        self.base_url()
+
+    def test_redirect(self):
+        r = self.request('GET', 'tree/foo/bar.ipynb')
+        self.assertEqual(r.url, self.base_url() + 'notebooks/foo/bar.ipynb')
+
+        r = self.request('GET', 'tree/foo/baz.txt', allow_redirects=False)
+        self.assertEqual(r.status_code, 302)
+        self.assertEqual(r.headers['Location'],
+             urlparse(url_path_join(self.base_url(), 'files/foo/baz.txt')).path)