balazs4 / alola Goto Github PK

const rgb = require('barecolor')

module.exports = function(headline) {
  const suite = [],
    before = [],
    after = [],
    only = []

  function self(name, fn) {
    suite.push({ name: name, fn: fn })
  }

  self.only = function(name, fn) {
    only.push({ name: name, fn: fn })
  }

  self.before = function(fn) { before.push(fn) }
  self.after = function(fn) { after.push(fn)  }
  self.skip = function(fn) {}

  // for testing Baretest
  self.run = async function() {
    const tests = only[0] ? only : suite

    rgb.cyan(headline + ' ')
    const failed = [];

    for (const test of tests) {
      try {
        for (const fn of before) await fn()
        await test.fn()
        rgb.gray('• ')

      } catch(e) {
        for (const fn of after) await fn()
        rgb.red('• ')
        failed.push(()=>{
           rgb.red(`\n\n! ${test.name} \n\n`)
           prettyError(e)
         });
        continue
      }
    }

    for (const fn of after) await fn()

    if (failed.length === 0){
      rgb.greenln(`✓ ${ tests.length }`)
      console.info('\n')
      return true
    }

    failed.forEach(fn => fn());
    rgb.redln(`x ${ failed.length } of ${tests.length}`)
    return false;
  }

  return self

}


function prettyError(e) {
  const msg = e.stack
  if (!msg) return rgb.yellow(e)

  const i = msg.indexOf('\n')
  rgb.yellowln(msg.slice(0, i))
  rgb.gray(msg.slice(i))
  console.info('\n')
}