Merge pull request #48 from Adslot/async-support

Support async migration functions

Author: bchew

lib/migrant.js

@@ -1,6 +1,7 @@
 var assert = require('assert'),
   path = require('path'),
   fs = require('fs'),
+  util = require('util'),
   async = require('async');
 
 var MetaDecorator = require('./meta/');
@@ -69,12 +70,13 @@ Migrant.prototype.migrateSingle = function(mode, filename, cb) {
   var self = this,
     migration = require(path.join(this.opts.dir, filename)),
     started = Date.now();
+  const migrateFn = util.types.isAsyncFunction(migration[mode]) ? util.callbackify(migration[mode]) : migration[mode];
 
   this.log(filename, mode);
   let msg;
   async.series([
     function(cbInner) {
-      migration[mode](function(err, res) {
+      migrateFn(function(err, res) {
         cbInner(err, res)
         // to check if the callback set correctly on each file.
         // if there is an error it will be written in the log file and will throw an error.

test/data/migrations/20150701-eee.js

@@ -0,0 +1,11 @@
+const { setTimeout } = require('timers/promises');
+
+exports.up = async () => {
+  await setTimeout(10);
+  return 'eee: up';
+};
+
+exports.down = async () => {
+  await setTimeout(10);
+  return 'eee: down';
+};

test/migrant_test.js

@@ -35,7 +35,13 @@ describe('index', function() {
       meta.set(null, noon)
       m.list('up', 10, function(err, list) {
         if (err) return done(err);
-        assert.deepEqual(list, ['20141028-aaa.js', '20141030-bbb.js', '20141031-ccc.js', '20150601-ddd.js']);
+        assert.deepEqual(list, [
+          '20141028-aaa.js',
+          '20141030-bbb.js',
+          '20141031-ccc.js',
+          '20150601-ddd.js',
+          '20150701-eee.js',
+        ]);
         done()
       });
     });
@@ -44,7 +50,12 @@ describe('index', function() {
       meta.set({migrations: [{filename: '20141028-aaa.js', ts: 11}]}, noon)
       m.list('up', 10, function(err, list) {
         if (err) return done(err);
-        assert.deepEqual(list, ['20141030-bbb.js', '20141031-ccc.js','20150601-ddd.js']);
+        assert.deepEqual(list, [
+          '20141030-bbb.js',
+          '20141031-ccc.js',
+          '20150601-ddd.js',
+          '20150701-eee.js',
+        ]);
         done()
       });
     });
@@ -138,7 +149,8 @@ describe('index', function() {
         assert.deepEqual(res, [
           ['aaa: up', undefined],
           ['bbb: up', undefined],
-          ['ccc: up', undefined]
+          ['ccc: up', undefined],
+          ['eee: up', undefined]
         ]);
 
         m.meta.get(function(err, data) {
@@ -152,7 +164,8 @@ describe('index', function() {
             '20150601-ddd.js',
             '20141028-aaa.js',
             '20141030-bbb.js',
-            '20141031-ccc.js'
+            '20141031-ccc.js',
+            '20150701-eee.js'
           ])
           done()
         })