fix(check): fix check via API not passing results to the callback

Signed-off-by: RandomSeeded <[email protected]>

Author: RandomSeeded

lib/commands/on-complete.js

@@ -1,9 +1,10 @@
 var assert = require('assert');
 var log = require('db-migrate-shared').log;
 
-module.exports = function (migrator, internals, callback, originalErr) {
+module.exports = function (migrator, internals, callback, originalErr, results) {
   if (typeof callback !== 'function') {
     originalErr = originalErr || callback;
+    results = results || originalErr;
   }
 
   migrator.driver.close(function (err) {
@@ -22,7 +23,7 @@ module.exports = function (migrator, internals, callback, originalErr) {
     }
 
     if (typeof callback === 'function') {
-      callback();
+      callback(null, results);
     }
   });
 };

test/on_complete_test.js

@@ -0,0 +1,27 @@
+var Code = require('code');
+var Lab = require('lab');
+var lab = (exports.lab = Lab.script());
+var onComplete = require('../lib/commands/on-complete');
+
+lab.experiment('on-complete', function () {
+  lab.test('should invoke the callback with the results', function (done) {
+    var migratorMock = {
+      driver: {
+        close: function (cb) {
+          cb();
+        }
+      }
+    };
+    var internalsMock = {
+      argv: { }
+    };
+
+    var resultsPassedToCallback = 'callback should be invoked with results';
+    var testCallback = function (err, res) {
+      Code.expect(err).to.equal(null);
+      Code.expect(res).to.equal(resultsPassedToCallback);
+      done();
+    };
+    onComplete(migratorMock, internalsMock, testCallback, null, resultsPassedToCallback);
+  });
+});