def test_disconnect_operational_error(self):
# Operation errors should be catched
disconnect()
self.assertEqual(gobworkflow.storage.storage.session, None)
self.assertEqual(gobworkflow.storage.storage.engine, None)
def test_disconnect(self, mock_rollback, mock_close, mock_dispose):
disconnect()
mock_rollback.assert_called()
mock_close.assert_called()
mock_dispose.assert_called()
self.assertEqual(gobworkflow.storage.storage.session, None)
self.assertEqual(gobworkflow.storage.storage.engine, None)
self.assertEqual(is_connected(), False)
def test_disconnect_other_error(self):
# Only operational errors should be catched
with self.assertRaises(MockException):
disconnect()