testswarm-browserstack.cli.js

#!/opt/local/bin/node
var program = require('commander'),
    tsbs = require('./testswarm-browserstack');

program
    .version('1.0')
    .option('--killAll', 'Kill all BrowserStack workers')
    .option('--killWorker [workerid]', 'Kill a BrowserStack worker by its worker ID', parseInt)
    .option('--getNeeded', 'Shows a list of browser IDs that have pending jobs in TestSwarm')
    .option('--kill', 'Kill BrowserStack workers if they are no longer needed (Only if --run is also specified)')
    .option('--run', 'Start new workers in BrowserStack based on the swarm state')
    .option('--dryRun', 'Use this option in combination with --kill, --run and/or --killAll. Will stop any action from taking place and only report what it would do in reality. Intended for debugging or getting statistics.')
    .option('-u, --user [username]', 'BrowserStack username', '')
    .option('-p, --pass [password]', 'BrowserStack password', '')
    .option('-v, --verbose', 'Output more debug messages (all output via console.log)')
    .option('--swarmUrl [url]', 'URL of TestSwarm root (without trailing slash)', '')
    .option('--swarmRunUrl [url]', 'URL to the TestSwarm run page (including client name), for BrowserStack workers to open', '')
    .option('--stackLimit [workers]', 'How many workers can be running simultaneously in BrowserStack (default: 4 workers)', parseInt)
    .option('--clientTimeout [min]', 'Number of minutes to run each client (default: 10 minutes)', parseInt)
    .parse(process.argv);

if (!process.argv[2]) {
    console.log(program.helpInformation());
    return;
}

console.log('\n--\n-- testswarm-browserstack.cli.js: ' + new Date().toString() + '\n--');

tsbs.options(program);

if (program.getNeeded) {
    if (!program.swarmUrl) {
        console.log('please set --swarmUrl. stopping.');
        return;
    }
    tsbs.getSwarmState(function (error, swarmState) {
        var browserID, stats,
            needed = [];
        if (error) {
            console.log('getting swarm state failed:\n', error);
            return;
        }
        for (browserID in swarmState.userAgents) {
            stats = swarmState.userAgents[browserID].stats;
            if (stats.onlineClients === 0 && stats.pendingRuns > 0) {
                needed.push(browserID);
            }
        }
        console.log(needed);
    });
}

if (program.run || program.killWorker || program.killAll) {
    if (!program.pass || !program.user) {
        console.log('please set --user and --pass for browserstack. stopping.');
        return;
    }
}

if (program.killAll) {
    tsbs.killAll(program.killWorker);
}

if (program.killWorker) {
    tsbs.killWorker(program.killWorker);
}

if (program.run) {
    if (!program.swarmUrl || !program.swarmRunUrl) {
        console.log('please set --swarmUrl and --swarmRunUrl. stopping.');
        return;
    }

    // Default options
    program.stackLimit = program.stackLimit || 4;
    program.clientTimeout = program.clientTimeout || 10;
    // Convert timeout to ming
    program.clientTimeout = program.clientTimeout * 60;
    tsbs.run();
}