def test_show_with_invalid_id(self):
self.mock_object(
db, 'volume_get',
mock.Mock(side_effect=exception.VolumeNotFound('fake_id')))
req = fakes.HTTPRequest.blank('/volumes/fake_id')
self.assertRaises(exception.VolumeNotFound, self.controller.show, req,
'fake_id')
def _volume_get(context, volume_id, session=None):
result = (_volume_get_query(
context, session=session).filter_by(id=volume_id).first())
if not result:
raise exception.VolumeNotFound(volume_id)
return result
def volumes_update(context, volumes):
"""Update multiple volumes."""
session = get_session()
with session.begin():
for vol in volumes:
LOG.debug('updating volume {0}:'.format(vol.get('id')))
query = _volume_get_query(context, session)
result = query.filter_by(id=vol.get('id')).update(vol)
if not result:
LOG.error(exception.VolumeNotFound(vol.get('id')))
def volumes_delete(context, volumes_id_list):
"""Delete multiple volumes."""
session = get_session()
with session.begin():
for vol_id in volumes_id_list:
LOG.debug('deleting volume {0}:'.format(vol_id))
query = _volume_get_query(context, session)
result = query.filter_by(id=vol_id).delete()
if not result:
LOG.error(exception.VolumeNotFound(vol_id))
return
