Improved denodeify

Author: TehShrike

lib/node-extensions.js

@@ -22,7 +22,10 @@ Promise.denodeify = function (fn, argumentCount) {
         if (err) reject(err)
         else resolve(res)
       })
-      fn.apply(self, args)
+      var res = fn.apply(self, args)
+      if (res && res.then) {
+        resolve(res)
+      }
     })
   }
 }

test/extensions-tests.js

@@ -55,6 +55,20 @@ describe('extensions', function () {
           done()
         })
     })
+    it('resolves correctly when the wrapped function returns a promise anyway', function (done) {
+      function wrap(val, key, callback) {
+        return new Promise(function(resolve, reject) {
+          resolve({val: val, key: key})
+        })
+      }
+      var pwrap = Promise.denodeify(wrap)
+      pwrap(sentinel, 'foo')
+        .then(function (wrapper) {
+          assert(wrapper.val === sentinel)
+          assert(wrapper.key === 'foo')
+          done()
+        })
+    })
   })
   describe('Promise.nodeify(fn)', function () {
     it('converts a promise returning function into a callback function', function (done) {