def test_close_session(self): stream = mock.MagicMock() sess1 = session.open_session(stream) sess2 = session.close_session(stream) self.assertEqual(sess1, sess2) self.close_brain_hook.assert_called_once_with(hook.HOOK_CLOSE_BRAIN, sess1)
def test_get_session_fails_after_close(self): stream = mock.MagicMock() session.open_session(stream) session.close_session(stream) self.assertRaises(KeyError, session.get_session, stream)
def on_close(self): log.debug("on_close: '%s'", self) session.close_session(self)