def test__get_deploy_steps_unsorted(self, mock_mgt_steps, mock_power_steps, mock_deploy_steps): mock_deploy_steps.return_value = [self.deploy_end, self.deploy_start, self.deploy_middle] with task_manager.acquire( self.context, self.node.uuid, shared=False) as task: steps = conductor_steps._get_deployment_steps(task, enabled=False, sort=False) self.assertEqual(mock_deploy_steps.return_value, steps) mock_mgt_steps.assert_called_once_with(mock.ANY, task) mock_power_steps.assert_called_once_with(mock.ANY, task) mock_deploy_steps.assert_called_once_with(mock.ANY, task)
def test__get_deployment_steps(self, mock_mgt_steps, mock_power_steps, mock_deploy_steps): # Test getting deploy steps, with one driver returning None, two # conflicting priorities, and asserting they are ordered properly. mock_power_steps.return_value = [self.power_disable, self.power_one] mock_deploy_steps.return_value = [ self.deploy_start, self.deploy_middle, self.deploy_end] expected = self.deploy_steps + [self.power_disable] with task_manager.acquire( self.context, self.node.uuid, shared=False) as task: steps = conductor_steps._get_deployment_steps(task, enabled=False) self.assertEqual(expected, steps) mock_mgt_steps.assert_called_once_with(mock.ANY, task) mock_power_steps.assert_called_once_with(mock.ANY, task) mock_deploy_steps.assert_called_once_with(mock.ANY, task)