ef44187a8b
Added autospec=True or spec_set= to mocks Fixed one incorrect call to has_calls() and changed to assert_has_calls() Change-Id: I01756a4db88e50a8a663a164644feebd9be90f2c
129 lines
6.3 KiB
Python
129 lines
6.3 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.
|
|
|
|
"""Test class for AMT Vendor methods."""
|
|
|
|
import mock
|
|
|
|
from ironic.common import boot_devices
|
|
from ironic.common import states
|
|
from ironic.conductor import task_manager
|
|
from ironic.drivers.modules.amt import management as amt_mgmt
|
|
from ironic.drivers.modules import pxe
|
|
from ironic.tests.conductor import utils as mgr_utils
|
|
from ironic.tests.db import base as db_base
|
|
from ironic.tests.db import utils as db_utils
|
|
from ironic.tests.objects import utils as obj_utils
|
|
|
|
INFO_DICT = db_utils.get_test_amt_info()
|
|
|
|
|
|
class AMTPXEVendorPassthruTestCase(db_base.DbTestCase):
|
|
|
|
def setUp(self):
|
|
super(AMTPXEVendorPassthruTestCase, self).setUp()
|
|
mgr_utils.mock_the_extension_manager(driver="pxe_amt")
|
|
self.node = obj_utils.create_test_node(self.context,
|
|
driver='pxe_amt', driver_info=INFO_DICT)
|
|
|
|
def test_vendor_routes(self):
|
|
expected = ['heartbeat', 'pass_deploy_info',
|
|
'pass_bootloader_install_info']
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=True) as task:
|
|
vendor_routes = task.driver.vendor.vendor_routes
|
|
self.assertIsInstance(vendor_routes, dict)
|
|
self.assertEqual(sorted(expected), sorted(list(vendor_routes)))
|
|
|
|
def test_driver_routes(self):
|
|
expected = ['lookup']
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=True) as task:
|
|
driver_routes = task.driver.vendor.driver_routes
|
|
self.assertIsInstance(driver_routes, dict)
|
|
self.assertEqual(sorted(expected), sorted(list(driver_routes)))
|
|
|
|
@mock.patch.object(amt_mgmt.AMTManagement, 'ensure_next_boot_device',
|
|
autospec=True)
|
|
@mock.patch.object(pxe.VendorPassthru, 'pass_deploy_info', autospec=True)
|
|
def test_vendorpassthru_pass_deploy_info_netboot(self,
|
|
mock_pxe_vendorpassthru,
|
|
mock_ensure):
|
|
kwargs = {'address': '123456'}
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
task.node.provision_state = states.DEPLOYWAIT
|
|
task.node.target_provision_state = states.ACTIVE
|
|
task.node.instance_info['capabilities'] = {
|
|
"boot_option": "netboot"
|
|
}
|
|
task.driver.vendor.pass_deploy_info(task, **kwargs)
|
|
mock_ensure.assert_called_with(
|
|
task.driver.management, task.node, boot_devices.PXE)
|
|
mock_pxe_vendorpassthru.assert_called_once_with(
|
|
task.driver.vendor, task, **kwargs)
|
|
|
|
@mock.patch.object(amt_mgmt.AMTManagement, 'ensure_next_boot_device',
|
|
autospec=True)
|
|
@mock.patch.object(pxe.VendorPassthru, 'pass_deploy_info', autospec=True)
|
|
def test_vendorpassthru_pass_deploy_info_localboot(self,
|
|
mock_pxe_vendorpassthru,
|
|
mock_ensure):
|
|
kwargs = {'address': '123456'}
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
task.node.provision_state = states.DEPLOYWAIT
|
|
task.node.target_provision_state = states.ACTIVE
|
|
task.node.instance_info['capabilities'] = {"boot_option": "local"}
|
|
task.driver.vendor.pass_deploy_info(task, **kwargs)
|
|
self.assertFalse(mock_ensure.called)
|
|
mock_pxe_vendorpassthru.assert_called_once_with(
|
|
task.driver.vendor, task, **kwargs)
|
|
|
|
@mock.patch.object(amt_mgmt.AMTManagement, 'ensure_next_boot_device',
|
|
autospec=True)
|
|
@mock.patch.object(pxe.VendorPassthru, 'continue_deploy', autospec=True)
|
|
def test_vendorpassthru_continue_deploy_netboot(self,
|
|
mock_pxe_vendorpassthru,
|
|
mock_ensure):
|
|
kwargs = {'address': '123456'}
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
task.node.provision_state = states.DEPLOYWAIT
|
|
task.node.target_provision_state = states.ACTIVE
|
|
task.node.instance_info['capabilities'] = {
|
|
"boot_option": "netboot"
|
|
}
|
|
task.driver.vendor.continue_deploy(task, **kwargs)
|
|
mock_ensure.assert_called_with(
|
|
task.driver.management, task.node, boot_devices.PXE)
|
|
mock_pxe_vendorpassthru.assert_called_once_with(
|
|
task.driver.vendor, task, **kwargs)
|
|
|
|
@mock.patch.object(amt_mgmt.AMTManagement, 'ensure_next_boot_device',
|
|
autospec=True)
|
|
@mock.patch.object(pxe.VendorPassthru, 'continue_deploy', autospec=True)
|
|
def test_vendorpassthru_continue_deploy_localboot(self,
|
|
mock_pxe_vendorpassthru,
|
|
mock_ensure):
|
|
kwargs = {'address': '123456'}
|
|
with task_manager.acquire(self.context, self.node.uuid,
|
|
shared=False) as task:
|
|
task.node.provision_state = states.DEPLOYWAIT
|
|
task.node.target_provision_state = states.ACTIVE
|
|
task.node.instance_info['capabilities'] = {"boot_option": "local"}
|
|
task.driver.vendor.continue_deploy(task, **kwargs)
|
|
self.assertFalse(mock_ensure.called)
|
|
mock_pxe_vendorpassthru.assert_called_once_with(
|
|
task.driver.vendor, task, **kwargs)
|