protractor.conf.js

/*global exports,browser */

'use strict';

require('babel-register');
require('babel-polyfill');

var Cookie = require('tough-cookie');
var request = require('request-promise');
var fs = require('fs');
var path = require('path');

function login(steamid) {
  steamid = steamid || 76561197970669109;

  var url = browser.params.apiUrl + '/startMockLogin?steamid=' + steamid;

  return request.get(url, {
    resolveWithFullResponse: true,
    followRedirect: false,
    simple: false
  }).then(function (res) {
    var authCookie = Cookie.parse(res.headers['set-cookie'][0]);

    browser.driver.get(browser.params.apiUrl);
    browser.manage().addCookie(authCookie.key, authCookie.value, authCookie.path, authCookie.domain);
  });
}

function dirExists(dirName) {
  try {
    var stat = fs.statSync(dirName);
    return stat.isDirectory();
  }
  catch (e) {
    return false;
  }
}

function writeScreenShot(data, filename) {
  var stream = fs.createWriteStream(filename);
  stream.write(new Buffer(data, 'base64'));
  stream.end();
}

function saveScreenshot(filename) {
  browser.takeScreenshot().then(function (png) {
    writeScreenShot(png, filename);
  });
}

exports.config = {
  framework: 'jasmine',

  baseUrl: 'http://localhost:8080/',
// disable, see if it works again with next protractor version
//  directConnect: true,

  specs: ['tests/**.js'],
  capabilities: {
    browserName: 'firefox'
  },

  params: {
    viewport: {
      width: 1200,
      height: 900
    },
    screenshots: 'screenshots'
  },

  onPrepare: function () {
    if (browser.params.screenshots && browser.params.screenshots !== 'false') {
      var screenshotsDir = browser.params.screenshots;

      if (!dirExists(screenshotsDir)) {
        fs.mkdirSync(screenshotsDir);
      }

      global.takeScreenshot = function (name) {
        saveScreenshot(path.join(screenshotsDir, name + '.png'));
      };
    } else {
      global.takeScreenshot = function () {};
    }

    global.login = login;

    var win = browser.manage().window();
    var viewport = browser.params.viewport;
    win.setSize(viewport.width, viewport.height);
  }
};