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)
