diff --git a/IPython/html/services/sessions/tests/test_sessionmanager.py b/IPython/html/services/sessions/tests/test_sessionmanager.py index 36980bd6a..78036c961 100644 --- a/IPython/html/services/sessions/tests/test_sessionmanager.py +++ b/IPython/html/services/sessions/tests/test_sessionmanager.py @@ -6,6 +6,7 @@ from tornado import web from ..sessionmanager import SessionManager from IPython.html.services.kernels.kernelmanager import MappingKernelManager +from IPython.html.services.contents.manager import ContentsManager class DummyKernel(object): def __init__(self, kernel_name='python'): @@ -28,10 +29,17 @@ class DummyMKM(MappingKernelManager): def shutdown_kernel(self, kernel_id, now=False): del self._kernels[kernel_id] + class TestSessionManager(TestCase): + def setUp(self): + self.sm = SessionManager( + kernel_manager=DummyMKM(), + contents_manager=ContentsManager(), + ) + def test_get_session(self): - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm session_id = sm.create_session(path='/path/to/test.ipynb', kernel_name='bar')['id'] model = sm.get_session(session_id=session_id) @@ -42,13 +50,13 @@ class TestSessionManager(TestCase): def test_bad_get_session(self): # Should raise error if a bad key is passed to the database. - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm session_id = sm.create_session(path='/path/to/test.ipynb', kernel_name='foo')['id'] self.assertRaises(TypeError, sm.get_session, bad_id=session_id) # Bad keyword def test_get_session_dead_kernel(self): - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm session = sm.create_session(path='/path/to/1/test1.ipynb', kernel_name='python') # kill the kernel sm.kernel_manager.shutdown_kernel(session['kernel']['id']) @@ -59,7 +67,7 @@ class TestSessionManager(TestCase): self.assertEqual(listed, []) def test_list_sessions(self): - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm sessions = [ sm.create_session(path='/path/to/1/test1.ipynb', kernel_name='python'), sm.create_session(path='/path/to/2/test2.ipynb', kernel_name='python'), @@ -84,7 +92,7 @@ class TestSessionManager(TestCase): self.assertEqual(sessions, expected) def test_list_sessions_dead_kernel(self): - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm sessions = [ sm.create_session(path='/path/to/1/test1.ipynb', kernel_name='python'), sm.create_session(path='/path/to/2/test2.ipynb', kernel_name='python'), @@ -107,7 +115,7 @@ class TestSessionManager(TestCase): self.assertEqual(listed, expected) def test_update_session(self): - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm session_id = sm.create_session(path='/path/to/test.ipynb', kernel_name='julia')['id'] sm.update_session(session_id, path='/path/to/new_name.ipynb') @@ -119,13 +127,13 @@ class TestSessionManager(TestCase): def test_bad_update_session(self): # try to update a session with a bad keyword ~ raise error - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm session_id = sm.create_session(path='/path/to/test.ipynb', kernel_name='ir')['id'] self.assertRaises(TypeError, sm.update_session, session_id=session_id, bad_kw='test.ipynb') # Bad keyword def test_delete_session(self): - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm sessions = [ sm.create_session(path='/path/to/1/test1.ipynb', kernel_name='python'), sm.create_session(path='/path/to/2/test2.ipynb', kernel_name='python'), @@ -147,7 +155,7 @@ class TestSessionManager(TestCase): def test_bad_delete_session(self): # try to delete a session that doesn't exist ~ raise error - sm = SessionManager(kernel_manager=DummyMKM()) + sm = self.sm sm.create_session(path='/path/to/test.ipynb', kernel_name='python') self.assertRaises(TypeError, sm.delete_session, bad_kwarg='23424') # Bad keyword self.assertRaises(web.HTTPError, sm.delete_session, session_id='23424') # nonexistant