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()
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')