From eac4058d9a582066a818e1a0ccd53f91f0082702 Mon Sep 17 00:00:00 2001 From: Tang Chen Date: Tue, 27 Sep 2016 18:55:11 +0800 Subject: Add missing unit tests for provision state commands There are several missing unit tests for provision state commands. This patch adds them. Change-Id: Ie5b08f682f3cb8fbc5016cb02e689bb1934798f3 --- .../tests/unit/osc/v1/test_baremetal_node.py | 185 +++++++++++++++++++++ 1 file changed, 185 insertions(+) diff --git a/ironicclient/tests/unit/osc/v1/test_baremetal_node.py b/ironicclient/tests/unit/osc/v1/test_baremetal_node.py index c09f541..31dd0d5 100644 --- a/ironicclient/tests/unit/osc/v1/test_baremetal_node.py +++ b/ironicclient/tests/unit/osc/v1/test_baremetal_node.py @@ -15,6 +15,7 @@ # import copy +import json from unittest import mock from osc_lib.tests import utils as oscutils @@ -36,6 +37,29 @@ class TestBaremetal(baremetal_fakes.TestBaremetal): self.baremetal_mock.reset_mock() +class TestAbort(TestBaremetal): + def setUp(self): + super(TestAbort, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.AbortBaremetalNode(self.app, None) + + def test_abort(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'abort'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'abort', cleansteps=None, configdrive=None, + deploysteps=None, rescue_password=None) + + class TestAdopt(TestBaremetal): def setUp(self): super(TestAdopt, self).setUp() @@ -109,6 +133,167 @@ class TestAdopt(TestBaremetal): poll_interval=2, timeout=0) +class TestClean(TestBaremetal): + def setUp(self): + super(TestClean, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.CleanBaremetalNode(self.app, None) + + def test_clean_without_steps(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'clean'), + ] + + self.assertRaises(oscutils.ParserException, self.check_parser, + self.cmd, arglist, verifylist) + + def test_clean_with_steps(self): + steps_dict = { + "clean_steps": [{ + "interface": "raid", + "step": "create_configuration", + "args": {"create_nonroot_volumes": False} + }, { + "interface": "deploy", + "step": "erase_devices" + }] + } + steps_json = json.dumps(steps_dict) + + arglist = ['--clean-steps', steps_json, 'node_uuid'] + verifylist = [ + ('clean_steps', steps_json), + ('provision_state', 'clean'), + ('node', 'node_uuid'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'clean', cleansteps=steps_dict, configdrive=None, + deploysteps=None, rescue_password=None) + + +class TestInspect(TestBaremetal): + def setUp(self): + super(TestInspect, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.InspectBaremetalNode(self.app, None) + + def test_inspect(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'inspect'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'inspect', cleansteps=None, configdrive=None, + deploysteps=None, rescue_password=None) + + +class TestManage(TestBaremetal): + def setUp(self): + super(TestManage, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.ManageBaremetalNode(self.app, None) + + def test_manage(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'manage'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'manage', cleansteps=None, configdrive=None, + deploysteps=None, rescue_password=None) + + +class TestProvide(TestBaremetal): + def setUp(self): + super(TestProvide, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.ProvideBaremetalNode(self.app, None) + + def test_provide(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'provide'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'provide', cleansteps=None, configdrive=None, + deploysteps=None, rescue_password=None) + + +class TestRebuild(TestBaremetal): + def setUp(self): + super(TestRebuild, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.RebuildBaremetalNode(self.app, None) + + def test_rebuild(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'rebuild'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'rebuild', cleansteps=None, configdrive=None, + deploysteps=None, rescue_password=None) + + +class TestUndeploy(TestBaremetal): + def setUp(self): + super(TestUndeploy, self).setUp() + + # Get the command object to test + self.cmd = baremetal_node.UndeployBaremetalNode(self.app, None) + + def test_undeploy(self): + arglist = ['node_uuid'] + verifylist = [ + ('node', 'node_uuid'), + ('provision_state', 'deleted'), + ] + + parsed_args = self.check_parser(self.cmd, arglist, verifylist) + + self.cmd.take_action(parsed_args) + + self.baremetal_mock.node.set_provision_state.assert_called_once_with( + 'node_uuid', 'deleted', cleansteps=None, configdrive=None, + deploysteps=None, rescue_password=None) + + class TestBootdeviceSet(TestBaremetal): def setUp(self): super(TestBootdeviceSet, self).setUp() -- cgit v1.2.1