123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
'use strict';const Hash = require('hashish');const {format} = require('util');
// capture terminal output, so that we might// assert against it.exports.checkOutput = function checkOutput(f, argv, cb) {  let exit = false;  let exitCode = 0;  const _exit = process.exit;  const _emit = process.emit;  const _emitWarning = process.emitWarning;  const _env = process.env;  const _argv = process.argv;  const _error = console.error;  const _log = console.log;  const _warn = console.warn;
  process.exit = code => {    exit = true;    exitCode = code;  };  process.env = Hash.merge(process.env, {_: 'node'});  process.argv = argv || ['./usage'];
  const errors = [];  const logs = [];  const warnings = [];  const emittedWarnings = [];
  console.error = (...msg) => {    errors.push(format(...msg));  };  console.log = (...msg) => {    logs.push(format(...msg));  };  console.warn = (...msg) => {    warnings.push(format(...msg));  };  process.emitWarning = warning => {    emittedWarnings.push(warning);  };
  let result;
  if (typeof cb === 'function') {    process.exit = () => {      exit = true;      cb(null, done());    };    process.emit = function emit(ev, value) {      if (ev === 'uncaughtException') {        cb(value, done());        return true;      }
      return _emit.apply(this, arguments);    };
    f();  } else {    try {      result = f();      if (typeof result.then === 'function') {        return result          .then(() => {            reset();            return done();          })          .catch(() => {            reset();            return done();          });      } else {        reset();      }    } catch (_err) {      reset();    }    return done();  }
  function reset() {    process.exit = _exit;    process.emit = _emit;    process.env = _env;    process.argv = _argv;    process.emitWarning = _emitWarning;
    console.error = _error;    console.log = _log;    console.warn = _warn;  }
  function done() {    reset();
    return {      errors,      logs,      warnings,      emittedWarnings,      exit,      exitCode,      result,    };  }};