[SRE-1128] Adds generic handler for wrapping executeStep

JupiterOne · Jul 27, 2023 · 38aeb99 · 38aeb99
1 parent eebc826
commit 38aeb99
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/packages/integration-sdk-runtime/src/execution/dependencyGraph.ts b/packages/integration-sdk-runtime/src/execution/dependencyGraph.ts
@@ -272,7 +272,7 @@ export function executeStepDependencyGraph<
                 },
                 operation: () =>
                   stepWrapper
-                    ? stepWrapper(step, () => executeStep(step))
+                    ? stepWrapper(step, async () => executeStep(step))
                     : executeStep(step),
               }).catch(handleUnexpectedError),
             );

diff --git a/packages/integration-sdk-runtime/src/execution/executeIntegration.ts b/packages/integration-sdk-runtime/src/execution/executeIntegration.ts
@@ -15,7 +15,6 @@ import {
   Relationship,
   StepExecutionContext,
   StepResultStatus,
-  StepWrapperFunction,
 } from '@jupiterone/integration-sdk-core';
 
 import {