def test_handle_session(self): session = mock.MagicMock() handler = SessionHandler(session) self.assertIsNone(handler.handle_session(session))
def test_handle_session_exception(self): session = mock.MagicMock() handler = SessionHandler(session) e = Exception() handler.handle_session(session, exc=e) self.assertTrue(session.rollback.called)
def test_get_session(self): session = mock.MagicMock() handler = SessionHandler(session) session_returned = handler.get_session() self.assertIs(session, session_returned)