1 files changed, 13 insertions, 5 deletions
diff --git a/test/unit/nodeify-test.js b/test/unit/nodeify-test.js
index a14d34338..c7b127889 100644
--- a/test/unit/nodeify-test.js
+++ b/test/unit/nodeify-test.js
@@ -1,22 +1,30 @@
 const assert = require('assert')
 const nodeify = require('../../app/scripts/lib/nodeify')
 
-describe('nodeify', function() {
-
+describe('nodeify', function () {
   var obj = {
     foo: 'bar',
     promiseFunc: function (a) {
       var solution = this.foo + a
       return Promise.resolve(solution)
-    }
+    },
   }
 
-  it('should retain original context', function(done) {
-    var nodified = nodeify(obj.promiseFunc).bind(obj)
+  it('should retain original context', function (done) {
+    var nodified = nodeify(obj.promiseFunc, obj)
     nodified('baz', function (err, res) {
       assert.equal(res, 'barbaz')
       done()
     })
   })
 
+  it('should allow the last argument to not be a function', function (done) {
+    const nodified = nodeify(obj.promiseFunc, obj)
+    try {
+      nodified('baz')
+      done()
+    } catch (err) {
+      done(new Error('should not have thrown if the last argument is not a function'))
+    }
+  })
 })