grunt-protractor.js

'use strict';

var terminate = require('terminate');
var spawn = require('cross-spawn');
var grunt = require('grunt');
var protractorDir = require.resolve('protractor/bin/elementexplorer.js').replace('elementexplorer.js', '');

module.exports = {
  updateWebdriver: function (done) {
    var p = spawn('node', [protractorDir + '/webdriver-manager', 'update', '--versions.chrome', '2.28']);
    p.stdout.pipe(process.stdout);
    p.stderr.pipe(process.stderr);
    p.on('exit', function (code) {
      if (code !== 0) {
        grunt.fail.warn('Webdriver failed to update');
      }
      done();
    });
  },

  startProtractor: function (config, done) {
    var allowedOptions = ['sauceUser', 'sauceKey', 'capabilities.tunnel-identifier',
                          'capabilities.build', 'browser', 'specs', 'baseUrl'];
    allowedOptions.forEach(function (option) {
      var value = grunt.option(option);
      if (value) {
        config[option] = value;
      }
    });

    var args = [protractorDir + '/protractor', config.configFile];
    args = args.concat(Object.keys(config).filter(function (key) {
      return key !== 'configFile';
    }).map(function (key) {
      return '--' + key + '=' + config[key];
    }));

    var p = spawn('node', args);
    var killed = false;
    var self = this;
    var retry;

    function scheduleRetry() {
      retry = setTimeout(function () {
        if (process.env.IS_BUILD_AGENT && process.env.SKIP_PROTRACTOR_RETRY !== 'true') {
          console.log('RETRY!!!');
          killed = true;
          terminate(p.pid, function () {
            console.log('TERMINATED!!!');
            self.startProtractor(config, done);
          });
        }
      }, 20000);
    }

    scheduleRetry();
    p.stdout.pipe(process.stdout);
    p.stderr.pipe(process.stderr);
    process.stdin.pipe(p.stdin);
    p.stdout.on('data', function (data) {
      if (data.indexOf('beforeLaunch: start') > -1) {
        clearTimeout(retry);
      } else if (data.indexOf('beforeLaunch: done') > -1) {
        clearTimeout(retry);
        scheduleRetry();
      } else if (data.indexOf('onPrepare!!!') > -1) {
        clearTimeout(retry);
      }
    });
    p.on('exit', function (code) {
      if (!killed) {
        if (code !== 0) {
          grunt.fail.warn('Protractor test(s) failed. Exit code: ' + code);
        }
        done();
      }
    });
  }
};