diff --git a/moto/swf/models/workflow_execution.py b/moto/swf/models/workflow_execution.py index f53c1028..c22d3e02 100644 --- a/moto/swf/models/workflow_execution.py +++ b/moto/swf/models/workflow_execution.py @@ -286,7 +286,7 @@ class WorkflowExecution(object): # handle each decision separately, in order for decision in decisions: decision_type = decision["decisionType"] - attributes_key = "{}EventAttributes".format(decapitalize(decision_type)) + attributes_key = "{}DecisionAttributes".format(decapitalize(decision_type)) attributes = decision.get(attributes_key, {}) if decision_type == "CompleteWorkflowExecution": self.complete(event_id, attributes.get("result")) diff --git a/tests/test_swf/test_decision_tasks.py b/tests/test_swf/test_decision_tasks.py index 6ccd0780..a07eecbf 100644 --- a/tests/test_swf/test_decision_tasks.py +++ b/tests/test_swf/test_decision_tasks.py @@ -144,7 +144,7 @@ def test_respond_decision_task_completed_with_complete_workflow_execution(): decisions = [{ "decisionType": "CompleteWorkflowExecution", - "completeWorkflowExecutionEventAttributes": {"result": "foo bar"} + "completeWorkflowExecutionDecisionAttributes": {"result": "foo bar"} }] resp = conn.respond_decision_task_completed(task_token, decisions=decisions) resp.should.be.none @@ -240,7 +240,7 @@ def test_respond_decision_task_completed_with_fail_workflow_execution(): decisions = [{ "decisionType": "FailWorkflowExecution", - "failWorkflowExecutionEventAttributes": {"reason": "my rules", "details": "foo"} + "failWorkflowExecutionDecisionAttributes": {"reason": "my rules", "details": "foo"} }] resp = conn.respond_decision_task_completed(task_token, decisions=decisions) resp.should.be.none