def volume_create(context, values):
"""Create a volume."""
if not values.get('id'):
values['id'] = uuidutils.generate_uuid()
vol_ref = models.Volume()
vol_ref.update(values)
session = get_session()
with session.begin():
session.add(vol_ref)
return _volume_get(context, vol_ref['id'], session=session)
def volumes_create(context, volumes):
"""Create multiple volumes."""
session = get_session()
vol_refs = []
with session.begin():
for vol in volumes:
LOG.debug('adding new volume for native_volume_id {0}:'.format(
vol.get('native_volume_id')))
if not vol.get('id'):
vol['id'] = uuidutils.generate_uuid()
vol_ref = models.Volume()
vol_ref.update(vol)
vol_refs.append(vol_ref)
session.add_all(vol_refs)
return vol_refs
def test_volume_create(self, mock_session):
fake_volume = models.Volume()
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_volume
result = db_api.volume_create(ctxt, fake_volume)
assert len(result) == 0
def test_shares_create(self, mock_session):
fake_share = [models.Volume()]
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_share
result = db_api.shares_create(ctxt, fake_share)
assert len(result) == 1
def test_filesystem_create(self, mock_session):
fake_filesystem = models.Volume()
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_filesystem
result = db_api.filesystem_create(ctxt, fake_filesystem)
assert len(result) == 0
def test_controller_create(self, mock_session):
fake_controller = models.Volume()
mock_session.return_value.__enter__.return_value.query.return_value \
= fake_controller
result = db_api.controller_create(ctxt, fake_controller)
assert len(result) == 0
