130 lines
4.9 KiB
Python
130 lines
4.9 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
"""Base class and convenience utilities for functional placement tests."""
|
|
|
|
from oslo_utils.fixture import uuidsentinel as uuids
|
|
from oslo_utils import uuidutils
|
|
|
|
from nova.api.openstack.placement import exception
|
|
from nova.api.openstack.placement.objects import consumer as consumer_obj
|
|
from nova.api.openstack.placement.objects import project as project_obj
|
|
from nova.api.openstack.placement.objects import resource_provider as rp_obj
|
|
from nova.api.openstack.placement.objects import user as user_obj
|
|
from nova.tests.functional.api.openstack.placement import base
|
|
|
|
|
|
def create_provider(context, name, *aggs, **kwargs):
|
|
parent = kwargs.get('parent')
|
|
root = kwargs.get('root')
|
|
uuid = kwargs.get('uuid', getattr(uuids, name))
|
|
rp = rp_obj.ResourceProvider(context, name=name, uuid=uuid)
|
|
if parent:
|
|
rp.parent_provider_uuid = parent
|
|
if root:
|
|
rp.root_provider_uuid = root
|
|
rp.create()
|
|
if aggs:
|
|
rp.set_aggregates(aggs)
|
|
return rp
|
|
|
|
|
|
def add_inventory(rp, rc, total, **kwargs):
|
|
kwargs.setdefault('max_unit', total)
|
|
inv = rp_obj.Inventory(rp._context, resource_provider=rp,
|
|
resource_class=rc, total=total, **kwargs)
|
|
inv.obj_set_defaults()
|
|
rp.add_inventory(inv)
|
|
return inv
|
|
|
|
|
|
def set_traits(rp, *traits):
|
|
tlist = []
|
|
for tname in traits:
|
|
try:
|
|
trait = rp_obj.Trait.get_by_name(rp._context, tname)
|
|
except exception.TraitNotFound:
|
|
trait = rp_obj.Trait(rp._context, name=tname)
|
|
trait.create()
|
|
tlist.append(trait)
|
|
rp.set_traits(rp_obj.TraitList(objects=tlist))
|
|
return tlist
|
|
|
|
|
|
def ensure_consumer(ctx, user, project, consumer_id=None):
|
|
# NOTE(efried): If not specified, use a random consumer UUID - we don't
|
|
# want to override any existing allocations from the test case.
|
|
consumer_id = consumer_id or uuidutils.generate_uuid()
|
|
try:
|
|
consumer = consumer_obj.Consumer.get_by_uuid(ctx, consumer_id)
|
|
except exception.NotFound:
|
|
consumer = consumer_obj.Consumer(
|
|
ctx, uuid=consumer_id, user=user, project=project)
|
|
consumer.create()
|
|
return consumer
|
|
|
|
|
|
def set_allocation(ctx, rp, consumer, rc_used_dict):
|
|
alloc = [
|
|
rp_obj.Allocation(
|
|
ctx, resource_provider=rp, resource_class=rc,
|
|
consumer=consumer, used=used)
|
|
for rc, used in rc_used_dict.items()
|
|
]
|
|
alloc_list = rp_obj.AllocationList(ctx, objects=alloc)
|
|
alloc_list.replace_all()
|
|
return alloc_list
|
|
|
|
|
|
class PlacementDbBaseTestCase(base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(PlacementDbBaseTestCase, self).setUp()
|
|
# we use context in some places and ctx in other. We should only use
|
|
# context, but let's paper over that for now.
|
|
self.ctx = self.context
|
|
self.user_obj = user_obj.User(self.ctx, external_id='fake-user')
|
|
self.user_obj.create()
|
|
self.project_obj = project_obj.Project(
|
|
self.ctx, external_id='fake-project')
|
|
self.project_obj.create()
|
|
# For debugging purposes, populated by _create_provider and used by
|
|
# _validate_allocation_requests to make failure results more readable.
|
|
self.rp_uuid_to_name = {}
|
|
|
|
def _create_provider(self, name, *aggs, **kwargs):
|
|
rp = create_provider(self.ctx, name, *aggs, **kwargs)
|
|
self.rp_uuid_to_name[rp.uuid] = name
|
|
return rp
|
|
|
|
def allocate_from_provider(self, rp, rc, used, consumer_id=None,
|
|
consumer=None):
|
|
if consumer is None:
|
|
consumer = ensure_consumer(
|
|
self.ctx, self.user_obj, self.project_obj, consumer_id)
|
|
alloc_list = set_allocation(self.ctx, rp, consumer, {rc: used})
|
|
return alloc_list
|
|
|
|
def _make_allocation(self, inv_dict, alloc_dict):
|
|
rp = self._create_provider('allocation_resource_provider')
|
|
disk_inv = rp_obj.Inventory(context=self.ctx,
|
|
resource_provider=rp, **inv_dict)
|
|
inv_list = rp_obj.InventoryList(objects=[disk_inv])
|
|
rp.set_inventory(inv_list)
|
|
consumer_id = alloc_dict['consumer_id']
|
|
consumer = ensure_consumer(
|
|
self.ctx, self.user_obj, self.project_obj, consumer_id)
|
|
alloc = rp_obj.Allocation(self.ctx, resource_provider=rp,
|
|
consumer=consumer, **alloc_dict)
|
|
alloc_list = rp_obj.AllocationList(self.ctx, objects=[alloc])
|
|
alloc_list.replace_all()
|
|
return rp, alloc
|