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
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)
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_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_provision_state_manage(self):
client_mock = mock.MagicMock()
args = mock.MagicMock()
args.node = 'node_uuid'
args.provision_state = 'manage'
args.config_drive = 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)
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
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)
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
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_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
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_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
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_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(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_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
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))
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
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))
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 = "-"
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)
)
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_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))
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_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))
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
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_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
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))
