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_handle_session(self):
session = mock.MagicMock()
handler = SessionHandler(session)
self.assertIsNone(handler.handle_session(session))
