diff options
author | Jenkins <jenkins@review.openstack.org> | 2017-07-05 11:32:54 +0000 |
---|---|---|
committer | Gerrit Code Review <review@openstack.org> | 2017-07-05 11:32:54 +0000 |
commit | 23e431b0deb0d4b50f60f7f50f44fb5942c2fc76 (patch) | |
tree | 421207c796554ad84b612cadef9287aa4c10584d | |
parent | 0a035933c813631f0c523bcc3b603dbd6ebf031e (diff) | |
parent | bee339d79c8ac318573fd54229d99e64fd27617a (diff) | |
download | oslo-db-23e431b0deb0d4b50f60f7f50f44fb5942c2fc76.tar.gz |
Merge "Add 'save_and_reraise_exception' method when call 'session.rollback()'"4.24.1
-rw-r--r-- | oslo_db/sqlalchemy/enginefacade.py | 9 | ||||
-rw-r--r-- | oslo_db/tests/sqlalchemy/test_enginefacade.py | 26 |
2 files changed, 29 insertions, 6 deletions
diff --git a/oslo_db/sqlalchemy/enginefacade.py b/oslo_db/sqlalchemy/enginefacade.py index af3a004..752c6ca 100644 --- a/oslo_db/sqlalchemy/enginefacade.py +++ b/oslo_db/sqlalchemy/enginefacade.py @@ -20,6 +20,7 @@ import warnings import debtcollector.removals as removals from oslo_config import cfg +from oslo_utils import excutils from oslo_db import exception from oslo_db import options @@ -636,12 +637,8 @@ class _TransactionContext(object): yield self.session self._end_session_transaction(self.session) except Exception: - self.session.rollback() - # TODO(zzzeek) do we need save_and_reraise() here, - # or do newer eventlets not have issues? we are using - # raw "raise" in many other places in oslo.db already - # (and one six.reraise()). - raise + with excutils.save_and_reraise_exception(): + self.session.rollback() finally: self.session.close() self.session = None diff --git a/oslo_db/tests/sqlalchemy/test_enginefacade.py b/oslo_db/tests/sqlalchemy/test_enginefacade.py index 206c335..303a8dd 100644 --- a/oslo_db/tests/sqlalchemy/test_enginefacade.py +++ b/oslo_db/tests/sqlalchemy/test_enginefacade.py @@ -13,6 +13,7 @@ import collections import contextlib import copy +import fixtures import pickle import warnings @@ -24,6 +25,7 @@ from sqlalchemy import Column from sqlalchemy import Integer from sqlalchemy import MetaData from sqlalchemy.orm import mapper +from sqlalchemy.orm import Session from sqlalchemy import select from sqlalchemy import String from sqlalchemy import Table @@ -1654,6 +1656,30 @@ class LiveFacadeTest(test_base.DbTestCase): session.query(self.User.name).scalar() ) + @mock.patch.object(Session, 'commit') + @mock.patch.object(Session, 'rollback') + def test_save_and_reraise_when_rollback_exception(self, + rollback_patch, + commit_patch): + context = oslo_context.RequestContext() + log = self.useFixture(fixtures.FakeLogger()) + + class RollbackException(Exception): + pass + + class CommitException(Exception): + pass + + commit_patch.side_effect = CommitException() + rollback_patch.side_effect = RollbackException() + + @enginefacade.writer + def go_session(context): + context.session.add(self.User(name="u1")) + + self.assertRaises(RollbackException, go_session, context) + self.assertIn('CommitException', log.output) + def test_flush_on_subtransaction(self): facade = enginefacade.transaction_context() facade.configure( |