python-ironicclient/ironicclient/tests/unit/v1/test_node_shell.py

1315 lines
50 KiB
Python

# Copyright 2013 IBM Corp
#
# 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 json
import tempfile
import mock
from ironicclient.common.apiclient import exceptions
from ironicclient.common import cliutils
from ironicclient.common import utils as commonutils
from ironicclient import exc
from ironicclient.tests.unit import utils
import ironicclient.v1.node_shell as n_shell
import ironicclient.v1.utils as v1_utils
class NodeShellTest(utils.BaseTestCase):
def test_node_show(self):
actual = {}
fake_print_dict = lambda data, *args, **kwargs: actual.update(data)
with mock.patch.object(cliutils, 'print_dict', fake_print_dict):
node = object()
n_shell._print_node_show(node)
exp = ['chassis_uuid',
'clean_step',
'created_at',
'console_enabled',
'driver',
'driver_info',
'driver_internal_info',
'extra',
'instance_info',
'instance_uuid',
'last_error',
'maintenance',
'maintenance_reason',
'name',
'boot_interface',
'console_interface',
'deploy_interface',
'inspect_interface',
'management_interface',
'network_interface',
'power_interface',
'raid_interface',
'storage_interface',
'vendor_interface',
'power_state',
'properties',
'provision_state',
'provision_updated_at',
'reservation',
'resource_class',
'target_power_state',
'target_provision_state',
'updated_at',
'inspection_finished_at',
'inspection_started_at',
'uuid',
'raid_config',
'target_raid_config']
act = actual.keys()
self.assertEqual(sorted(exp), sorted(act))
def test_do_node_delete(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = ['node_uuid']
n_shell.do_node_delete(client_mock, args)
client_mock.node.delete.assert_called_once_with('node_uuid')
def test_do_node_delete_multiple(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = ['node_uuid1', 'node_uuid2']
n_shell.do_node_delete(client_mock, args)
client_mock.node.delete.assert_has_calls(
[mock.call('node_uuid1'), mock.call('node_uuid2')])
def test_do_node_delete_multiple_with_exception(self):
client_mock = mock.MagicMock()
client_mock.node.delete.side_effect = (
[exceptions.ClientException, None])
args = mock.MagicMock()
args.node = ['node_uuid1', 'node_uuid2']
self.assertRaises(exceptions.ClientException, n_shell.do_node_delete,
client_mock, args)
client_mock.node.delete.assert_has_calls(
[mock.call('node_uuid1'), mock.call('node_uuid2')])
def test_do_node_update(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.op = 'add'
args.attributes = [['arg1=val1', 'arg2=val2']]
args.json = False
n_shell.do_node_update(client_mock, args)
patch = commonutils.args_array_to_patch(args.op, args.attributes[0])
client_mock.node.update.assert_called_once_with('node_uuid', patch)
def test_do_node_update_wrong_op(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.op = 'foo'
args.attributes = [['arg1=val1', 'arg2=val2']]
args.json = False
self.assertRaises(exceptions.CommandError,
n_shell.do_node_update,
client_mock, args)
self.assertFalse(client_mock.node.update.called)
def test_do_node_create(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with()
def test_do_node_create_with_driver(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.driver = 'driver'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
driver='driver')
def test_do_node_create_with_chassis_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.chassis_uuid = 'chassis_uuid'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
chassis_uuid='chassis_uuid')
def test_do_node_create_with_driver_info(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.driver_info = ['arg1=val1', 'arg2=val2']
args.json = False
n_shell.do_node_create(client_mock, args)
kwargs = {'driver_info': {'arg1': 'val1', 'arg2': 'val2'}}
client_mock.node.create.assert_called_once_with(**kwargs)
def test_do_node_create_with_properties(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.properties = ['arg1=val1', 'arg2=val2']
args.json = False
n_shell.do_node_create(client_mock, args)
kwargs = {'properties': {'arg1': 'val1', 'arg2': 'val2'}}
client_mock.node.create.assert_called_once_with(**kwargs)
def test_do_node_create_with_extra(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.driver = 'driver_name'
args.extra = ['arg1=val1', 'arg2=val2']
args.json = False
n_shell.do_node_create(client_mock, args)
kwargs = {
'driver': 'driver_name',
'extra': {'arg1': 'val1', 'arg2': 'val2'},
}
client_mock.node.create.assert_called_once_with(**kwargs)
def test_do_node_create_with_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.uuid = 'fef99cb8-a0d1-43df-b084-17b3b42b3cbd'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(uuid=args.uuid)
def test_do_node_create_with_name(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.name = 'node_name'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(name=args.name)
def test_do_node_create_with_boot_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.boot_interface = 'boot'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
boot_interface='boot')
def test_do_node_create_with_console_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.console_interface = 'console'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
console_interface='console')
def test_do_node_create_with_deploy_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.deploy_interface = 'deploy'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
deploy_interface='deploy')
def test_do_node_create_with_inspect_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.inspect_interface = 'inspect'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
inspect_interface='inspect')
def test_do_node_create_with_management_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.management_interface = 'management'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
management_interface='management')
def test_do_node_create_with_network_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.network_interface = 'neutron'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
network_interface='neutron')
def test_do_node_create_with_power_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.power_interface = 'power'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
power_interface='power')
def test_do_node_create_with_raid_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.raid_interface = 'raid'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
raid_interface='raid')
def test_do_node_create_with_storage_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.storage_interface = 'storage'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
storage_interface='storage')
def test_do_node_create_with_vendor_interface(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.vendor_interface = 'vendor'
args.json = False
n_shell.do_node_create(client_mock, args)
client_mock.node.create.assert_called_once_with(
vendor_interface='vendor')
def test_do_node_show(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.instance_uuid = False
args.fields = None
args.json = False
n_shell.do_node_show(client_mock, args)
client_mock.node.get.assert_called_once_with('node_uuid', fields=None)
# assert get_by_instance_uuid() wasn't called
self.assertFalse(client_mock.node.get_by_instance_uuid.called)
def test_do_node_show_by_instance_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'instance_uuid'
args.instance_uuid = True
args.fields = None
args.json = False
n_shell.do_node_show(client_mock, args)
client_mock.node.get_by_instance_uuid.assert_called_once_with(
'instance_uuid', fields=None)
# assert get() wasn't called
self.assertFalse(client_mock.node.get.called)
def test_do_node_show_by_space_node_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = ' '
args.instance_uuid = False
args.json = False
self.assertRaises(exceptions.CommandError,
n_shell.do_node_show,
client_mock, args)
def test_do_node_show_by_space_instance_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = ' '
args.instance_uuid = True
args.json = False
self.assertRaises(exceptions.CommandError,
n_shell.do_node_show,
client_mock, args)
def test_do_node_show_by_empty_node_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = ''
args.instance_uuid = False
args.json = False
self.assertRaises(exceptions.CommandError,
n_shell.do_node_show,
client_mock, args)
def test_do_node_show_by_empty_instance_uuid(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = ''
args.instance_uuid = True
args.json = False
self.assertRaises(exceptions.CommandError,
n_shell.do_node_show,
client_mock, args)
def test_do_node_show_fields(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.instance_uuid = False
args.fields = [['uuid', 'power_state']]
args.json = False
n_shell.do_node_show(client_mock, args)
client_mock.node.get.assert_called_once_with(
'node_uuid', fields=['uuid', 'power_state'])
def test_do_node_show_invalid_fields(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.instance_uuid = False
args.fields = [['foo', 'bar']]
args.json = False
self.assertRaises(exceptions.CommandError,
n_shell.do_node_show, client_mock, args)
def test_do_node_set_maintenance_true(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'true'
args.reason = 'reason'
n_shell.do_node_set_maintenance(client_mock, args)
client_mock.node.set_maintenance.assert_called_once_with(
'node_uuid', True, maint_reason='reason')
def test_do_node_set_maintenance_false(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'false'
# NOTE(jroll) None is the default. <3 mock.
args.reason = None
n_shell.do_node_set_maintenance(client_mock, args)
client_mock.node.set_maintenance.assert_called_once_with(
'node_uuid', False, maint_reason=None)
def test_do_node_set_maintenance_bad(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'yuck'
# NOTE(jroll) None is the default. <3 mock.
args.reason = None
self.assertRaises(exceptions.CommandError,
n_shell.do_node_set_maintenance, client_mock, args)
self.assertFalse(client_mock.node.set_maintenance.called)
def test_do_node_set_maintenance_false_with_reason_fails(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'false'
args.reason = 'reason'
self.assertRaises(exceptions.CommandError,
n_shell.do_node_set_maintenance,
client_mock, args)
def test_do_node_set_maintenance_on(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'on'
args.reason = 'reason'
n_shell.do_node_set_maintenance(client_mock, args)
client_mock.node.set_maintenance.assert_called_once_with(
'node_uuid', True, maint_reason='reason')
def test_do_node_set_maintenance_off(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'off'
# NOTE(jroll) None is the default. <3 mock.
args.reason = None
n_shell.do_node_set_maintenance(client_mock, args)
client_mock.node.set_maintenance.assert_called_once_with(
'node_uuid', False, maint_reason=None)
def test_do_node_set_maintenance_off_with_reason_fails(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.maintenance_mode = 'off'
args.reason = 'reason'
self.assertRaises(exceptions.CommandError,
n_shell.do_node_set_maintenance,
client_mock, args)
def _do_node_set_power_state_helper(self, power_state,
soft=False, timeout=None, error=False):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.power_state = power_state
args.soft = soft
args.power_timeout = timeout
if error:
client_mock.node = mock.MagicMock()
client_mock.node.set_power_state = mock.MagicMock()
client_mock.node.set_power_state.side_effect = ValueError("fake")
self.assertRaises(exc.CommandError,
n_shell.do_node_set_power_state,
client_mock, args)
else:
n_shell.do_node_set_power_state(client_mock, args)
client_mock.node.set_power_state.assert_called_once_with(
'node_uuid', power_state, soft, timeout=timeout)
def test_do_node_set_power_state_on(self):
self._do_node_set_power_state_helper('on')
def test_do_node_set_power_state_off(self):
self._do_node_set_power_state_helper('off')
def test_do_node_set_power_state_reboot(self):
self._do_node_set_power_state_helper('reboot')
def test_do_node_set_power_state_on_timeout(self):
self._do_node_set_power_state_helper('on', timeout=10)
def test_do_node_set_power_state_on_timeout_fail(self):
self._do_node_set_power_state_helper('on', timeout=0, error=True)
def test_do_node_set_power_state_off_timeout(self):
self._do_node_set_power_state_helper('off', timeout=10)
def test_do_node_set_power_state_reboot_timeout(self):
self._do_node_set_power_state_helper('reboot', timeout=10)
def test_do_node_set_power_state_soft_on_fail(self):
self._do_node_set_power_state_helper('on', soft=True, error=True)
def test_do_node_set_power_state_soft_off(self):
self._do_node_set_power_state_helper('off', soft=True)
def test_do_node_set_power_state_soft_reboot(self):
self._do_node_set_power_state_helper('reboot', soft=True)
def test_do_node_set_power_state_soft_on_timeout_fail(self):
self._do_node_set_power_state_helper('on', soft=True, timeout=10,
error=True)
def test_do_node_set_power_state_soft_off_timeout(self):
self._do_node_set_power_state_helper('off', soft=True, timeout=10)
def test_do_node_set_power_state_soft_reboot_timeout(self):
self._do_node_set_power_state_helper('reboot', soft=True, timeout=10)
def test_do_node_set_target_raid_config_file(self):
contents = '{"raid": "config"}'
with tempfile.NamedTemporaryFile(mode='w') as f:
f.write(contents)
f.flush()
node_manager_mock = mock.MagicMock(spec=['set_target_raid_config'])
client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock)
args = mock.MagicMock()
args.node = 'node_uuid'
args.target_raid_config = f.name
n_shell.do_node_set_target_raid_config(client_mock, args)
node_manager_mock.set_target_raid_config.assert_called_once_with(
'node_uuid', json.loads(contents))
def test_do_node_set_target_raid_config_string(self):
node_manager_mock = mock.MagicMock(spec=['set_target_raid_config'])
client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock)
target_raid_config_string = (
'{"logical_disks": [{"size_gb": 100, "raid_level": "1"}]}')
expected_target_raid_config_string = json.loads(
target_raid_config_string)
args = mock.MagicMock(node='node',
target_raid_config=target_raid_config_string)
n_shell.do_node_set_target_raid_config(client_mock, args)
node_manager_mock.set_target_raid_config.assert_called_once_with(
'node', expected_target_raid_config_string)
@mock.patch.object(commonutils, 'get_from_stdin', autospec=True)
def test_set_target_raid_config_stdin(self, stdin_read_mock):
node_manager_mock = mock.MagicMock(spec=['set_target_raid_config'])
client_mock = mock.MagicMock(spec=['node'], node=node_manager_mock)
target_raid_config_string = (
'{"logical_disks": [{"size_gb": 100, "raid_level": "1"}]}')
stdin_read_mock.return_value = target_raid_config_string
args_mock = mock.MagicMock(node='node',
target_raid_config='-')
expected_target_raid_config_string = json.loads(
target_raid_config_string)
n_shell.do_node_set_target_raid_config(client_mock, args_mock)
stdin_read_mock.assert_called_once_with('target_raid_config')
client_mock.node.set_target_raid_config.assert_called_once_with(
'node', expected_target_raid_config_string)
@mock.patch.object(commonutils, 'get_from_stdin', autospec=True)
def test_set_target_raid_config_stdin_exception(self, stdin_read_mock):
client_mock = mock.MagicMock()
stdin_read_mock.side_effect = exc.InvalidAttribute('bad')
args_mock = mock.MagicMock(node='node',
target_raid_config='-')
self.assertRaises(exc.InvalidAttribute,
n_shell.do_node_set_target_raid_config,
client_mock, args_mock)
stdin_read_mock.assert_called_once_with('target_raid_config')
self.assertFalse(client_mock.set_target_raid_config.called)
def test_do_node_validate(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
n_shell.do_node_validate(client_mock, args)
client_mock.node.validate.assert_called_once_with('node_uuid')
def test_do_node_vendor_passthru_with_args(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.http_method = 'POST'
args.method = 'method'
args.arguments = [['arg1=val1', 'arg2=val2']]
n_shell.do_node_vendor_passthru(client_mock, args)
client_mock.node.vendor_passthru.assert_called_once_with(
args.node, args.method, args={'arg1': 'val1', 'arg2': 'val2'},
http_method=args.http_method)
def test_do_node_vendor_passthru_without_args(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.http_method = 'POST'
args.method = 'method'
args.arguments = [[]]
n_shell.do_node_vendor_passthru(client_mock, args)
client_mock.node.vendor_passthru.assert_called_once_with(
args.node, args.method, args={}, http_method=args.http_method)
def test_do_node_set_provision_state_active(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'active'
args.config_drive = 'foo'
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'active', configdrive='foo', cleansteps=None)
self.assertFalse(client_mock.node.wait_for_provision_state.called)
def test_do_node_set_provision_state_active_wait(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'active'
args.config_drive = 'foo'
args.clean_steps = None
args.wait_timeout = 0
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'active', configdrive='foo', cleansteps=None)
client_mock.node.wait_for_provision_state.assert_called_once_with(
'node_uuid', expected_state='active', timeout=0,
poll_interval=v1_utils._LONG_ACTION_POLL_INTERVAL)
def test_do_node_set_provision_state_deleted(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'deleted'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'deleted', configdrive=None, cleansteps=None)
def test_do_node_set_provision_state_rebuild(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'rebuild'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'rebuild', configdrive=None, cleansteps=None)
def test_do_node_set_provision_state_not_active_fails(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'deleted'
args.config_drive = 'foo'
args.clean_steps = None
args.wait_timeout = None
self.assertRaises(exceptions.CommandError,
n_shell.do_node_set_provision_state,
client_mock, args)
self.assertFalse(client_mock.node.set_provision_state.called)
def test_do_node_set_provision_state_inspect(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'inspect'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'inspect', configdrive=None, cleansteps=None)
def test_do_node_set_provision_state_manage(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'manage'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'manage', configdrive=None, cleansteps=None)
def test_do_node_set_provision_state_provide(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'provide'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'provide', configdrive=None, cleansteps=None)
def test_do_node_set_provision_state_clean(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'clean'
args.config_drive = None
clean_steps = '[{"step": "upgrade", "interface": "deploy"}]'
args.clean_steps = clean_steps
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'clean', configdrive=None,
cleansteps=json.loads(clean_steps))
@mock.patch.object(commonutils, 'get_from_stdin', autospec=True)
def test_do_node_set_provision_state_clean_stdin(self, mock_stdin):
clean_steps = '[{"step": "upgrade", "interface": "deploy"}]'
mock_stdin.return_value = clean_steps
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'clean'
args.config_drive = None
args.clean_steps = '-'
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
mock_stdin.assert_called_once_with('clean steps')
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'clean', configdrive=None,
cleansteps=json.loads(clean_steps))
@mock.patch.object(commonutils, 'get_from_stdin', autospec=True)
def test_do_node_set_provision_state_clean_stdin_fails(self, mock_stdin):
mock_stdin.side_effect = exc.InvalidAttribute('bad')
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'clean'
args.config_drive = None
args.clean_steps = '-'
args.wait_timeout = None
self.assertRaises(exc.InvalidAttribute,
n_shell.do_node_set_provision_state,
client_mock, args)
mock_stdin.assert_called_once_with('clean steps')
self.assertFalse(client_mock.node.set_provision_state.called)
def test_do_node_set_provision_state_clean_file(self):
contents = '[{"step": "upgrade", "interface": "deploy"}]'
with tempfile.NamedTemporaryFile(mode='w') as f:
f.write(contents)
f.flush()
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'clean'
args.config_drive = None
args.clean_steps = f.name
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'clean', configdrive=None,
cleansteps=json.loads(contents))
def test_do_node_set_provision_state_clean_fails(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'clean'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
# clean_steps isn't specified
self.assertRaisesRegex(exceptions.CommandError,
'clean-steps.*must be specified',
n_shell.do_node_set_provision_state,
client_mock, args)
self.assertFalse(client_mock.node.set_provision_state.called)
def test_do_node_set_provision_state_not_clean_fails(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'deleted'
args.config_drive = None
clean_steps = '[{"step": "upgrade", "interface": "deploy"}]'
args.clean_steps = clean_steps
args.wait_timeout = None
# clean_steps specified but not cleaning
self.assertRaisesRegex(exceptions.CommandError,
'clean-steps.*only valid',
n_shell.do_node_set_provision_state,
client_mock, args)
self.assertFalse(client_mock.node.set_provision_state.called)
def test_do_node_set_provision_state_abort(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'abort'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = None
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'abort', configdrive=None, cleansteps=None)
def test_do_node_set_provision_state_adopt(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'adopt'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = 0
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'adopt', cleansteps=None, configdrive=None)
def test_do_node_set_provision_state_abort_no_wait(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'abort'
args.config_drive = None
args.clean_steps = None
args.wait_timeout = 0
self.assertRaisesRegex(exceptions.CommandError,
"not supported for provision state 'abort'",
n_shell.do_node_set_provision_state,
client_mock, args)
self.assertFalse(client_mock.node.set_provision_state.called)
self.assertFalse(client_mock.node.wait_for_provision_state.called)
def test_do_node_set_console_mode(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.enabled = 'true'
n_shell.do_node_set_console_mode(client_mock, args)
client_mock.node.set_console_mode.assert_called_once_with(
'node_uuid', True)
def test_do_node_set_console_mode_bad(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.enabled = 'yuck'
self.assertRaises(exceptions.CommandError,
n_shell.do_node_set_console_mode, client_mock, args)
self.assertFalse(client_mock.node.set_console_mode.called)
def test_do_node_set_boot_device(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.persistent = False
args.device = 'pxe'
n_shell.do_node_set_boot_device(client_mock, args)
client_mock.node.set_boot_device.assert_called_once_with(
'node_uuid', 'pxe', False)
def test_do_node_set_boot_device_persistent(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.persistent = True
args.device = 'disk'
n_shell.do_node_set_boot_device(client_mock, args)
client_mock.node.set_boot_device.assert_called_once_with(
'node_uuid', 'disk', True)
def test_do_node_get_boot_device(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.json = False
n_shell.do_node_get_boot_device(client_mock, args)
client_mock.node.get_boot_device.assert_called_once_with('node_uuid')
def test_do_node_inject_nmi(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
n_shell.do_node_inject_nmi(client_mock, args)
client_mock.node.inject_nmi.assert_called_once_with('node_uuid')
def test_do_node_get_supported_boot_devices(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.json = False
n_shell.do_node_get_supported_boot_devices(client_mock, args)
client_mock.node.get_supported_boot_devices.assert_called_once_with(
'node_uuid')
def _get_client_mock_args(self, node=None, associated=None,
maintenance=None, marker=None, limit=None,
sort_dir=None, sort_key=None, detail=False,
fields=None, provision_state=None, driver=None,
json=False, resource_class=None):
args = mock.MagicMock()
args.node = node
args.associated = associated
args.maintenance = maintenance
args.provision_state = provision_state
args.marker = marker
args.limit = limit
args.sort_dir = sort_dir
args.sort_key = sort_key
args.detail = detail
args.fields = fields
args.driver = driver
args.json = json
args.resource_class = resource_class
return args
def test_do_node_list(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args()
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(detail=False)
def test_do_node_list_detail(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(detail=True)
def test_do_node_list_provision_state(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(provision_state='wait call-back',
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(
provision_state='wait call-back',
detail=False)
def test_do_node_list_detail_provision_state(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(provision_state='wait call-back',
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(
provision_state='wait call-back',
detail=True)
def test_do_node_list_driver(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(driver='fake',
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(driver='fake',
detail=False)
def test_do_node_list_detail_driver(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(driver='fake',
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(driver='fake',
detail=True)
def test_do_node_list_resource_class(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(resource_class='foo',
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(resource_class='foo',
detail=False)
def test_do_node_list_detail_resource_class(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(resource_class='foo',
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(resource_class='foo',
detail=True)
def test_do_node_list_sort_key(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_key='created_at',
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(sort_key='created_at',
detail=False)
def test_do_node_list_wrong_sort_key(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_key='chassis_uuid',
detail=False)
self.assertRaises(exceptions.CommandError,
n_shell.do_node_list,
client_mock, args)
self.assertFalse(client_mock.node.list.called)
def test_do_node_list_detail_sort_key(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_key='created_at',
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(sort_key='created_at',
detail=True)
def test_do_node_list_detail_wrong_sort_key(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_key='chassis_uuid',
detail=True)
self.assertRaises(exceptions.CommandError,
n_shell.do_node_list,
client_mock, args)
self.assertFalse(client_mock.node.list.called)
def test_do_node_list_sort_dir(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_dir='desc',
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(sort_dir='desc',
detail=False)
def test_do_node_list_detail_sort_dir(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_dir='asc',
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(sort_dir='asc',
detail=True)
def test_do_node_list_wrong_sort_dir(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(sort_dir='abc',
detail=False)
self.assertRaises(exceptions.CommandError,
n_shell.do_node_list,
client_mock, args)
self.assertFalse(client_mock.node.list.called)
def test_do_node_list_maintenance(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(maintenance=True,
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(maintenance=True,
detail=False)
def test_do_node_list_detail_maintenance(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(maintenance=True,
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(maintenance=True,
detail=True)
def test_do_node_list_associated(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(associated=True,
detail=False)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(associated=True,
detail=False)
def test_do_node_list_detail_associated(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(associated=True,
detail=True)
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(associated=True,
detail=True)
def test_do_node_list_fields(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(fields=[['uuid', 'provision_state']])
n_shell.do_node_list(client_mock, args)
client_mock.node.list.assert_called_once_with(
fields=['uuid', 'provision_state'], detail=False)
def test_do_node_list_invalid_fields(self):
client_mock = mock.MagicMock()
args = self._get_client_mock_args(fields=[['foo', 'bar']])
self.assertRaises(exceptions.CommandError,
n_shell.do_node_list, client_mock, args)
def test_do_node_show_states(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.json = False
n_shell.do_node_show_states(client_mock, args)
client_mock.node.states.assert_called_once_with('node_uuid')
def test_do_node_port_list(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock)
n_shell.do_node_port_list(client_mock, args)
client_mock.node.list_ports.assert_called_once_with(
node_mock, detail=False)
def test_do_node_port_list_detail(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock, detail=True)
n_shell.do_node_port_list(client_mock, args)
client_mock.node.list_ports.assert_called_once_with(
node_mock, detail=True)
def test_do_node_port_list_sort_key(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock,
sort_key='created_at',
detail=False)
n_shell.do_node_port_list(client_mock, args)
client_mock.node.list_ports.assert_called_once_with(
node_mock, sort_key='created_at', detail=False)
def test_do_node_port_list_wrong_sort_key(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock,
sort_key='node_uuid',
detail=False)
self.assertRaises(exceptions.CommandError,
n_shell.do_node_port_list,
client_mock, args)
self.assertFalse(client_mock.node.list_ports.called)
def test_do_node_port_list_detail_sort_key(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock,
sort_key='created_at',
detail=True)
n_shell.do_node_port_list(client_mock, args)
client_mock.node.list_ports.assert_called_once_with(
node_mock, sort_key='created_at', detail=True)
def test_do_node_port_list_detail_wrong_sort_key(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock,
sort_key='node_uuid',
detail=True)
self.assertRaises(exceptions.CommandError,
n_shell.do_node_port_list,
client_mock, args)
self.assertFalse(client_mock.node.list_ports.called)
def test_do_node_port_list_sort_dir(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock, sort_dir='desc',
detail=False)
n_shell.do_node_port_list(client_mock, args)
client_mock.node.list_ports.assert_called_once_with(
node_mock, sort_dir='desc', detail=False)
def test_do_node_port_list_wrong_sort_dir(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock, sort_dir='abc',
detail=False)
self.assertRaises(exceptions.CommandError,
n_shell.do_node_port_list,
client_mock, args)
self.assertFalse(client_mock.node.list_ports.called)
def test_do_node_port_list_fields(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock,
fields=[['uuid', 'address']])
n_shell.do_node_port_list(client_mock, args)
client_mock.node.list_ports.assert_called_once_with(
node_mock, fields=['uuid', 'address'], detail=False)
def test_do_node_port_list_invalid_fields(self):
client_mock = mock.MagicMock()
node_mock = mock.MagicMock(spec_set=[])
args = self._get_client_mock_args(node=node_mock,
fields=[['foo', 'bar']])
self.assertRaises(exceptions.CommandError,
n_shell.do_node_port_list, client_mock, args)
def test_do_node_get_vendor_passthru_methods(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
n_shell.do_node_get_vendor_passthru_methods(client_mock, args)
client_mock.node.get_vendor_passthru_methods.assert_called_once_with(
'node_uuid')
def test_do_node_vif_list(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
n_shell.do_node_vif_list(client_mock, args)
client_mock.node.vif_list.assert_called_once_with(
'node_uuid')
def test_do_node_vif_attach(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.vif_id = 'aaa-aaa'
n_shell.do_node_vif_attach(client_mock, args)
client_mock.node.vif_attach.assert_called_once_with(
'node_uuid', 'aaa-aaa')
def test_do_node_vif_attach_custom_fields(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.vif_id = 'aaa-aaa'
args.vif_info = ['aaa=bbb', 'ccc=ddd']
n_shell.do_node_vif_attach(client_mock, args)
client_mock.node.vif_attach.assert_called_once_with(
'node_uuid', 'aaa-aaa', aaa='bbb', ccc='ddd')
def test_do_node_vif_detach(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.vif_id = 'aaa-aaa'
n_shell.do_node_vif_detach(client_mock, args)
client_mock.node.vif_detach.assert_called_once_with(
'node_uuid', 'aaa-aaa')