feat(@embark/testing): introduce proper request2 api for async/await

embarklabs · Jan 15, 2020 · c947517 · c947517
1 parent dc9171a
commit c947517
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/packages/utils/testing/src/lib/events.js b/packages/utils/testing/src/lib/events.js
@@ -40,7 +40,18 @@ class Events {
 
   request2(cmd, ...args) {
     assert(this.commandHandlers[cmd], `command handler for ${cmd} not registered`);
-    this.commandHandlers[cmd](...args);
+    return new Promise((resolve, reject) => {
+      args.push((err, ...res) => {
+        if (err) {
+          return reject(err);
+        }
+        if (res.length && res.length > 1) {
+          return resolve(res);
+        }
+        return resolve(res[0]);
+      });
+      this.commandHandlers[cmd](...args);
+    });
   }
 }