104 lines
4.2 KiB
Python
104 lines
4.2 KiB
Python
# Copyright (c) 2021 Dell Inc. or its subsidiaries.
|
|
#
|
|
# 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.
|
|
|
|
from unittest import mock
|
|
|
|
from oslo_utils import importutils
|
|
|
|
from ironic.common import exception
|
|
from ironic.conductor import task_manager
|
|
from ironic.drivers.modules.drac import utils as drac_utils
|
|
from ironic.drivers.modules.redfish import utils as redfish_utils
|
|
from ironic.tests.unit.drivers.modules.drac import utils as test_utils
|
|
from ironic.tests.unit.objects import utils as obj_utils
|
|
|
|
sushy = importutils.try_import('sushy')
|
|
|
|
INFO_DICT = test_utils.INFO_DICT
|
|
|
|
|
|
@mock.patch.object(redfish_utils, 'get_system', autospec=True)
|
|
class DracUtilsOemManagerTestCase(test_utils.BaseDracTest):
|
|
|
|
def setUp(self):
|
|
super(DracUtilsOemManagerTestCase, self).setUp()
|
|
self.node = obj_utils.create_test_node(self.context,
|
|
driver='idrac',
|
|
driver_info=INFO_DICT)
|
|
self.config(enabled_hardware_types=['idrac'],
|
|
enabled_management_interfaces=['idrac-redfish'])
|
|
|
|
def test_execute_oem_manager_method(self, mock_get_system):
|
|
fake_manager_oem = mock.Mock()
|
|
fake_manager_oem.test_method.return_value = 42
|
|
fake_manager = mock.Mock()
|
|
fake_manager.get_oem_extension.return_value = fake_manager_oem
|
|
mock_get_system.return_value.managers = [fake_manager]
|
|
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
result = drac_utils.execute_oem_manager_method(
|
|
task, 'test method', lambda m: m.test_method())
|
|
|
|
self.assertEqual(42, result)
|
|
|
|
def test_execute_oem_manager_method_no_managers(self, mock_get_system):
|
|
mock_get_system.return_value.managers = []
|
|
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
self.assertRaises(
|
|
exception.RedfishError,
|
|
drac_utils.execute_oem_manager_method,
|
|
task,
|
|
'test method',
|
|
lambda m: m.test_method())
|
|
|
|
def test_execute_oem_manager_method_oem_not_found(self, mock_get_system):
|
|
fake_manager = mock.Mock()
|
|
fake_manager.get_oem_extension.side_effect = (
|
|
sushy.exceptions.OEMExtensionNotFoundError)
|
|
mock_get_system.return_value.managers = [fake_manager]
|
|
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
self.assertRaises(
|
|
exception.RedfishError,
|
|
drac_utils.execute_oem_manager_method,
|
|
task,
|
|
'test method',
|
|
lambda m: m.test_method())
|
|
|
|
def test_execute_oem_manager_method_managers_fail(self, mock_get_system):
|
|
fake_manager_oem1 = mock.Mock()
|
|
fake_manager_oem1.test_method.side_effect = (
|
|
sushy.exceptions.SushyError)
|
|
fake_manager1 = mock.Mock()
|
|
fake_manager1.get_oem_extension.return_value = fake_manager_oem1
|
|
fake_manager_oem2 = mock.Mock()
|
|
fake_manager_oem2.test_method.side_effect = (
|
|
sushy.exceptions.SushyError)
|
|
fake_manager2 = mock.Mock()
|
|
fake_manager2.get_oem_extension.return_value = fake_manager_oem2
|
|
mock_get_system.return_value.managers = [fake_manager1, fake_manager2]
|
|
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
self.assertRaises(
|
|
exception.RedfishError,
|
|
drac_utils.execute_oem_manager_method,
|
|
task,
|
|
'test method',
|
|
lambda m: m.test_method())
|