fix pulseWhileDone promise handling

Small an oversight from the Bluebird -> Promise refactor. Fix #1695
npm · Aug 21, 2020 · 10fcff7 · 10fcff7
1 parent 1faa5b3
commit 10fcff7
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/lib/utils/pulse-till-done.js b/lib/utils/pulse-till-done.js
@@ -25,18 +25,17 @@ module.exports = function (prefix, cb) {
     cb(er, ...args)
   }
 }
-module.exports.withPromise = pulseWhile
 
-function pulseWhile (prefix, promise) {
+const pulseWhile = async (prefix, promise) => {
   if (!promise) {
     promise = prefix
     prefix = ''
   }
   pulseStart(prefix)
-  return Promise.resolve(promise)
-    .then(() => pulseStop())
-    .catch(er => {
-      pulseStop()
-      throw er
-    })
+  try {
+    return await promise
+  } finally {
+    pulseStop()
+  }
 }
+module.exports.withPromise = pulseWhile