def test_close(self): from pyramid_sockjs import Session, STATE_CLOSING session = Session('id') session.open() session.close() self.assertFalse(session.expired) self.assertEqual(session.state, STATE_CLOSING)
def test_close_event(self): from pyramid_sockjs import Session closing = [] class TestSession(Session): def on_close(self): closing.append(True) session = TestSession('id') session.open() session.close() self.assertTrue(closing[0])
def test_close_on_message_exception(self): from pyramid_sockjs import Session class TestSession(Session): def on_close(self): raise Exception() session = TestSession('id') session.open() err = None try: session.close() except Exception as exc: # pragma: no cover err = exc self.assertIsNone(err)