'use strict';

const _ = require('lodash');

const RunnerEvents = require('./constants/events');

const STATS = {

total: 'total',

updated: 'updated',

passed: 'passed',

failed: 'failed',

skipped: 'skipped',

retries: 'retries'

};

module.exports = class Stats {

static create() {

return new Stats();

}

constructor() {

this._stats = {};

}

attachRunner(runner) {

runner

.on(RunnerEvents.SKIP_STATE, (test) => this._addStat(STATS.skipped, test))

.on(RunnerEvents.ERROR, (test) => this._addStat(STATS.failed, test))

.on(RunnerEvents.UPDATE_RESULT, (test) => {

return test.updated ? this._addStat(STATS.updated, test) : this._addStat(STATS.passed, test);

})

.on(RunnerEvents.TEST_RESULT, (test) => {

return test.equal ? this._addStat(STATS.passed, test) : this._addStat(STATS.failed, test);

})

.on(RunnerEvents.RETRY, (test) => this._getSuiteStats(test).retries++);

}

_addStat(stat, test) {

this._getSuiteStats(test).states[test.state.name] = stat;

}

_getSuiteStats(test) {

const key = this._buildSuiteKey(test);

if (!this._stats[key]) {

this._stats[key] = {

retries: 0,

states: {}

};

}

return this._stats[key];

}

_buildSuiteKey(test) {

return `${test.suite.fullName} ${test.browserId}`;

}

getResult() {

const statNames = _.keys(STATS);

const result = _.zipObject(statNames, _.fill(Array(statNames.length), 0));

_.forEach(this._stats, (suiteStats) => {

result.retries += suiteStats.retries;

_.forEach(suiteStats.states, (stateStatus) => {

result.total++;

result[stateStatus]++;

});

});

return result;

}

};