diff options
Diffstat (limited to 'ironic/tests/unit/drivers/modules/xclarity/test_power.py')
-rw-r--r-- | ironic/tests/unit/drivers/modules/xclarity/test_power.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/ironic/tests/unit/drivers/modules/xclarity/test_power.py b/ironic/tests/unit/drivers/modules/xclarity/test_power.py index 6cec8cd31..b01ee6623 100644 --- a/ironic/tests/unit/drivers/modules/xclarity/test_power.py +++ b/ironic/tests/unit/drivers/modules/xclarity/test_power.py @@ -26,6 +26,7 @@ import mock from oslo_utils import importutils +from ironic.common import exception from ironic.common import states from ironic.conductor import task_manager from ironic.drivers.modules.xclarity import common @@ -84,9 +85,9 @@ class XClarityPowerDriverTestCase(db_base.DbTestCase): if 'ironic.drivers.modules.xclarity' in sys.modules: six.moves.reload_module( sys.modules['ironic.drivers.modules.xclarity']) - ex = common.XClarityError('E') + ex = exception.XClarityError('E') mock_xc_client.return_value.get_node_power_status.side_effect = ex - self.assertRaises(common.XClarityError, + self.assertRaises(exception.XClarityError, task.driver.power.get_power_state, task) @@ -121,9 +122,9 @@ class XClarityPowerDriverTestCase(db_base.DbTestCase): if 'ironic.drivers.modules.xclarity' in sys.modules: six.moves.reload_module( sys.modules['ironic.drivers.modules.xclarity']) - ex = common.XClarityError('E') + ex = exception.XClarityError('E') mock_xc_client.return_value.set_node_power_status.side_effect = ex - self.assertRaises(common.XClarityError, + self.assertRaises(exception.XClarityError, task.driver.power.set_power_state, task, states.POWER_OFF) |