runner.js

'use strict';
var util = require('util');
var EventEmitter = require('events').EventEmitter;
var Promise = require('bluebird');
var Test = require('./test');

function Runner(opts) {
	if (!(this instanceof Runner)) {
		return new Runner(opts);
	}

	EventEmitter.call(this);

	this.results = [];

	this.stats = {
		failCount: 0,
		testCount: 0
	};

	this.tests = {
		concurrent: [],
		serial: [],
		before: [],
		after: []
	};

	this._assertModule = null;
}

util.inherits(Runner, EventEmitter);
module.exports = Runner;

Runner.prototype.addTest = function (title, cb) {
	this.stats.testCount++;
	this.tests.concurrent.push(new Test(title, this._assertModule, cb));
};

Runner.prototype.addSerialTest = function (title, cb) {
	this.stats.testCount++;
	this.tests.serial.push(new Test(title, this._assertModule, cb));
};

Runner.prototype.addBeforeHook = function (title, cb) {
	this.tests.before.push(new Test(title, this._assertModule, cb));
};

Runner.prototype.addAfterHook = function (title, cb) {
	this.tests.after.push(new Test(title, this._assertModule, cb));
};

Runner.prototype.concurrent = function (tests) {
	var self = this;

	// run all tests
	return Promise.all(tests.map(function (test) {
		return test.run()
			.catch(function () {
				// in case of error, don't reject a promise
				return;
			})
			.then(function () {
				self._addTestResult(test);
			});
	}));
};

Runner.prototype.serial = function (tests) {
	var self = this;

	return Promise.resolve(tests).each(function (test) {
		return test.run()
			.catch(function () {
				return;
			})
			.then(function () {
				self._addTestResult(test);
			});
	});
};

Runner.prototype._addTestResult = function (test) {
	if (test.assertError) {
		this.stats.failCount++;
	}

	this.results.push({
		duration: test.duration,
		title: test.title,
		error: test.assertError
	});

	this.emit('test', test.assertError, test.title, test.duration);
};

Runner.prototype.run = function () {
	var self = this;
	var tests = this.tests;
	var stats = this.stats;

	return this.serial(tests.before)
		.then(function () {
			if (stats.failCount > 0) {
				return Promise.reject();
			}
		})
		.then(function () {
			return self.serial(tests.serial);
		})
		.then(function () {
			return self.concurrent(tests.concurrent);
		})
		.then(function () {
			return self.serial(tests.after);
		})
		.catch(function () {
			return;
		})
		.then(function () {
			stats.passCount = stats.testCount - stats.failCount;
		});
};

// Set custom assert module
Runner.prototype.setAssertModule = function (assertModule) {
	this._assertModule = assertModule;
};