def test_list_executions(self): scenario = utils.MistralScenario(context=self.context) return_executions_list = scenario._list_executions() self.assertEqual(return_executions_list, self.clients("mistral").executions.list.return_value) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.list_executions")
def test_create_execution_with_params(self): scenario = utils.MistralScenario(context=self.context) mock_wait_for_status = self.mock_wait_for_status.mock wf_name = "fake_wf_name" mock_create_exec = self.clients("mistral").executions.create self.assertEqual(mock_wait_for_status.return_value, scenario._create_execution(wf_name, **PARAMS_EXAMPLE)) mock_create_exec.assert_called_once_with(wf_name, workflow_input=None, namespace="", **PARAMS_EXAMPLE) args, kwargs = mock_wait_for_status.call_args self.assertEqual(mock_create_exec.return_value, args[0]) self.assertEqual(["ERROR"], kwargs["failure_statuses"]) self.assertEqual(["SUCCESS"], kwargs["ready_statuses"]) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.create_execution") args, kwargs = mock_wait_for_status.call_args self.assertEqual(mock_create_exec.return_value, args[0]) self.assertEqual(["ERROR"], kwargs["failure_statuses"]) self.assertEqual(["SUCCESS"], kwargs["ready_statuses"]) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.create_execution")
def test_delete_workbook(self): scenario = utils.MistralScenario(context=self.context) scenario._delete_workbook("wb_name") self.clients("mistral").workbooks.delete.assert_called_once_with( "wb_name", namespace="") self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.delete_workbook")
def test_create_workbook(self): definition = "version: \"2.0\"\nname: wb" scenario = utils.MistralScenario(context=self.context) self.assertEqual( self.clients("mistral").workbooks.create.return_value, scenario._create_workbook(definition)) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.create_workbook")
def test_list_workbooks(self): scenario = utils.MistralScenario(context=self.context) return_wbs_list = scenario._list_workbooks() self.assertEqual( self.clients("mistral").workbooks.list.return_value, return_wbs_list) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.list_workbooks")
def test_delete_execution(self): scenario = utils.MistralScenario(context=self.context) execution = fakes.FakeMistralClient().execution.create() scenario._delete_execution(execution) self.clients("mistral").executions.delete.assert_called_once_with( execution.id) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.delete_execution")
def test_delete_workflow(self): wf_identifier = "wf_identifier" namespace = "delete_wf_test" scenario = utils.MistralScenario(context=self.context) scenario._delete_workflow(wf_identifier, namespace=namespace) self.clients("mistral").workflows.delete.assert_called_once_with( wf_identifier, namespace=namespace) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.delete_workflow")
def test_create_execution_with_input(self): scenario = utils.MistralScenario(context=self.context) mock_wait_for_status = self.mock_wait_for_status.mock wf_name = "fake_wf_name" mock_create_exec = self.clients("mistral").executions.create self.assertEqual( mock_wait_for_status.return_value, scenario._create_execution(wf_name, wf_input=str(INPUT_EXAMPLE))) mock_create_exec.assert_called_once_with(wf_name, workflow_input=INPUT_EXAMPLE, namespace="")
def test_create_workflow(self): scenario = utils.MistralScenario(context=self.context) definition = """ wf: type: direct tasks: task1: action: std.noop """ self.assertEqual( self.clients("mistral").workflows.create.return_value, scenario._create_workflow(definition)) self._test_atomic_action_timer(scenario.atomic_actions(), "mistral.create_workflow")