support async test

browserify · Feb 27, 2021 · 6adca6c · 6adca6c
1 parent 7a0ed07
commit 6adca6c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/tests/events-once.js b/tests/events-once.js
@@ -175,7 +175,7 @@ function onceWithEventTargetError() {
   });
 }
 
-Promise.all([
+module.exports = Promise.all([
   onceAnEvent(),
   onceAnEventWithTwoArgs(),
   catchesErrors(),
@@ -184,7 +184,4 @@ Promise.all([
   onceWithEventTarget(),
   onceWithEventTargetTwoArgs(),
   onceWithEventTargetError()
-]).catch(function (err) {
-  console.error(err.stack)
-  process.exit(1)
-});
+]);
diff --git a/tests/index.js b/tests/index.js
@@ -14,7 +14,15 @@ var require = function(file) {
         t.on('end', function () { delete common.test; });
         common.test = t;
 
-        try { orig_require(file); } catch (err) { t.fail(err); }
+        try {
+          var exp = orig_require(file);
+          if (exp && exp.then) {
+            exp.then(function () { t.end(); }, t.fail);
+            return;
+          }
+        } catch (err) {
+          t.fail(err);
+        }
         t.end();
     });
 };