html2canvas/tests/mocha/selenium.js

var wd = require('wd');
var http = require('http');
var https = require('https');
var url = require('url');
var path = require('path');
var Promise = require('bluebird');
var _ = require('lodash');
var humanizeDuration = require('humanize-duration');
var utils = require('../utils');
var colors = utils.colors;
var port = 8080;

function runTestWithRetries(browser, test, retries) {
    retries = retries || 0;
    return runTest(browser, test).timeout(30000).catch(Promise.TimeoutError, function() {
        if (retries < 3) {
            console.log(colors.violet, 'Retry', retries + 1, test);
            return runTestWithRetries(browser, test, retries + 1);
        } else {
            throw new Error("Couldn't run test after 3 retries");
        }
    });
}

function getResults(browser) {
    return function() {
        return Promise.props({
            dataUrl: browser
                .waitForElementByCss("body[data-complete='true']", 90000)
                .then(function() {
                    return browser.elementsByCssSelector('.test.fail');
                })
                .then(function(nodes) {
                    return Array.isArray(nodes)
                        ? Promise.map(nodes, function(node) {
                              return browser.text(node).then(function(error) {
                                  return Promise.reject(error);
                              });
                          })
                        : Promise.resolve([]);
                })
        });
    };
}

function runTest(browser, test) {
    return Promise.resolve(
        browser.then(utils.loadTestPage(browser, test, port)).then(getResults(browser))
    ).cancellable();
}

exports.tests = function(browsers, singleTest) {
    var path = 'tests/mocha';
    return (singleTest ? Promise.resolve([singleTest]) : utils.getTests(path)).then(function(
        tests
    ) {
        return Promise.map(
            browsers,
            function(settings) {
                var name = [settings.browserName, settings.version, settings.platform].join('-');
                var count = 0;
                var browser = utils.initBrowser(settings);
                return Promise.using(browser, function() {
                    return Promise.map(
                        tests,
                        function(test, index, total) {
                            console.log(
                                colors.green,
                                'STARTING',
                                '(' + ++count + '/' + total + ')',
                                name,
                                test,
                                colors.clear
                            );
                            var start = Date.now();
                            return runTestWithRetries(browser, test).then(function() {
                                console.log(
                                    colors.green,
                                    'COMPLETE',
                                    humanizeDuration(Date.now() - start),
                                    '(' + count + '/' + total + ')',
                                    name,
                                    colors.clear
                                );
                            });
                        },
                        {concurrency: 1}
                    )
                        .settle()
                        .catch(function(error) {
                            console.error(colors.red, 'ERROR', name, error);
                            throw error;
                        });
                });
            },
            {concurrency: 3}
        );
    });
};