96 lines
3.2 KiB
Python
96 lines
3.2 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.
|
|
|
|
import collections
|
|
|
|
from oslo_utils import importutils
|
|
|
|
from ironic.tests.unit.db import base as db_base
|
|
from ironic.tests.unit.db import utils as db_utils
|
|
|
|
|
|
INFO_DICT = db_utils.get_test_drac_info()
|
|
|
|
dracclient_job = importutils.try_import('dracclient.resources.job')
|
|
dracclient_raid = importutils.try_import('dracclient.resources.raid')
|
|
|
|
|
|
class BaseDracTest(db_base.DbTestCase):
|
|
def setUp(self):
|
|
super(BaseDracTest, self).setUp()
|
|
self.config(enabled_hardware_types=['idrac', 'fake-hardware'],
|
|
enabled_power_interfaces=['idrac-wsman', 'fake'],
|
|
enabled_management_interfaces=['idrac-wsman', 'fake'],
|
|
enabled_inspect_interfaces=[
|
|
'idrac-wsman', 'fake', 'no-inspect'],
|
|
enabled_vendor_interfaces=[
|
|
'idrac-wsman', 'fake', 'no-vendor'],
|
|
enabled_raid_interfaces=['idrac-wsman', 'fake', 'no-raid'])
|
|
|
|
|
|
class DictToObj(object):
|
|
"""Returns a dictionary into a class"""
|
|
def __init__(self, dictionary):
|
|
for key in dictionary:
|
|
setattr(self, key, dictionary[key])
|
|
|
|
|
|
def dict_to_namedtuple(name='GenericNamedTuple', values=None,
|
|
tuple_class=None):
|
|
"""Converts a dict to a collections.namedtuple"""
|
|
|
|
if values is None:
|
|
values = {}
|
|
|
|
if tuple_class is None:
|
|
tuple_class = collections.namedtuple(name, list(values))
|
|
else:
|
|
# Support different versions of the driver as fields change.
|
|
values = {field: values.get(field) for field in tuple_class._fields}
|
|
|
|
return tuple_class(**values)
|
|
|
|
|
|
def dict_of_object(data):
|
|
"""Create a dictionary object"""
|
|
|
|
for k, v in data.items():
|
|
if isinstance(v, dict):
|
|
dict_obj = DictToObj(v)
|
|
data[k] = dict_obj
|
|
return data
|
|
|
|
|
|
def make_job(job_dict):
|
|
tuple_class = dracclient_job.Job if dracclient_job else None
|
|
return dict_to_namedtuple(values=job_dict,
|
|
tuple_class=tuple_class)
|
|
|
|
|
|
def make_raid_controller(raid_controller_dict):
|
|
tuple_class = dracclient_raid.RAIDController if dracclient_raid else None
|
|
return dict_to_namedtuple(values=raid_controller_dict,
|
|
tuple_class=tuple_class)
|
|
|
|
|
|
def make_virtual_disk(virtual_disk_dict):
|
|
tuple_class = dracclient_raid.VirtualDisk if dracclient_raid else None
|
|
return dict_to_namedtuple(values=virtual_disk_dict,
|
|
tuple_class=tuple_class)
|
|
|
|
|
|
def make_physical_disk(physical_disk_dict):
|
|
tuple_class = dracclient_raid.PhysicalDisk if dracclient_raid else None
|
|
return dict_to_namedtuple(values=physical_disk_dict,
|
|
tuple_class=tuple_class)
|