fix: correct run-state-machine state resource when waiting on a heart…

…best
wmfs · Jun 26, 2018 · b4efc6d · b4efc6d
1 parent 58455a2
commit b4efc6d
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/lib/plugin/components/state-resources/run-state-machine/index.js b/lib/plugin/components/state-resources/run-state-machine/index.js
@@ -36,11 +36,9 @@ class RunStateMachine {
         cause: execDesc.errorMessage
       })
     } else {
-      context.sendTaskHeartbeat(execDesc.ctx, (err, executionDescription) => {
-        if (err) throw new Error(err)
-        executionDescription.currentResource = execDesc.currentResource
-        done(executionDescription)
-      })
+      const executionDescription = await context.sendTaskHeartbeat(execDesc.ctx)
+      executionDescription.currentResource = execDesc.currentResource
+      done(executionDescription)
     }
   } // run
 } // class RunStateMachine