fix(stepfunction): add two extra methods to track (#448)

Co-authored-by: Daniele Frasca <[email protected]> Co-authored-by: Ran Ribenzaft <[email protected]>
epsagon · Feb 25, 2021 · 9f16226 · 9f16226
1 parent a97b972
commit 9f16226
Showing 1 changed file with 8 additions and 4 deletions.
diff --git 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;
         }