CyberChef/tests/operations/TestRegister.mjs

/**
 * TestRegister.js
 *
 * This is so individual files can register their tests in one place, and
 * ensure that they will get run by the frontend.
 *
 * @author tlwr [toby@toby.codes]
 * @copyright Crown Copyright 2017
 * @license Apache-2.0
 */
import Chef from "../../src/core/Chef";

(function() {
    /**
     * Object to store and run the list of tests.
     *
     * @class
     * @constructor
     */
    function TestRegister() {
        this.tests = [];
    }


    /**
     * Add a list of tests to the register.
     *
     * @param {Object[]} tests
     */
    TestRegister.prototype.addTests = function(tests) {
        this.tests = this.tests.concat(tests);
    };


    /**
     * Runs all the tests in the register.
     */
    TestRegister.prototype.runTests = function() {
        return Promise.all(
            this.tests.map(function(test, i) {
                const chef = new Chef();

                return chef.bake(
                    test.input,
                    test.recipeConfig,
                    {},
                    0,
                    false
                ).then(function(result) {
                    const ret = {
                        test: test,
                        status: null,
                        output: null,
                    };

                    if (result.error) {
                        if (test.expectedError) {
                            ret.status = "passing";
                        } else {
                            ret.status = "erroring";
                            ret.output = result.error.displayStr;
                        }
                    } else {
                        if (test.expectedError) {
                            ret.status = "failing";
                            ret.output = "Expected an error but did not receive one.";
                        } else if (result.result === test.expectedOutput) {
                            ret.status = "passing";
                        } else if (test.hasOwnProperty("expectedMatch") && test.expectedMatch.test(result.result)) {
                            ret.status = "passing";
                        } else {
                            ret.status = "failing";
                            const expected = test.expectedOutput ? test.expectedOutput :
                                test.expectedMatch ? test.expectedMatch.toString() : "unknown";
                            ret.output = [
                                "Expected",
                                "\t" + expected.replace(/\n/g, "\n\t"),
                                "Received",
                                "\t" + result.result.replace(/\n/g, "\n\t"),
                            ].join("\n");
                        }
                    }

                    return ret;
                });
            })
        );
    };


    // Singleton TestRegister, keeping things simple and obvious.
    global.TestRegister = global.TestRegister || new TestRegister();
})();

export default global.TestRegister;