|
|
"""Tests for the session manager."""
|
|
|
|
|
|
import os
|
|
|
|
|
|
from unittest import TestCase
|
|
|
from tempfile import NamedTemporaryFile
|
|
|
|
|
|
from IPython.utils.tempdir import TemporaryDirectory
|
|
|
from IPython.utils.traitlets import TraitError
|
|
|
|
|
|
from ..sessionmanager import SessionManager
|
|
|
|
|
|
class TestSessionManager(TestCase):
|
|
|
|
|
|
def test_get_session(self):
|
|
|
sm = SessionManager()
|
|
|
session_id = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id, name='test.ipynb', path='/path/to/', kernel='5678')
|
|
|
model = sm.get_session(id=session_id)
|
|
|
expected = {'id':session_id, 'name':u'test.ipynb', 'path': u'/path/to/', 'kernel':{'id':u'5678', 'ws_url': u''}}
|
|
|
self.assertEqual(model, expected)
|
|
|
|
|
|
def test_bad_get_session(self):
|
|
|
# Should raise error if a bad key is passed to the database.
|
|
|
sm = SessionManager()
|
|
|
session_id = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id, name='test.ipynb', path='/path/to/', kernel='5678')
|
|
|
self.assertRaises(TraitError, sm.get_session, bad_id=session_id) # Bad keyword
|
|
|
|
|
|
def test_list_sessions(self):
|
|
|
sm = SessionManager()
|
|
|
session_id1 = sm.get_session_id()
|
|
|
session_id2 = sm.get_session_id()
|
|
|
session_id3 = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id1, name='test1.ipynb', path='/path/to/1/', kernel='5678')
|
|
|
sm.save_session(session_id=session_id2, name='test2.ipynb', path='/path/to/2/', kernel='5678')
|
|
|
sm.save_session(session_id=session_id3, name='test3.ipynb', path='/path/to/3/', kernel='5678')
|
|
|
sessions = sm.list_sessions()
|
|
|
expected = [{'id':session_id1, 'name':u'test1.ipynb',
|
|
|
'path': u'/path/to/1/', 'kernel':{'id':u'5678', 'ws_url': u''}},
|
|
|
{'id':session_id2, 'name':u'test2.ipynb',
|
|
|
'path': u'/path/to/2/', 'kernel':{'id':u'5678', 'ws_url': u''}},
|
|
|
{'id':session_id3, 'name':u'test3.ipynb',
|
|
|
'path': u'/path/to/3/', 'kernel':{'id':u'5678', 'ws_url': u''}}]
|
|
|
self.assertEqual(sessions, expected)
|
|
|
|
|
|
def test_update_session(self):
|
|
|
sm = SessionManager()
|
|
|
session_id = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id, name='test.ipynb', path='/path/to/', kernel=None)
|
|
|
sm.update_session(session_id, kernel='5678')
|
|
|
sm.update_session(session_id, name='new_name.ipynb')
|
|
|
model = sm.get_session(id=session_id)
|
|
|
expected = {'id':session_id, 'name':u'new_name.ipynb', 'path': u'/path/to/', 'kernel':{'id':u'5678', 'ws_url': u''}}
|
|
|
self.assertEqual(model, expected)
|
|
|
|
|
|
def test_bad_update_session(self):
|
|
|
# try to update a session with a bad keyword ~ raise error
|
|
|
sm = SessionManager()
|
|
|
session_id = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id, name='test.ipynb', path='/path/to/', kernel='5678')
|
|
|
self.assertRaises(TraitError, sm.update_session, session_id=session_id, bad_kw='test.ipynb') # Bad keyword
|
|
|
|
|
|
def test_delete_session(self):
|
|
|
sm = SessionManager()
|
|
|
session_id1 = sm.get_session_id()
|
|
|
session_id2 = sm.get_session_id()
|
|
|
session_id3 = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id1, name='test1.ipynb', path='/path/to/1/', kernel='5678')
|
|
|
sm.save_session(session_id=session_id2, name='test2.ipynb', path='/path/to/2/', kernel='5678')
|
|
|
sm.save_session(session_id=session_id3, name='test3.ipynb', path='/path/to/3/', kernel='5678')
|
|
|
sm.delete_session(session_id2)
|
|
|
sessions = sm.list_sessions()
|
|
|
expected = [{'id':session_id1, 'name':u'test1.ipynb',
|
|
|
'path': u'/path/to/1/', 'kernel':{'id':u'5678', 'ws_url': u''}},
|
|
|
{'id':session_id3, 'name':u'test3.ipynb',
|
|
|
'path': u'/path/to/3/', 'kernel':{'id':u'5678', 'ws_url': u''}}]
|
|
|
self.assertEqual(sessions, expected)
|
|
|
|
|
|
def test_bad_delete_session(self):
|
|
|
# try to delete a session that doesn't exist ~ raise error
|
|
|
sm = SessionManager()
|
|
|
session_id = sm.get_session_id()
|
|
|
sm.save_session(session_id=session_id, name='test.ipynb', path='/path/to/', kernel='5678')
|
|
|
self.assertRaises(TraitError, sm.delete_session, session_id='23424') # Bad keyword
|
|
|
|
|
|
|
