e2ed
Version:
E2E testing framework over Playwright
44 lines (43 loc) • 1.75 kB
JavaScript
;
Object.defineProperty(exports, "__esModule", { value: true });
exports.unmockWebSocketRoute = void 0;
const webSocketMockState_1 = require("../../context/webSocketMockState");
const step_1 = require("../../step");
const useContext_1 = require("../../useContext");
const asserts_1 = require("../../utils/asserts");
const fn_1 = require("../../utils/fn");
/**
* 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 (webSocketMockFunction) {
(0, fn_1.setCustomInspectOnFunction)(webSocketMockFunction);
}
await (0, step_1.step)(`Unmock WebSocket for route "${Route.name}"`, async () => {
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);
}
}, {
payload: { routeWasMocked, webSocketMockFunction },
skipLogs: skipLogs ?? false,
type: 5 /* LogEventType.InternalAction */,
});
};
exports.unmockWebSocketRoute = unmockWebSocketRoute;