diff options
author | Jenkins <jenkins@review.openstack.org> | 2015-06-27 03:20:57 +0000 |
---|---|---|
committer | Gerrit Code Review <review@openstack.org> | 2015-06-27 03:20:57 +0000 |
commit | 239c4719726694793beda82f77d709e3d878216a (patch) | |
tree | fedda9dc5f861580cd34de1b3bcede2d7d23d476 | |
parent | b0e9c35d3aa7cc19ebcc9090f3224c31f76ce64b (diff) | |
parent | 2e79681670f7b8e6843abf29d6e95ffda0c3147c (diff) | |
download | oslo-db-239c4719726694793beda82f77d709e3d878216a.tar.gz |
Merge "Fix sqlalchemy.ModelBase.__contains__() behaviour"2.0.0
-rw-r--r-- | oslo_db/sqlalchemy/models.py | 10 | ||||
-rw-r--r-- | oslo_db/tests/sqlalchemy/test_models.py | 12 |
2 files changed, 21 insertions, 1 deletions
diff --git a/oslo_db/sqlalchemy/models.py b/oslo_db/sqlalchemy/models.py index 4293c9d..0c4e0de 100644 --- a/oslo_db/sqlalchemy/models.py +++ b/oslo_db/sqlalchemy/models.py @@ -54,7 +54,15 @@ class ModelBase(six.Iterator): return getattr(self, key) def __contains__(self, key): - return hasattr(self, key) + # Don't use hasattr() because hasattr() catches any exception, not only + # AttributeError. We want to passthrough SQLAlchemy exceptions + # (ex: sqlalchemy.orm.exc.DetachedInstanceError). + try: + getattr(self, key) + except AttributeError: + return False + else: + return True def get(self, key, default=None): return getattr(self, key, default) diff --git a/oslo_db/tests/sqlalchemy/test_models.py b/oslo_db/tests/sqlalchemy/test_models.py index 525573a..279c5c3 100644 --- a/oslo_db/tests/sqlalchemy/test_models.py +++ b/oslo_db/tests/sqlalchemy/test_models.py @@ -70,6 +70,18 @@ class ModelBaseTest(test_base.DbTestCase): self.assertFalse('non-existent-key' in mb) + def test_modelbase_contains_exc(self): + class ErrorModel(models.ModelBase): + @property + def bug(self): + raise ValueError + + model = ErrorModel() + model.update({'attr': 5}) + + self.assertTrue('attr' in model) + self.assertRaises(ValueError, lambda: 'bug' in model) + def test_modelbase_items_iteritems(self): h = {'a': '1', 'b': '2'} expected = { |