protractor-junit-flake/src/logger.js

A wrapper for protractor to automatically re-run failed tests for a specific number of attempts

import chalk from 'chalk'

const LOG_LEVELS = {
  debug: 1,
  protractor: 2,
  info: 3,
  silent: 4
}

class Logger {
  constructor (color) {
    this.color = color
  }

  log (levelName, message, useColor = true) {
    let currentLevel = LOG_LEVELS[process.env.PROTRACTOR_FLAKE_LOG_LEVEL] || LOG_LEVELS.protractor
    let incomingLevel = LOG_LEVELS[levelName]

    if (incomingLevel >= currentLevel) {
      message = this.colorize(message, useColor)
      process.stdout.write(message)
    }
  }

  protractor (message) {
    this.log('protractor', message, false)
  }

  info (message) {
    this.log('info', message)
  }

  colorize (message, useColor) {
    if (useColor && chalk.supportsColor && this.color) {
      return chalk[this.color](message)
    } else {
      return message
    }
  }
}

export default Logger