acuminous/zUnit

const StreamReporter = require('./StreamReporter');
const Events = require('../Events');
const Outcomes = require('../Outcomes');

class TapReporter extends StreamReporter {
  constructor(options = {}) {
    super({ ...options });
    this._testOutcomeHandlers = {
      [Outcomes.PASSED]: (test) => this._testPassed(test),
      [Outcomes.FAILED]: (test) => this._testFailed(test),
      [Outcomes.SKIPPED]: (test) => this._testSkipped(test),
    };
  }

  withHarness(harness) {
    harness.on(Events.STARTED, () => {
      this._writeln('TAP version 13');
      this._writeln(`1..${harness.numberOfTests}`);
    });
    harness.on(Events.FINISHED, () => {
      this.end();
    });
    return this;
  }

  withSuite(suite) {
    suite.on(Events.FAILED, (result, error) => {
      this._diagnostic(error);
    });
    return new TapReporter(this._options);
  }

  withTest(test) {
    test.on(Events.FINISHED, (result) => {
      this._testOutcomeHandlers[result](test);
    });
    test.on(Events.FAILED, (result, error) => {
      this._diagnostic(error);
    });
    return this;
  }

  _testPassed(test) {
    this._writeln(`ok ${test.point} - ${test.description}`);
  }

  _testFailed(test) {
    this._writeln(`not ok ${test.point} - ${test.description}`);
  }

  _testSkipped(test) {
    this._writeln(`ok ${test.point} - ${test.description} # skip ${test.reason}`);
  }

  _diagnostic(error) {
    const text = error.stack || String(error);
    this._writeln(text.replace(/^/gm, '# '));
  }
}

module.exports = TapReporter;