From 9f16226d1e184723181cf91800d78a2b44fff201 Mon Sep 17 00:00:00 2001
From: Dan <78874812+ymwjbxxq@users.noreply.github.com>
Date: Thu, 25 Feb 2021 14:02:50 +0100
Subject: [PATCH] fix(stepfunction): add two extra methods to track (#448)

Co-authored-by: Daniele Frasca <daniele.frasca@prosiebensat1digital.de>
Co-authored-by: Ran Ribenzaft <ranrib@users.noreply.github.com>
---
 src/events/aws_sdk.js | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/src/events/aws_sdk.js b/src/events/aws_sdk.js
index 9079f640..56036d14 100644
--- a/src/events/aws_sdk.js
+++ b/src/events/aws_sdk.js
@@ -575,14 +575,18 @@ const stepFunctionsEventCreator = {
      */
     patchInput(request, event) {
         switch (request.operation) {
-        case 'startExecution': {
+        case 'startExecution':
             initializeStepsDict(request, 'input', event);
             break;
-        }
-        case 'sendTaskSuccess': {
+        case 'stopExecution':
+            initializeStepsDict(request, 'input', event);
+            break;
+        case 'sendTaskSuccess':
+            initializeStepsDict(request, 'output', event);
+            break;
+        case 'sendTaskFailure':
             initializeStepsDict(request, 'output', event);
             break;
-        }
         default:
             break;
         }