e2ed/actions/mock/unmockWebSocketRoute.js

E2E testing framework over Playwright

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.unmockWebSocketRoute = void 0;
const webSocketMockState_1 = require("../../context/webSocketMockState");
const useContext_1 = require("../../useContext");
const asserts_1 = require("../../utils/asserts");
const fn_1 = require("../../utils/fn");
const log_1 = require("../../utils/log");
/**
 * Unmock WebSocket (remove mock, if any) for some WebSocket route.
 */
const unmockWebSocketRoute = async (Route) => {
    const webSocketMockState = (0, webSocketMockState_1.getWebSocketMockState)();
    const { optionsByRoute, requestsFilter } = webSocketMockState;
    let webSocketMockFunction;
    let routeWasMocked = false;
    let skipLogs;
    if (optionsByRoute?.has(Route)) {
        const options = optionsByRoute.get(Route);
        webSocketMockFunction = options?.webSocketMockFunction;
        skipLogs = options?.skipLogs;
        routeWasMocked = true;
        optionsByRoute.delete(Route);
    }
    if (optionsByRoute?.size === 0) {
        (0, asserts_1.assertValueIsDefined)(requestsFilter, 'requestsFilter is defined', {
            routeName: Route.name,
            routeWasMocked,
        });
        const page = (0, useContext_1.getPlaywrightPage)();
        await page.unroute(requestsFilter);
    }
    if (webSocketMockFunction) {
        (0, fn_1.setCustomInspectOnFunction)(webSocketMockFunction);
    }
    if (skipLogs !== true) {
        (0, log_1.log)(`Unmock WebSocket for route "${Route.name}"`, { routeWasMocked, webSocketMockFunction }, 5 /* LogEventType.InternalAction */);
    }
};
exports.unmockWebSocketRoute = unmockWebSocketRoute;