def __init__(self, session_inactivity_timeout=60, session_storage=None):
self.session_storage = session_storage or MemorySessionStorage()
self.sessions = {}
self.session_inactivity_timeout = session_inactivity_timeout
self.timers = {}
def setUp(self):
self.session_source = MemorySessionStorage()
self.switch_descriptor = mock.Mock()
def setUp(self):
self.session_source = MemorySessionStorage()
self.switch_descriptor = mock.Mock()
class SessionStorageTest(TestCase):
def setUp(self):
self.session_source = MemorySessionStorage()
self.switch_descriptor = mock.Mock()
def test_add_session(self):
self.session_source.add('some_session', self.switch_descriptor)
assert_that(self.session_source.get('some_session'),
is_(self.switch_descriptor))
def test_get_session(self):
self.session_source.add('some_session', self.switch_descriptor)
assert_that(self.session_source.get('some_session'), is_(self.switch_descriptor))
def test_get_nonexistent_session_is_none(self):
assert_that(self.session_source.get('nonexistent_session'), is_(none()))
def test_remove_session(self):
self.session_source.add('some_session', self.switch_descriptor)
self.session_source.remove('some_session')
assert_that(self.session_source.get('some_session'), is_(none()))
def test_add_session_that_already_exists_fails(self):
self.session_source.add('some_session', self.switch_descriptor)
with self.assertRaises(SessionAlreadyExists):
self.session_source.add('some_session', self.switch_descriptor)
def test_remove_nonexistent_session_fails(self):
self.session_source.add('other_session', self.switch_descriptor)
with self.assertRaises(UnknownSession):
self.session_source.remove('some_session')
class SessionStorageTest(TestCase):
def setUp(self):
self.session_source = MemorySessionStorage()
self.switch_descriptor = mock.Mock()
def test_add_session(self):
self.session_source.add('some_session', self.switch_descriptor)
assert_that(self.session_source.get('some_session'),
is_(self.switch_descriptor))
def test_get_session(self):
self.session_source.add('some_session', self.switch_descriptor)
assert_that(self.session_source.get('some_session'),
is_(self.switch_descriptor))
def test_get_nonexistent_session_is_none(self):
assert_that(self.session_source.get('nonexistent_session'),
is_(none()))
def test_remove_session(self):
self.session_source.add('some_session', self.switch_descriptor)
self.session_source.remove('some_session')
assert_that(self.session_source.get('some_session'), is_(none()))
def test_add_session_that_already_exists_fails(self):
self.session_source.add('some_session', self.switch_descriptor)
with self.assertRaises(SessionAlreadyExists):
self.session_source.add('some_session', self.switch_descriptor)
def test_remove_nonexistent_session_fails(self):
self.session_source.add('other_session', self.switch_descriptor)
with self.assertRaises(UnknownSession):
self.session_source.remove('some_session')
