knex/test/tape/harness.js

'use strict';
var tape = require('tape');
var Promise = require('bluebird');
var debug = require('debug')('knex:tests');

module.exports = function(tableName, knex) {
  return function(name, dialects, cb) {
    if (arguments.length === 2) {
      cb = dialects;
    } else {
      if (!Array.isArray(dialects)) {
        dialects = [dialects];
      }
      if (dialects.indexOf(knex.client.dialect) === -1) {
        debug('Skipping dialect ' + knex.client.dialect + ' for test ' + name);
        return;
      }
    }

    return tape(name, function(t) {
      var hasPlanned = false;

      t.on('plan', function() {
        hasPlanned = true;
      });

      var disposable = Promise.resolve(true).disposer(function() {
        return knex.truncate(tableName).finally(function() {
          t.end();
        });
      });

      Promise.using(disposable, function() {
        var val = cb(t);
        if (val && typeof val.then === 'function') {
          return val.catch(function(err) {
            t.error(err);
          });
        } else {
          t.error(new Error('A promise should be returned to test ' + name));
          t.end();
        }
      });
    });
  };
};