zulip/frontend_tests/zjsunit/zblueslip.js

exports.make_zblueslip = function () {
    const lib = {};

    const opts = {
        // Silently swallow all debug, log and info calls.
        debug: false,
        log: false,
        info: false,
        // Check against expected error values for the following.
        warn: true,
        error: true,
        fatal: true,
    };

    // Store valid test data for options.
    lib.test_data = {};
    lib.test_logs = {};
    Object.keys(opts).forEach(name => {
        lib.test_data[name] = [];
        lib.test_logs[name] = [];
    });
    lib.set_test_data = (name, message) => {
        lib.test_data[name].push(message);
    };
    lib.clear_test_data = (name) => {
        if (!name) {
            // Clear all data
            Object.keys(opts).forEach(name => {
                lib.test_data[name] = [];
                lib.test_logs[name] = [];
            });
            return;
        }
        lib.test_data[name] = [];
        lib.test_logs[name] = [];
    };

    lib.get_test_logs = (name) => {
        return lib.test_logs[name];
    };

    // Create logging functions
    Object.keys(opts).forEach(name => {
        if (!opts[name]) {
            // should just log the message.
            lib[name] = function (message, more_info, stack) {
                lib.test_logs[name].push({message, more_info, stack});
            };
            return;
        }
        lib[name] = function (message, more_info, stack) {
            if (typeof message !== 'string') {
                // We may catch exceptions in blueslip, and if
                // so our stub should include that.
                if (message.toString().includes('exception')) {
                    message = message.toString();
                } else {
                    throw Error('message should be string: ' + message);
                }
            }
            lib.test_logs[name].push({message, more_info, stack});
            const exact_match_fail = !lib.test_data[name].includes(message);
            if (exact_match_fail) {
                const error = Error(`Invalid ${name} message: "${message}".`);
                error.blueslip = true;
                throw error;
            }
        };
    });

    lib.exception_msg = function (ex) {
        return ex.message;
    };

    lib.start_timing = () => {
        return () => {};
    };

    return lib;
};