testem/lib/utils/run-timeout.js

Test'em 'scripts! Javascript Unit testing made easy.

'use strict';

const Bluebird = require('bluebird');
const EventEmitter = require('events').EventEmitter;

class RunTimeout {
  constructor(timeout) {
    this.timeout = timeout;
  }

  static with(timeout) {
    let runTimeout = new RunTimeout(timeout);

    return runTimeout.start().disposer(() => runTimeout.stop());
  }

  start() {
    let self = this;

    if (this.timeout) {
      this.timeoutID = setTimeout(() => {
        self.setTimedOut();
      }, this.timeout * 1000);
    }

    return Bluebird.resolve(this);
  }

  setTimedOut() {
    this.timedOut = true;
    this.emit('timeout');
  }

  stop() {
    clearTimeout(this.timeoutID);
    this.timeoutID = null;
    this.timedOut = null;
  }

  try(fn) {
    if (this.timedOut) {
      return Bluebird.reject(new Error('Run timed out.'));
    }

    return fn();
  }
}

RunTimeout.prototype.__proto__ = EventEmitter.prototype;

module.exports = RunTimeout;