def test_update_range_reraise_error(self):
session = mock.Mock()
session.query.side_effect = orm_exc.ObjectDeletedError(None, None)
self.assertRaises(db_exc.RetryRequest,
self.subnet_manager.update_range,
session,
mock.Mock(),
first_ip='1.2.3.5')
def test_delete_range_reraise_error(self):
session = mock.Mock()
session.query.side_effect = orm_exc.ObjectDeletedError(None, None)
@ndb_api.retry_db_errors
def go():
self.subnet_manager.delete_range(session, mock.Mock())
self.assertRaises(ipam_exc.IPAllocationFailed, go)
def test_exceptions(self):
class Foo(object):
pass
users = self.tables.users
mapper(User, users)
for sa_exc in (
orm_exc.UnmappedInstanceError(Foo()),
orm_exc.UnmappedClassError(Foo),
orm_exc.ObjectDeletedError(attributes.instance_state(User())),
):
for loads, dumps in picklers():
repickled = loads(dumps(sa_exc))
eq_(repickled.args[0], sa_exc.args[0])
