protractor-junit-flake/dist/index.js

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

'use strict';

Object.defineProperty(exports, '__esModule', {
  value: true
});

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { 'default': obj }; }

var _child_process = require('child_process');

var _junitXml = require('./junit-xml');

var _parseOptions = require('./parse-options');

var _parseOptions2 = _interopRequireDefault(_parseOptions);

require('core-js/shim');

var _logger = require('./logger');

var _logger2 = _interopRequireDefault(_logger);

var _lodash = require('lodash');

var _lodash2 = _interopRequireDefault(_lodash);

exports['default'] = function () {
  var options = arguments.length <= 0 || arguments[0] === undefined ? {} : arguments[0];
  var callback = arguments.length <= 1 || arguments[1] === undefined ? function noop() {} : arguments[1];

  var parsedOptions = (0, _parseOptions2['default'])(options);
  var logger = new _logger2['default'](parsedOptions.color);

  logger.info('Running attempt ' + parsedOptions.testAttempt + ' of ' + parsedOptions.maxAttempts + '\n');
  logger.info('Using resultsXMLPath: ' + parsedOptions.resultsXmlPath + '\n');
  var testAttempt = parsedOptions.testAttempt || 1;

  function rerunFailedTests(status, output) {
    var failedSpecNames = (0, _junitXml.processResults)(parsedOptions.resultsXmlPath, testAttempt);

    ++testAttempt;
    logger.info('Failed specs = ' + failedSpecNames);
    if (!failedSpecNames || failedSpecNames.length === 0) {
      logger.info('\nNo failed specs were found. Exiting test attempt ' + testAttempt + '.\n');
      status = 0;
      callback(status, output);
    } else {
      logger.info('\nRe-running test attempt ' + testAttempt + ' with ' + failedSpecNames.length + ' tests\n');
      var specRegex = failedSpecNames.map(function (name) {
        return _lodash2['default'].escapeRegExp(name).replace(/[/]/g, '\\/');
      }).join('|');
      return startProtractor(specRegex, true);
    }
  }

  function handleTestEnd(status) {
    var output = arguments.length <= 1 || arguments[1] === undefined ? '' : arguments[1];

    logger.info('Test ended with status  ' + status + '\n');
    if (!status || testAttempt >= parsedOptions.maxAttempts) {
      (0, _junitXml.processLastRunResults)(parsedOptions.resultsXmlPath, testAttempt);
      callback(status, output);
    } else {
      return rerunFailedTests(status, output);
    }
  }

  function startProtractor() {
    var specRegex = arguments.length <= 0 || arguments[0] === undefined ? '' : arguments[0];
    var retry = arguments.length <= 1 || arguments[1] === undefined ? false : arguments[1];

    var protractorArgs = [parsedOptions.protractorPath].concat(parsedOptions.protractorArgs);

    if (retry) {
      protractorArgs.push('--params.flake.retry', true);
    }

    if (specRegex) {
      protractorArgs.push('--jasmineNodeOpts.grep', specRegex);
    }

    protractorArgs.push('--testAttempt', testAttempt);
    protractorArgs.push('--disableChecks');

    var protractor = (0, _child_process.spawn)(parsedOptions.nodeBin, protractorArgs, parsedOptions.protractorSpawnOptions);

    protractor.stdout.on('data', function (buffer) {
      var text = buffer.toString();
      logger.protractor(text);
    });

    protractor.stderr.on('data', function (buffer) {
      var text = buffer.toString();
      logger.protractor(text);
    });

    protractor.on('exit', function (code, signal) {
      logger.info('Exited attempt ' + testAttempt + ' with code: ' + code + ' and signal: ' + signal + '\n');
      handleTestEnd(code, '');
    });

    protractor.on('error', function (err) {
      logger.info('Protractor error ' + err + '\n', true);
      handleTestEnd(1, 'Protractor error ' + err + '\n');
    });
  }

  if (testAttempt > 1 && testAttempt <= parsedOptions.maxAttempts) {
    // We increment testAttempt in rerunFailedTests, so move it back first
    --testAttempt;
    rerunFailedTests(0, '');
  } else {
    startProtractor();
  }
};

module.exports = exports['default'];