Esempio n. 1
0
 def test_handle_session(self):
     session = mock.MagicMock()
     handler = SessionHandler(session)
     self.assertIsNone(handler.handle_session(session))
Esempio n. 2
0
 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)
Esempio n. 3
0
 def test_get_session(self):
     session = mock.MagicMock()
     handler = SessionHandler(session)
     session_returned = handler.get_session()
     self.assertIs(session, session_returned)