codeceptjs/lib/mochaFactory.js

Modern Era Acceptance Testing Framework for NodeJS

const Mocha = require('mocha');
const fsPath = require('path');
const fs = require('fs');
const reporter = require('./reporter/cli');
const gherkinParser = require('./interfaces/gherkin.js');
const output = require('./output');
const { genTestId } = require('./utils');
const ConnectionRefused = require('./helper/errors/ConnectionRefused');

const scenarioUi = fsPath.join(__dirname, './ui.js');

let mocha;

class MochaFactory {
  static create(config, opts) {
    mocha = new Mocha(Object.assign(config, opts));
    output.process(opts.child);
    mocha.ui(scenarioUi);

    // process.on('unhandledRejection', (reason) => {
    //   output.error('Unhandled rejection');
    //   console.log(Error.captureStackTrace(reason));
    //   output.error(reason);
    // });


    Mocha.Runner.prototype.uncaught = function (err) {
      if (err) {
        if (err.toString().indexOf('ECONNREFUSED') >= 0) {
          err = new ConnectionRefused(err);
        }
        output.error(err);
        output.print(err.stack);
        process.exit(1);
      }
      output.error('Uncaught undefined exception');
      process.exit(1);
    };

    mocha.loadFiles = (fn) => {
      // load features
      mocha.files
        .filter(file => file.match(/\.feature$/))
        .map(file => fs.readFileSync(file, 'utf8'))
        .forEach(content => mocha.suite.addSuite(gherkinParser(content)));

      // remove feature files
      mocha.files = mocha.files.filter(file => !file.match(/\.feature$/));

      Mocha.prototype.loadFiles.call(mocha, fn);

      // add ids for each test
      mocha.suite.eachTest(test => test.id = genTestId(test));
    };

    // use standard reporter
    if (!opts.reporter) {
      mocha.reporter(reporter, opts);
      return mocha;
    }

    // load custom reporter with options
    const reporterOptions = Object.assign(config.reporterOptions || {});

    if (opts.reporterOptions !== undefined) {
      opts.reporterOptions.split(',').forEach((opt) => {
        const L = opt.split('=');
        if (L.length > 2 || L.length === 0) {
          throw new Error(`invalid reporter option '${opt}'`);
        } else if (L.length === 2) {
          reporterOptions[L[0]] = L[1];
        } else {
          reporterOptions[L[0]] = true;
        }
      });
    }

    const attributes = Object.getOwnPropertyDescriptor(reporterOptions, 'codeceptjs-cli-reporter');
    if (reporterOptions['codeceptjs-cli-reporter'] && attributes) {
      Object.defineProperty(
        reporterOptions, 'codeceptjs/lib/reporter/cli',
        attributes,
      );
      delete reporterOptions['codeceptjs-cli-reporter'];
    }

    // custom reporters
    mocha.reporter(opts.reporter, reporterOptions);
    return mocha;
  }
}

module.exports = MochaFactory;