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])