def storage_pool_create(context, values): """Create a storage_pool from the values dictionary.""" if not values.get('id'): values['id'] = uuidutils.generate_uuid() storage_pool_ref = models.StoragePool() storage_pool_ref.update(values) session = get_session() with session.begin(): session.add(storage_pool_ref) return _storage_pool_get(context, storage_pool_ref['id'], session=session)
def fake_storage_pool_create(): fake_storage_pools = [models.StoragePool(), models.StoragePool()] fake_storage_pools[0] = { 'id': '14155a1f-f053-4ccb-a846-ed67e4387428', 'storage_id': '12c2d52f-01bc-41f5-b73f' '-7abf6f38a2a6', 'name': 'SRP_1', 'status': 'normal', 'created_at': '2020-06-10T07:17:08.707356', 'updated_at': '2020-06-10T07:17:08.707356', 'native_storage_pool_id': 'SRP_1', 'storage_type': 'block', 'total_capacity': 26300318136401, 'used_capacity': 19054536509358, 'free_capacity': 7245781627043, 'subscribed_capacity': 219902325555200, "description": "fake storage Pool", } fake_storage_pools[1] = { 'id': "95f7b7ed-bd7f-426e-b05f-f1ffeb4f09df", 'storage_id': "84d0c5f7-2349-401c-8672" "-f76214d13cab", 'name': "SRP_2", 'status': "normal", 'created_at': "2020-06-10T07:17:08.707356", 'updated_at': "2020-06-10T07:17:08.707356", 'native_storage_pool_id': "SRP_2", 'extra': "extra attrib", # invalid key 'storage_type': "block", 'total_capacity': 26300318136401, 'used_capacity': 19054536509358, 'free_capacity': 7245781627043, 'subscribed_capacity': 219902325555200, 'description': "fake storage Pool", } return fake_storage_pools
def storage_pools_create(context, storage_pools): """Create a storage_pool from the values dictionary.""" session = get_session() storage_pool_refs = [] with session.begin(): for storage_pool in storage_pools: LOG.debug( 'adding new storage_pool for native_storage_pool_id {0}:'. format(storage_pool.get('native_storage_pool_id'))) if not storage_pool.get('id'): storage_pool['id'] = uuidutils.generate_uuid() storage_pool_ref = models.StoragePool() storage_pool_ref.update(storage_pool) storage_pool_refs.append(storage_pool_ref) session.add_all(storage_pool_refs) return storage_pool_refs
def test_storage_pool_create(self, mock_session): fake_storage_pool = models.StoragePool() mock_session.return_value.__enter__.return_value.query.return_value \ = fake_storage_pool result = db_api.storage_pool_create(context, fake_storage_pool) assert len(result) == 0
def test_storage_pools_delete(self, mock_session): fake_storage_pools = [models.StoragePool().id] mock_session.return_value.__enter__.return_value.query.return_value \ = fake_storage_pools result = db_api.storage_pools_delete(context, fake_storage_pools) assert result is None