embark/lib/tests/reporter.js

const Base = require('mocha/lib/reporters/base');
const color = Base.color;

class EmbarkSpec extends Base {
  constructor(runner, options) {
    super(runner, options);

    const self = this;
    self.embarkEvents = options.reporterOptions.events;
    self.gasDetails = options.reporterOptions.gasDetails;
    self.gasLimit = options.reporterOptions.gasLimit;
    let indents = 0;
    let n = 0;
    self.stats.gasCost = 0;

    function onContractReceipt(receipt) {
      self.stats.gasCost += receipt.gasUsed;
      if (self.gasDetails) {
        const fmt = color('bright pass', ' ') +
          color('green', ' %s') +
          ' deployed for ' +
          color(self.getGasColor(receipt.gasUsed), '%s') +
          color('light', ' gas');

        console.log(fmt, receipt.className, receipt.gasUsed);
      }
    }

    self.embarkEvents.on("deploy:contract:receipt", onContractReceipt);

    function indent() {
      return Array(indents).join('  ');
    }

    runner.on('start', function () {
      console.log();
    });

    runner.on('suite', function (suite) {
      ++indents;
      if (self.gasDetails) {
        console.log();
      }
      console.log(color('suite', '%s%s'), indent(), suite.title);
    });

    runner.on('suite end', function () {
      --indents;
      if (indents === 1) {
        console.log();
      }
    });

    runner.on('pending', function (test) {
      const fmt = indent() + color('pending', '  - %s');
      console.log(fmt, test.title);
    });

    runner.on('pass', function (test) {
      let fmt;
      if (test.speed === 'fast') {
        fmt =
          indent() +
          color('checkmark', '  ' + Base.symbols.ok) +
          color('pass', ' %s');
        console.log(fmt, test.title);
      } else {
        fmt =
          indent() +
          color('checkmark', '  ' + Base.symbols.ok) +
          color('pass', ' %s') +
          color(test.speed, ' (%dms)');
        console.log(fmt, test.title, test.duration);
      }
    });

    runner.on('fail', function (test) {
      console.log(indent() + color('fail', '  %d) %s'), ++n, test.title);
    });

    runner.once('end', function () {
      runner.removeAllListeners();
      self.embarkEvents.removeListener("deploy:contract:receipt", onContractReceipt);
      self.epilogue();
    });
  }

  getGasColor(gasCost) {
    if (gasCost <= this.gasLimit/10) {
      return 'fast';
    }
    if (gasCost <= 3 * (this.gasLimit/4)) {
      return 'medium';
    }
    return 'slow';
  }

  epilogue() {
    const stats = this.stats;
    let fmt;

    console.log();

    // passes
    fmt = color('bright pass', ' ') +
      color('green', ' %d passing') +
      color(this.getGasColor(stats.gasCost), ' (%s gas)');

    console.log(fmt,
      stats.passes || 0,
      stats.gasCost);

    // pending
    if (stats.pending) {
      fmt = color('pending', ' ') +
        color('pending', ' %d pending');

      console.log(fmt, stats.pending);
    }

    // failures
    if (stats.failures) {
      fmt = color('fail', '  %d failing');

      console.log(fmt, stats.failures);

      Base.list(this.failures);
      console.log();
    }

    console.log();
  }
}

module.exports = EmbarkSpec;