From d72d3f7e308e6e1e7b0da16325753147f247dbf6 Mon Sep 17 00:00:00 2001 From: MinRK Date: Tue, 19 Jun 2012 20:55:25 -0700 Subject: [PATCH] add tests for notebook_dir validation --- .../html/notebook/tests/test_nbmanager.py | 34 +++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 IPython/frontend/html/notebook/tests/test_nbmanager.py diff --git a/IPython/frontend/html/notebook/tests/test_nbmanager.py b/IPython/frontend/html/notebook/tests/test_nbmanager.py new file mode 100644 index 000000000..815872501 --- /dev/null +++ b/IPython/frontend/html/notebook/tests/test_nbmanager.py @@ -0,0 +1,34 @@ +"""Tests for the notebook manager.""" + +import os +from unittest import TestCase +from tempfile import NamedTemporaryFile + +from IPython.utils.tempdir import TemporaryDirectory +from IPython.utils.traitlets import TraitError + +from IPython.frontend.html.notebook.notebookmanager import NotebookManager + +class TestNotebookManager(TestCase): + + def test_nb_dir(self): + with TemporaryDirectory() as td: + km = NotebookManager(notebook_dir=td) + self.assertEquals(km.notebook_dir, td) + + def test_create_nb_dir(self): + with TemporaryDirectory() as td: + nbdir = os.path.join(td, 'notebooks') + km = NotebookManager(notebook_dir=nbdir) + self.assertEquals(km.notebook_dir, nbdir) + + def test_missing_nb_dir(self): + with TemporaryDirectory() as td: + nbdir = os.path.join(td, 'notebook', 'dir', 'is', 'missing') + self.assertRaises(TraitError, NotebookManager, notebook_dir=nbdir) + + def test_invalid_nb_dir(self): + with NamedTemporaryFile() as tf: + self.assertRaises(TraitError, NotebookManager, notebook_dir=tf.name) + +