def test_open_session(self): stream = mock.MagicMock() sess = session.open_session(stream) self.assertEqual(sess.stream, stream) self.open_brain_hook.assert_called_once_with(hook.HOOK_OPEN_BRAIN, sess)
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 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_after_open(self): stream = mock.MagicMock() self.assertRaises(KeyError, session.get_session, stream) sess1 = session.open_session(stream) sess2 = session.get_session(stream) self.assertEqual(sess1, sess2)
def on_connect(self): log.debug("on_connect: '%s'", self) self.read_delimiter = "\r\n" session.open_session(self)