node: mark promises as handled as soon as possible

Fixes: https://github.com/nodejs/io.js/issues/1912
PR-URL: https://github.com/nodejs/io.js/pull/1952
Reviewed-By: Domenic Denicola <d@domenic.me>
Reviewed-By: Petka Antonov <petka_antonov@hotmail.com>

diff --git a/src/node.js b/src/node.js
index 25164784cfebbe692a74b78464c9e8862df3073b..292c7b22133ee6f62f41831a840b6b26cf0f9cb0 100644 (file)
--- a/src/node.js
+++ b/src/node.js
@@ -515,8 +515,12 @@
       var hasBeenNotified = hasBeenNotifiedProperty.get(promise);
       if (hasBeenNotified !== undefined) {
         hasBeenNotifiedProperty.delete(promise);
-        if (hasBeenNotified === true)
-          process.emit('rejectionHandled', promise);
+        if (hasBeenNotified === true) {
+          process.nextTick(function() {
+            process.emit('rejectionHandled', promise);
+          });
+        }
+
       }
     }
 
@@ -524,9 +528,7 @@
       if (event === promiseRejectEvent.unhandled)
         unhandledRejection(promise, reason);
       else if (event === promiseRejectEvent.handled)
-        process.nextTick(function() {
-          rejectionHandled(promise);
-        });
+        rejectionHandled(promise);
       else
         NativeModule.require('assert').fail('unexpected PromiseRejectEvent');
     });

diff --git a/test/parallel/test-promises-unhandled-rejections.js b/test/parallel/test-promises-unhandled-rejections.js
index 9a186de8dfec7937cc1863c6ccc102b640c795dc..e5b3e6f35c0d45abd6790ba125e8728dd2e4a1dd 100644 (file)
--- a/test/parallel/test-promises-unhandled-rejections.js
+++ b/test/parallel/test-promises-unhandled-rejections.js
@@ -275,6 +275,21 @@ asyncTest('Attaching a promise catch in a process.nextTick is soon enough to' +
   });
 });
 
+asyncTest('While inside setImmediate, catching a rejected promise derived ' +
+          'from returning a rejected promise in a fulfillment handler ' +
+          'prevents unhandledRejection', function(done) {
+  onUnhandledFail(done);
+
+  setImmediate(function() {
+    // reproduces on first tick and inside of setImmediate
+    Promise
+      .resolve('resolve')
+      .then(function() {
+        return Promise.reject('reject');
+      }).catch(function(e) {});
+  });
+});
+
 // State adapation tests
 asyncTest('catching a promise which is asynchronously rejected (via' +
           'resolution to an asynchronously-rejected promise) prevents' +