Add 'rebuild' option to node-set-provision-state

This patch set add 'rebuild' option to "ironic node-set-provision-state"
command.

Change-Id: I9e9268aa61b6f48697147e3c2cff519ea1c265c2
Closes-Bug: #1310446
This commit is contained in:
Yuiko Takada
2014-09-12 09:55:21 +09:00
committed by Ruby Loo
parent 747a9e8014
commit 2d071aee6d
2 changed files with 32 additions and 2 deletions

View File

@@ -96,6 +96,36 @@ class NodeShellTest(utils.BaseTestCase):
}
client_mock.node.vendor_passthru.assert_called_once_with(**kwargs)
def test_do_node_set_provision_state_active(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'active'
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'active')
def test_do_node_set_provision_state_deleted(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'deleted'
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'deleted')
def test_do_node_set_provision_state_rebuild(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'rebuild'
n_shell.do_node_set_provision_state(client_mock, args)
client_mock.node.set_provision_state.assert_called_once_with(
'node_uuid', 'rebuild')
def test_do_node_set_boot_device(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()

View File

@@ -266,8 +266,8 @@ def do_node_set_power_state(cc, args):
@cliutils.arg(
'provision_state',
metavar='<provision state>',
choices=['active', 'deleted'],
help="Supported states: 'active' or 'deleted'")
choices=['active', 'deleted', 'rebuild'],
help="Supported states: 'active' or 'deleted' or 'rebuild'")
def do_node_set_provision_state(cc, args):
"""Provision an instance on, or delete an instance from a node."""
cc.node.set_provision_state(args.node, args.provision_state)