gemini-core/lib/base-stats/index.js

Utility which contains common modules for gemini and hermione

'use strict';

const _ = require('lodash');

module.exports = class BaseStats {
    static create(...args) {
        return new this(...args);
    }

    constructor(statNames) {
        this._tests = new Set();
        this._retries = 0;
        this._statNames = statNames;

        this._stats = this._fillEmptyStats();
    }

    addPassed(test) {
        this._addStat(this._statNames.PASSED, test);
    }

    addFailed(test) {
        this._addStat(this._statNames.FAILED, test);
    }

    addSkipped(test) {
        this._addStat(this._statNames.SKIPPED, test);
    }

    addRetries() {
        this._retries++;
    }

    _addStat(stat, test, statsStorage = this._stats, testsStorage = this._tests) {
        const key = `${this._buildSuiteKey(test)} ${this._buildStateKey(test)}`;

        statsStorage[stat]++;
        testsStorage.add(key);
    }

    _fillEmptyStats() {
        const statValues = _.values(this._statNames);

        return _.zipObject(statValues, Array(statValues.length).fill(0));
    }

    _buildStateKey() {
        throw new Error('Method must be implemented in child classes');
    }

    _buildSuiteKey() {
        throw new Error('Method must be implemented in child classes');
    }

    getResult() {
        return _.extend(this._stats, {
            total: this._tests.size,
            retries: this._retries
        });
    }
};