acuminous/zUnit

const Suite = require('./Suite');
const Test = require('./Test');
const Hook = require('./Hook');
const context = new Array();

function describe(name, fn, options) {
  return _describe(name, fn, { ...options });
}

function xdescribe(name, fn, options) {
  return _describe(name, fn, { ...options, skip: true });
}

function odescribe(name, fn, options) {
  return _describe(name, fn, { ...options, exclusive: true });
}

function _describe(name, fn, options) {
  const suite = new Suite(name, options);
  context.push(suite);
  fn();
  context.pop();
  publish(suite);
  return suite;
}

function it(name, fn, options) {
  return _it(name, fn, { ...options });
}

function xit(name, fn, options) {
  return _it(name, fn, { ...options, skip: true });
}

function oit(name, fn, options) {
  return _it(name, fn, { ...options, exclusive: true });
}

function _it(name, fn, options) {
  const test = new Test(name, fn, options);
  publish(test);
  return test;
}

function before(...args) {
  const hook = createHook('before', args);
  currentSuite().before(hook);
}

function beforeEach(...args) {
  const hook = createHook('beforeEach', args);
  currentSuite().beforeEach(hook);
}

function afterEach(...args) {
  const hook = createHook('afterEach', args);
  currentSuite().afterEach(hook);
}

function after(...args) {
  const hook = createHook('after', args);
  currentSuite().after(hook);
}

function createHook(defaultName, args) {
  const { name, fn, options } = { name: defaultName, ...getHookParameters(args) };
  return new Hook(name, fn, options);
}

function getHookParameters(args) {
  return typeof args[0] === 'string'
    ? {
        name: args[0],
        fn: args[1],
        options: args[2],
      }
    : {
        fn: args[0],
        options: args[1],
      };
}

function include(...testables) {
  currentSuite().add(testables);
}

function currentSuite() {
  return context[context.length - 1];
}

function publish(testable) {
  if (context.length > 0) {
    currentSuite().add(testable);
  } else {
    process.emit('zunit:testable', testable);
  }
}

module.exports = {
  describe,
  xdescribe,
  odescribe,
  it,
  xit,
  oit,
  before,
  beforeEach,
  after,
  afterEach,
  include,
};