testplane 9.0.0-rc.4 → 9.0.0

package/build/src/browser/cdp/selectivity/index.js

@@ -1,6 +1,11 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.startSelectivity = exports.updateSelectivityHashes = void 0;
+exports.startSelectivity = exports.clearUnusedSelectivityDumps = exports.updateSelectivityHashes = void 0;
+const node_path_1 = __importDefault(require("node:path"));
+const fs_extra_1 = __importDefault(require("fs-extra"));
 const css_selectivity_1 = require("./css-selectivity");
 const js_selectivity_1 = require("./js-selectivity");
 const test_dependencies_writer_1 = require("./test-dependencies-writer");
@@ -10,16 +15,24 @@ const types_1 = require("./types");
 const testplane_selectivity_1 = require("./testplane-selectivity");
 const hash_reader_1 = require("./hash-reader");
 const events_1 = require("../../../events");
+const modes_1 = require("./modes");
+const debug_1 = require("./debug");
+const used_dumps_tracker_1 = require("./used-dumps-tracker");
 /**
- * Called at the end of successfull testplane run
+ * Called at the end of testplane run
  * Not using "Promise.all" here because all hashes are already calculated and cached at the start
  */
-const updateSelectivityHashes = async (config) => {
+const updateSelectivityHashes = async (config, isRunFailed) => {
     const browserIds = config.getBrowserIds();
+    const processedRoots = new Set();
     for (const browserId of browserIds) {
         const browserConfig = config.forBrowser(browserId);
-        const { enabled, testDependenciesPath, compression, disableSelectivityPatterns } = browserConfig.selectivity;
-        if (!enabled) {
+        const { enabled, testDependenciesPath, compression, disableSelectivityPatterns, saveIncompleteDumpOnFail } = browserConfig.selectivity;
+        const rootKey = `${testDependenciesPath}#${compression}`;
+        if ((isRunFailed || browserConfig.lastFailed.only) && !saveIncompleteDumpOnFail) {
+            continue;
+        }
+        if (!(0, modes_1.selectivityShouldWrite)(enabled) || processedRoots.has(rootKey)) {
             continue;
         }
         const hashReader = (0, hash_reader_1.getHashReader)(testDependenciesPath, compression);
@@ -30,13 +43,84 @@ const updateSelectivityHashes = async (config) => {
                 hashWriter.addPatternDependencyHash(pattern);
             }
         }
-        await hashWriter.commit();
+        try {
+            await hashWriter.save();
+        }
+        catch (cause) {
+            throw new Error(["Selectivity: couldn't save test dependencies hash", String(cause)].join("\n"));
+        }
+        processedRoots.add(rootKey);
     }
 };
 exports.updateSelectivityHashes = updateSelectivityHashes;
+/**
+ * Called at the end of testplane run
+ */
+const clearUnusedSelectivityDumps = async (config, isRunFailed) => {
+    if (process.env.TESTPLANE_SELECTIVITY_CLEAR_UNUSED_DUMPS === "false") {
+        return;
+    }
+    const usedDumpsTracker = (0, used_dumps_tracker_1.getUsedDumpsTracker)();
+    const browserIds = config.getBrowserIds();
+    const selectivityRoots = [];
+    for (const browserId of browserIds) {
+        const browserConfig = config.forBrowser(browserId);
+        const { enabled, testDependenciesPath, saveIncompleteDumpOnFail } = browserConfig.selectivity;
+        if ((isRunFailed && !saveIncompleteDumpOnFail) || browserConfig.lastFailed.only) {
+            continue;
+        }
+        else if ((0, modes_1.selectivityShouldWrite)(enabled) && !selectivityRoots.includes(testDependenciesPath)) {
+            selectivityRoots.push(testDependenciesPath);
+        }
+    }
+    let filesTotal = 0;
+    let filesDeleted = 0;
+    // eslint-disable-next-line no-bitwise
+    const rwMode = fs_extra_1.default.constants.R_OK | fs_extra_1.default.constants.W_OK;
+    await Promise.all(selectivityRoots.map(async (selectivityRoot) => {
+        if (!usedDumpsTracker.usedDumpsFor(selectivityRoot)) {
+            return;
+        }
+        const testsPath = (0, utils_1.getSelectivityTestsPath)(selectivityRoot);
+        const accessError = await fs_extra_1.default.access(testsPath, rwMode).catch((err) => err);
+        if (accessError) {
+            if (!("code" in accessError && accessError.code === "ENOENT")) {
+                (0, debug_1.debugSelectivity)(`Couldn't access "${testsPath}" to clear stale files: %O`, accessError);
+            }
+            return;
+        }
+        const testsFileNames = await fs_extra_1.default.readdir(testsPath);
+        filesTotal += testsFileNames.length;
+        for (const testFileName of testsFileNames) {
+            const extensionPosition = testFileName.indexOf(".json");
+            // If the file does not look like selectivity test dependencies, skip it
+            if (extensionPosition === -1) {
+                continue;
+            }
+            const dumpId = testFileName.slice(0, extensionPosition);
+            if (!usedDumpsTracker.wasUsed(dumpId, selectivityRoot)) {
+                const filePath = node_path_1.default.join(testsPath, testFileName);
+                await fs_extra_1.default
+                    .unlink(filePath)
+                    .then(() => {
+                    filesDeleted++;
+                })
+                    .catch(err => {
+                    (0, debug_1.debugSelectivity)(`Couldn't remove stale file "${filePath}": %O`, err);
+                });
+            }
+        }
+    }));
+    if (filesDeleted) {
+        (0, debug_1.debugSelectivity)(`Out of ${filesTotal} dump files, ${filesDeleted} were considered as outdated and deleted`);
+    }
+};
+exports.clearUnusedSelectivityDumps = clearUnusedSelectivityDumps;
+const testplaneCoverageBreakScriptName = "__testplane_cdp_coverage_snapshot_pause";
+const scriptToEvaluateOnNewDocument = `window.addEventListener("beforeunload", function ${testplaneCoverageBreakScriptName}() {debugger;});`;
 const startSelectivity = async (browser) => {
-    const { enabled, compression, sourceRoot, testDependenciesPath, mapDependencyRelativePath } = browser.config.selectivity;
-    if (!enabled || !browser.publicAPI.isChromium) {
+    const { enabled, compression, sourceRoot, testDependenciesPath, mapDependencyRelativePath, mapSourceMapUrl } = browser.config.selectivity;
+    if (!(0, modes_1.selectivityShouldWrite)(enabled) || !browser.publicAPI.isChromium) {
         return () => Promise.resolve();
     }
     if (compression === types_1.Compression.ZSTD && !process.versions.zstd) {
@@ -45,9 +129,9 @@ const startSelectivity = async (browser) => {
     if (!browser.cdp) {
         throw new Error("Selectivity: Devtools connection is not established, couldn't record selectivity without it");
     }
-    const cdpTaget = browser.cdp.target;
+    const cdp = browser.cdp;
     const handle = await browser.publicAPI.getWindowHandle();
-    const { targetInfos } = await cdpTaget.getTargets();
+    const { targetInfos } = await cdp.target.getTargets();
     const cdpTargetId = targetInfos.find(t => handle.includes(t.targetId))?.targetId;
     if (!cdpTargetId) {
         throw new Error([
@@ -56,29 +140,68 @@ const startSelectivity = async (browser) => {
             `\n\t- cdp targets: ${targetInfos.map(t => `"${t.targetId}"`).join(", ")}`,
         ].join(""));
     }
-    const sessionId = await cdpTaget.attachToTarget(cdpTargetId).then(r => r.sessionId);
-    const cssSelectivity = new css_selectivity_1.CSSSelectivity(browser.cdp, sessionId, sourceRoot);
-    const jsSelectivity = new js_selectivity_1.JSSelectivity(browser.cdp, sessionId, sourceRoot);
-    await Promise.all([cssSelectivity.start(), jsSelectivity.start()]);
+    const wdSessionId = browser.sessionId;
+    const cdpSessionId = await cdp.target.attachToTarget(cdpTargetId).then(r => r.sessionId);
+    const cssSelectivity = new css_selectivity_1.CSSSelectivity(cdp, cdpSessionId, wdSessionId, sourceRoot, mapSourceMapUrl);
+    const jsSelectivity = new js_selectivity_1.JSSelectivity(cdp, cdpSessionId, sourceRoot, mapSourceMapUrl);
+    await Promise.all([
+        cdp.dom.enable(cdpSessionId).then(() => cdp.css.enable(cdpSessionId)),
+        cdp.target.setAutoAttach(cdpSessionId, { autoAttach: true, waitForDebuggerOnStart: false }),
+        cdp.debugger.enable(cdpSessionId),
+        cdp.page.enable(cdpSessionId),
+        cdp.profiler.enable(cdpSessionId),
+    ]);
+    await Promise.allSettled([cssSelectivity.start(), jsSelectivity.start()]).then(async ([css, js]) => {
+        if (css.status === "rejected" || js.status === "rejected") {
+            await Promise.all([cssSelectivity.stop(true), jsSelectivity.stop(true)]);
+            const originalError = css.status === "rejected" ? css.reason : js.status === "rejected" ? js.reason : "unknown reason";
+            throw new Error(["Selectivity: Couldn't start selectivity:", String(originalError)].join("\n"));
+        }
+    });
+    let pageSwitchPromise = Promise.resolve();
+    let isSelectivityStopped = false;
+    const debuggerPausedFn = ({ callFrames }, eventCdpSessionId) => {
+        if (eventCdpSessionId !== cdpSessionId) {
+            return;
+        }
+        if (callFrames[0]?.functionName !== testplaneCoverageBreakScriptName || isSelectivityStopped) {
+            cdp.debugger.resume(cdpSessionId).catch(() => { });
+            return;
+        }
+        pageSwitchPromise = pageSwitchPromise.finally(() => Promise.all([cssSelectivity.takeCoverageSnapshot(), jsSelectivity.takeCoverageSnapshot()])
+            .catch(err => {
+            console.error("Selectivity: couldn't take snapshot while navigating:", err);
+        })
+            .then(() => {
+            cdp.debugger.resume(cdpSessionId).catch(() => { });
+        }));
+    };
+    cdp.debugger.on("paused", debuggerPausedFn);
+    await cdp.page.addScriptToEvaluateOnNewDocument(cdpSessionId, { source: scriptToEvaluateOnNewDocument });
     /** @param drop only performs cleanup without writing anything. Should be "true" if test is failed */
     return async function stopSelectivity(test, drop) {
-        const [cssDependencies, jsDependencies] = await Promise.all([
+        isSelectivityStopped = true;
+        await pageSwitchPromise;
+        const [cssDependenciesPromise, jsDependenciesPromise] = await Promise.allSettled([
             cssSelectivity.stop(drop),
             jsSelectivity.stop(drop),
         ]);
-        cdpTaget.detachFromTarget(sessionId).catch(() => { });
+        cdp.debugger.off("paused", debuggerPausedFn);
+        cdp.target.detachFromTarget(cdpSessionId).catch(() => { });
+        if (jsDependenciesPromise.status === "rejected") {
+            throw jsDependenciesPromise.reason;
+        }
+        if (cssDependenciesPromise.status === "rejected") {
+            throw cssDependenciesPromise.reason;
+        }
+        const cssDependencies = cssDependenciesPromise.value;
+        const jsDependencies = jsDependenciesPromise.value;
         if (drop || (!cssDependencies?.size && !jsDependencies?.size)) {
             return;
         }
-        const mapBrowserDepsRelativePath = mapDependencyRelativePath
-            ? (relativePath) => mapDependencyRelativePath({ scope: "browser", relativePath })
-            : null;
-        const mapTestplaneDepsRelativePath = mapDependencyRelativePath
-            ? (relativePath) => mapDependencyRelativePath({ scope: "testplane", relativePath })
-            : null;
         const testDependencyWriter = (0, test_dependencies_writer_1.getTestDependenciesWriter)(testDependenciesPath, compression);
-        const browserDeps = (0, utils_1.transformSourceDependencies)(cssDependencies, jsDependencies, mapBrowserDepsRelativePath);
-        const testplaneDeps = (0, utils_1.transformSourceDependencies)(null, (0, testplane_selectivity_1.getCollectedTestplaneDependencies)(), mapTestplaneDepsRelativePath);
+        const browserDeps = (0, utils_1.transformSourceDependencies)({ css: cssDependencies, js: jsDependencies, png: null }, mapDependencyRelativePath, "browser");
+        const testplaneDeps = (0, utils_1.transformSourceDependencies)({ css: null, js: (0, testplane_selectivity_1.getCollectedTestplaneJsDependencies)(), png: (0, testplane_selectivity_1.getCollectedTestplanePngDependencies)() }, mapDependencyRelativePath, "testplane");
         process.send?.({
             event: events_1.MasterEvents.TEST_DEPENDENCIES,
             context: {

package/build/src/browser/cdp/selectivity/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/browser/cdp/selectivity/index.ts"],"names":[],"mappings":";;;AAAA,uDAAmD;AACnD,qDAAiD;AAEjD,yEAAuE;AAEvE,mCAA+E;AAC/E,+CAA8C;AAC9C,mCAAsC;AACtC,mEAA4E;AAC5E,+CAA8C;AAE9C,4CAA+C;AAI/C;;;GAGG;AACI,MAAM,uBAAuB,GAAG,KAAK,EAAE,MAAc,EAAiB,EAAE;IAC3E,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;IAE1C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACjC,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,WAAW,EAAE,0BAA0B,EAAE,GAAG,aAAa,CAAC,WAAW,CAAC;QAE7G,IAAI,CAAC,OAAO,EAAE,CAAC;YACX,SAAS;QACb,CAAC;QAED,MAAM,UAAU,GAAG,IAAA,2BAAa,EAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,IAAA,2BAAa,EAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;QAEpE,KAAK,MAAM,OAAO,IAAI,0BAA0B,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAE9D,IAAI,SAAS,EAAE,CAAC;gBACZ,UAAU,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;YACjD,CAAC;QACL,CAAC;QAED,MAAM,UAAU,CAAC,MAAM,EAAE,CAAC;IAC9B,CAAC;AACL,CAAC,CAAC;AAxBW,QAAA,uBAAuB,2BAwBlC;AAEK,MAAM,gBAAgB,GAAG,KAAK,EAAE,OAAwB,EAA8B,EAAE;IAC3F,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,oBAAoB,EAAE,yBAAyB,EAAE,GACvF,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC;IAE/B,IAAI,CAAC,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;QAC5C,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IACnC,CAAC;IAED,IAAI,WAAW,KAAK,mBAAW,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC7D,MAAM,IAAI,KAAK,CACX,+GAA+G,CAClH,CAAC;IACN,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,6FAA6F,CAAC,CAAC;IACnH,CAAC;IAED,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC;IACpC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;IACzD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,QAAQ,CAAC,UAAU,EAAE,CAAC;IACpD,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,QAAQ,CAAC;IAEjF,IAAI,CAAC,WAAW,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACX;YACI,0CAA0C;YAC1C,2BAA2B,MAAM,EAAE;YACnC,sBAAsB,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SAC7E,CAAC,IAAI,CAAC,EAAE,CAAC,CACb,CAAC;IACN,CAAC;IAED,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEpF,MAAM,cAAc,GAAG,IAAI,gCAAc,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IAC9E,MAAM,aAAa,GAAG,IAAI,8BAAa,CAAC,OAAO,CAAC,GAAG,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IAE5E,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,KAAK,EAAE,EAAE,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAEnE,qGAAqG;IACrG,OAAO,KAAK,UAAU,eAAe,CAAC,IAAU,EAAE,IAAa;QAC3D,MAAM,CAAC,eAAe,EAAE,cAAc,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACxD,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,QAAQ,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAErD,IAAI,IAAI,IAAI,CAAC,CAAC,eAAe,EAAE,IAAI,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,EAAE,CAAC;YAC5D,OAAO;QACX,CAAC;QAED,MAAM,0BAA0B,GAAG,yBAAyB;YACxD,CAAC,CAAC,CAAC,YAAoB,EAAiB,EAAE,CAAC,yBAAyB,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;YACxG,CAAC,CAAC,IAAI,CAAC;QAEX,MAAM,4BAA4B,GAAG,yBAAyB;YAC1D,CAAC,CAAC,CAAC,YAAoB,EAAiB,EAAE,CAAC,yBAAyB,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC;YAC1G,CAAC,CAAC,IAAI,CAAC;QAEX,MAAM,oBAAoB,GAAG,IAAA,oDAAyB,EAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;QAC1F,MAAM,WAAW,GAAG,IAAA,mCAA2B,EAAC,eAAe,EAAE,cAAc,EAAE,0BAA0B,CAAC,CAAC;QAC7G,MAAM,aAAa,GAAG,IAAA,mCAA2B,EAC7C,IAAI,EACJ,IAAA,yDAAiC,GAAE,EACnC,4BAA4B,CAC/B,CAAC;QAEF,OAAO,CAAC,IAAI,EAAE,CAAC;YACX,KAAK,EAAE,qBAAY,CAAC,iBAAiB;YACrC,OAAO,EAAE;gBACL,oBAAoB;gBACpB,WAAW;gBACX,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE;gBAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;aACF;YAC3B,IAAI,EAAE,IAAA,+BAAuB,EAAC,WAAW,EAAE,aAAa,CAAwB;SACnF,CAAC,CAAC;QAEH,MAAM,oBAAoB,CAAC,OAAO,CAAC,IAAI,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IACzE,CAAC,CAAC;AACN,CAAC,CAAC;AAnFW,QAAA,gBAAgB,oBAmF3B"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/browser/cdp/selectivity/index.ts"],"names":[],"mappings":";;;;;;AAAA,0DAA6B;AAC7B,wDAA0B;AAC1B,uDAAmD;AACnD,qDAAiD;AAEjD,yEAAuE;AAEvE,mCAAwG;AACxG,+CAA8C;AAC9C,mCAAsC;AACtC,mEAAoH;AACpH,+CAA8C;AAE9C,4CAA+C;AAC/C,mCAAiD;AACjD,mCAA2C;AAC3C,6DAA2D;AAM3D;;;GAGG;AACI,MAAM,uBAAuB,GAAG,KAAK,EAAE,MAAc,EAAE,WAAoB,EAAiB,EAAE;IACjG,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;IAC1C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAE,CAAC;IAEjC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACjC,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,WAAW,EAAE,0BAA0B,EAAE,wBAAwB,EAAE,GACtG,aAAa,CAAC,WAAW,CAAC;QAC9B,MAAM,OAAO,GAAG,GAAG,oBAAoB,IAAI,WAAW,EAAE,CAAC;QAEzD,IAAI,CAAC,WAAW,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAC9E,SAAS;QACb,CAAC;QAED,IAAI,CAAC,IAAA,8BAAsB,EAAC,OAAO,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAClE,SAAS;QACb,CAAC;QAED,MAAM,UAAU,GAAG,IAAA,2BAAa,EAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,IAAA,2BAAa,EAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;QAEpE,KAAK,MAAM,OAAO,IAAI,0BAA0B,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,MAAM,UAAU,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;YAE9D,IAAI,SAAS,EAAE,CAAC;gBACZ,UAAU,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;YACjD,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACD,MAAM,UAAU,CAAC,IAAI,EAAE,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,CAAC,mDAAmD,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACrG,CAAC;QAED,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;AACL,CAAC,CAAC;AArCW,QAAA,uBAAuB,2BAqClC;AAEF;;GAEG;AACI,MAAM,2BAA2B,GAAG,KAAK,EAAE,MAAc,EAAE,WAAoB,EAAiB,EAAE;IACrG,IAAI,OAAO,CAAC,GAAG,CAAC,wCAAwC,KAAK,OAAO,EAAE,CAAC;QACnE,OAAO;IACX,CAAC;IAED,MAAM,gBAAgB,GAAG,IAAA,wCAAmB,GAAE,CAAC;IAC/C,MAAM,UAAU,GAAG,MAAM,CAAC,aAAa,EAAE,CAAC;IAC1C,MAAM,gBAAgB,GAAa,EAAE,CAAC;IAEtC,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;QACjC,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACnD,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,wBAAwB,EAAE,GAAG,aAAa,CAAC,WAAW,CAAC;QAE9F,IAAI,CAAC,WAAW,IAAI,CAAC,wBAAwB,CAAC,IAAI,aAAa,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;YAC9E,SAAS;QACb,CAAC;aAAM,IAAI,IAAA,8BAAsB,EAAC,OAAO,CAAC,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,oBAAoB,CAAC,EAAE,CAAC;YAC7F,gBAAgB,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAChD,CAAC;IACL,CAAC;IAED,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,IAAI,YAAY,GAAG,CAAC,CAAC;IAErB,sCAAsC;IACtC,MAAM,MAAM,GAAG,kBAAE,CAAC,SAAS,CAAC,IAAI,GAAG,kBAAE,CAAC,SAAS,CAAC,IAAI,CAAC;IAErD,MAAM,OAAO,CAAC,GAAG,CACb,gBAAgB,CAAC,GAAG,CAAC,KAAK,EAAC,eAAe,EAAC,EAAE;QACzC,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,eAAe,CAAC,EAAE,CAAC;YAClD,OAAO;QACX,CAAC;QAED,MAAM,SAAS,GAAG,IAAA,+BAAuB,EAAC,eAAe,CAAC,CAAC;QAC3D,MAAM,WAAW,GAAG,MAAM,kBAAE,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC,KAAK,CAAC,CAAC,GAAU,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAElF,IAAI,WAAW,EAAE,CAAC;YACd,IAAI,CAAC,CAAC,MAAM,IAAI,WAAW,IAAI,WAAW,CAAC,IAAI,KAAK,QAAQ,CAAC,EAAE,CAAC;gBAC5D,IAAA,wBAAgB,EAAC,oBAAoB,SAAS,4BAA4B,EAAE,WAAW,CAAC,CAAC;YAC7F,CAAC;YAED,OAAO;QACX,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,kBAAE,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAEnD,UAAU,IAAI,cAAc,CAAC,MAAM,CAAC;QAEpC,KAAK,MAAM,YAAY,IAAI,cAAc,EAAE,CAAC;YACxC,MAAM,iBAAiB,GAAG,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAExD,wEAAwE;YACxE,IAAI,iBAAiB,KAAK,CAAC,CAAC,EAAE,CAAC;gBAC3B,SAAS;YACb,CAAC;YAED,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;YAExD,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,EAAE,eAAe,CAAC,EAAE,CAAC;gBACrD,MAAM,QAAQ,GAAG,mBAAI,CAAC,IAAI,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;gBAEpD,MAAM,kBAAE;qBACH,MAAM,CAAC,QAAQ,CAAC;qBAChB,IAAI,CAAC,GAAG,EAAE;oBACP,YAAY,EAAE,CAAC;gBACnB,CAAC,CAAC;qBACD,KAAK,CAAC,GAAG,CAAC,EAAE;oBACT,IAAA,wBAAgB,EAAC,+BAA+B,QAAQ,OAAO,EAAE,GAAG,CAAC,CAAC;gBAC1E,CAAC,CAAC,CAAC;YACX,CAAC;QACL,CAAC;IACL,CAAC,CAAC,CACL,CAAC;IAEF,IAAI,YAAY,EAAE,CAAC;QACf,IAAA,wBAAgB,EAAC,UAAU,UAAU,gBAAgB,YAAY,0CAA0C,CAAC,CAAC;IACjH,CAAC;AACL,CAAC,CAAC;AA5EW,QAAA,2BAA2B,+BA4EtC;AAEF,MAAM,gCAAgC,GAAG,yCAAyC,CAAC;AACnF,MAAM,6BAA6B,GAAG,oDAAoD,gCAAgC,kBAAkB,CAAC;AAEtI,MAAM,gBAAgB,GAAG,KAAK,EAAE,OAAwB,EAA8B,EAAE;IAC3F,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,UAAU,EAAE,oBAAoB,EAAE,yBAAyB,EAAE,eAAe,EAAE,GACxG,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC;IAE/B,IAAI,CAAC,IAAA,8BAAsB,EAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;QACpE,OAAO,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;IACnC,CAAC;IAED,IAAI,WAAW,KAAK,mBAAW,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC7D,MAAM,IAAI,KAAK,CACX,+GAA+G,CAClH,CAAC;IACN,CAAC;IAED,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,6FAA6F,CAAC,CAAC;IACnH,CAAC;IAED,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;IACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;IACzD,MAAM,EAAE,WAAW,EAAE,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,UAAU,EAAE,CAAC;IACtD,MAAM,WAAW,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,QAAQ,CAAC;IAEjF,IAAI,CAAC,WAAW,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CACX;YACI,0CAA0C;YAC1C,2BAA2B,MAAM,EAAE;YACnC,sBAAsB,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE;SAC7E,CAAC,IAAI,CAAC,EAAE,CAAC,CACb,CAAC;IACN,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,CAAC,SAAS,CAAC;IACtC,MAAM,YAAY,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAEzF,MAAM,cAAc,GAAG,IAAI,gCAAc,CAAC,GAAG,EAAE,YAAY,EAAE,WAAW,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;IACvG,MAAM,aAAa,GAAG,IAAI,8BAAa,CAAC,GAAG,EAAE,YAAY,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;IAExF,MAAM,OAAO,CAAC,GAAG,CAAC;QACd,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACrE,GAAG,CAAC,MAAM,CAAC,aAAa,CAAC,YAAY,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,sBAAsB,EAAE,KAAK,EAAE,CAAC;QAC3F,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC;QACjC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QAC7B,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC;KACpC,CAAC,CAAC;IAEH,MAAM,OAAO,CAAC,UAAU,CAAC,CAAC,cAAc,CAAC,KAAK,EAAE,EAAE,aAAa,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE;QAC/F,IAAI,GAAG,CAAC,MAAM,KAAK,UAAU,IAAI,EAAE,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YACxD,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAEzE,MAAM,aAAa,GACf,GAAG,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,gBAAgB,CAAC;YAErG,MAAM,IAAI,KAAK,CAAC,CAAC,0CAA0C,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACpG,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,iBAAiB,GAAkB,OAAO,CAAC,OAAO,EAAE,CAAC;IACzD,IAAI,oBAAoB,GAAG,KAAK,CAAC;IAEjC,MAAM,gBAAgB,GAAG,CAAC,EAAE,UAAU,EAA4B,EAAE,iBAAgC,EAAQ,EAAE;QAC1G,IAAI,iBAAiB,KAAK,YAAY,EAAE,CAAC;YACrC,OAAO;QACX,CAAC;QAED,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,YAAY,KAAK,gCAAgC,IAAI,oBAAoB,EAAE,CAAC;YAC3F,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAClD,OAAO;QACX,CAAC;QAED,iBAAiB,GAAG,iBAAiB,CAAC,OAAO,CAAC,GAAG,EAAE,CAC/C,OAAO,CAAC,GAAG,CAAC,CAAC,cAAc,CAAC,oBAAoB,EAAE,EAAE,aAAa,CAAC,oBAAoB,EAAE,CAAC,CAAC;aACrF,KAAK,CAAC,GAAG,CAAC,EAAE;YACT,OAAO,CAAC,KAAK,CAAC,uDAAuD,EAAE,GAAG,CAAC,CAAC;QAChF,CAAC,CAAC;aACD,IAAI,CAAC,GAAG,EAAE;YACP,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtD,CAAC,CAAC,CACT,CAAC;IACN,CAAC,CAAC;IAEF,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;IAE5C,MAAM,GAAG,CAAC,IAAI,CAAC,gCAAgC,CAAC,YAAY,EAAE,EAAE,MAAM,EAAE,6BAA6B,EAAE,CAAC,CAAC;IAEzG,qGAAqG;IACrG,OAAO,KAAK,UAAU,eAAe,CAAC,IAAU,EAAE,IAAa;QAC3D,oBAAoB,GAAG,IAAI,CAAC;QAE5B,MAAM,iBAAiB,CAAC;QAExB,MAAM,CAAC,sBAAsB,EAAE,qBAAqB,CAAC,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC;YAC7E,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC;YACzB,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC;SAC3B,CAAC,CAAC;QAEH,GAAG,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;QAC7C,GAAG,CAAC,MAAM,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAE1D,IAAI,qBAAqB,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAC9C,MAAM,qBAAqB,CAAC,MAAM,CAAC;QACvC,CAAC;QAED,IAAI,sBAAsB,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAC/C,MAAM,sBAAsB,CAAC,MAAM,CAAC;QACxC,CAAC;QAED,MAAM,eAAe,GAAG,sBAAsB,CAAC,KAAK,CAAC;QACrD,MAAM,cAAc,GAAG,qBAAqB,CAAC,KAAK,CAAC;QAEnD,IAAI,IAAI,IAAI,CAAC,CAAC,eAAe,EAAE,IAAI,IAAI,CAAC,cAAc,EAAE,IAAI,CAAC,EAAE,CAAC;YAC5D,OAAO;QACX,CAAC;QAED,MAAM,oBAAoB,GAAG,IAAA,oDAAyB,EAAC,oBAAoB,EAAE,WAAW,CAAC,CAAC;QAC1F,MAAM,WAAW,GAAG,IAAA,mCAA2B,EAC3C,EAAE,GAAG,EAAE,eAAe,EAAE,EAAE,EAAE,cAAc,EAAE,GAAG,EAAE,IAAI,EAAE,EACvD,yBAAyB,EACzB,SAAS,CACZ,CAAC;QACF,MAAM,aAAa,GAAG,IAAA,mCAA2B,EAC7C,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE,IAAA,2DAAmC,GAAE,EAAE,GAAG,EAAE,IAAA,4DAAoC,GAAE,EAAE,EACrG,yBAAyB,EACzB,WAAW,CACd,CAAC;QAEF,OAAO,CAAC,IAAI,EAAE,CAAC;YACX,KAAK,EAAE,qBAAY,CAAC,iBAAiB;YACrC,OAAO,EAAE;gBACL,oBAAoB;gBACpB,WAAW;gBACX,MAAM,EAAE,IAAI,CAAC,EAAE;gBACf,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE;gBAC3B,SAAS,EAAE,IAAI,CAAC,SAAS;aACF;YAC3B,IAAI,EAAE,IAAA,+BAAuB,EAAC,WAAW,EAAE,aAAa,CAAwB;SACnF,CAAC,CAAC;QAEH,MAAM,oBAAoB,CAAC,OAAO,CAAC,IAAI,EAAE,WAAW,EAAE,aAAa,CAAC,CAAC;IACzE,CAAC,CAAC;AACN,CAAC,CAAC;AA7IW,QAAA,gBAAgB,oBA6I3B"}

package/build/src/browser/cdp/selectivity/js-selectivity.d.ts

@@ -1,16 +1,23 @@
 import type { CDP } from "..";
 import type { CDPSessionId } from "../types";
+import type { SelectivityMapSourceMapUrlFn } from "../../../config/types";
 export declare class JSSelectivity {
     private readonly _cdp;
     private readonly _sessionId;
     private readonly _sourceRoot;
-    private _debuggerOnPausedFn;
+    private readonly _mapSourceMapUrl;
     private _debuggerOnScriptParsedFn;
-    private _scriptsSource;
-    private _scriptsSourceMap;
-    constructor(cdp: CDP, sessionId: CDPSessionId, sourceRoot?: string);
+    private readonly _scriptsSource;
+    private readonly _scriptsSourceMap;
+    private readonly _scriptIdToSourceUrl;
+    private readonly _scriptIdToSourceMapUrl;
+    private readonly _coverageResult;
+    constructor(cdp: CDP, sessionId: CDPSessionId, sourceRoot: string, mapSourceMapUrl: SelectivityMapSourceMapUrlFn | null);
     private _processScript;
+    private _ensureScriptsAreLoading;
+    private _waitForLoadingScripts;
     start(): Promise<void>;
+    takeCoverageSnapshot(): Promise<void>;
     /** @param drop only performs cleanup without providing actual deps. Should be "true" if test is failed */
     stop(drop?: boolean): Promise<Set<string> | null>;
 }

package/build/src/browser/cdp/selectivity/js-selectivity.js

@@ -5,124 +5,237 @@ const lodash_1 = require("lodash");
 const node_url_1 = require("node:url");
 const constants_1 = require("../../../error-snippets/constants");
 const utils_1 = require("./utils");
+const fs_cache_1 = require("./fs-cache");
+const debug_1 = require("./debug");
 const SOURCE_CODE_EXTENSIONS = [".js", ".jsx", ".ts", ".tsx", ".mjs", ".cjs", ".mts", ".cts"];
 const isSourceCodeFile = (sourceFileName) => {
     return SOURCE_CODE_EXTENSIONS.some(ext => sourceFileName.endsWith(ext));
 };
 class JSSelectivity {
-    constructor(cdp, sessionId, sourceRoot = "") {
-        this._debuggerOnPausedFn = null;
+    constructor(cdp, sessionId, sourceRoot, mapSourceMapUrl) {
         this._debuggerOnScriptParsedFn = null;
         this._scriptsSource = {};
         this._scriptsSourceMap = {};
+        this._scriptIdToSourceUrl = {};
+        this._scriptIdToSourceMapUrl = {};
+        this._coverageResult = [];
         this._cdp = cdp;
         this._sessionId = sessionId;
         this._sourceRoot = sourceRoot;
+        this._mapSourceMapUrl = mapSourceMapUrl;
     }
-    _processScript({ scriptId, url, sourceMapURL }) {
-        if (!this._sessionId) {
+    _processScript({ scriptId, url, sourceMapURL }, cdpSessionId) {
+        if (!this._sessionId || cdpSessionId !== this._sessionId) {
             return;
         }
-        if (!url || !sourceMapURL) {
+        this._scriptIdToSourceUrl[scriptId] ||= url;
+        if (!url || !sourceMapURL || url.startsWith("chrome-error://")) {
             this._scriptsSource[scriptId] ||= null;
             this._scriptsSourceMap[scriptId] ||= null;
             return;
         }
-        this._scriptsSource[scriptId] ||= this._cdp.debugger
-            .getScriptSource(this._sessionId, scriptId)
-            .then(res => res.scriptSource)
-            .catch((err) => err);
-        this._scriptsSourceMap[scriptId] ||= (0, utils_1.fetchTextWithBrowserFallback)((0, node_url_1.resolve)(url, sourceMapURL), this._cdp.runtime, this._sessionId);
+        if (this._scriptsSource[scriptId] && this._scriptsSourceMap[scriptId]) {
+            return;
+        }
+        let sourceMapResolvedUrl = (0, node_url_1.resolve)(url, sourceMapURL);
+        const mapResult = this._mapSourceMapUrl
+            ? this._mapSourceMapUrl({ type: "js", sourceUrl: url, sourceMapUrl: sourceMapResolvedUrl })
+            : true;
+        if (!mapResult) {
+            this._scriptsSource[scriptId] ||= null;
+            this._scriptsSourceMap[scriptId] ||= null;
+            return;
+        }
+        if (mapResult !== true) {
+            this._scriptIdToSourceMapUrl[scriptId] = sourceMapResolvedUrl = mapResult;
+        }
+        this._scriptsSource[scriptId] ||= (0, fs_cache_1.hasCachedSelectivityFile)(fs_cache_1.CacheType.Asset, url).then(isCached => {
+            return isCached
+                ? true
+                : this._cdp.debugger
+                    .getScriptSource(this._sessionId, scriptId)
+                    .then(res => res.scriptSource)
+                    .then(data => (0, fs_cache_1.setCachedSelectivityFile)(fs_cache_1.CacheType.Asset, url, data)
+                    .then(() => true)
+                    .catch(err => {
+                    (0, debug_1.debugSelectivity)(`Couldn't offload asset from "${url}" to fs-cache: %O`, err);
+                    return data;
+                }))
+                    .catch((err) => err);
+        });
+        // Embedded source maps are not cached on file system because of their large cache key
+        if ((0, utils_1.isDataProtocol)(sourceMapResolvedUrl)) {
+            this._scriptIdToSourceMapUrl[scriptId] = null;
+            this._scriptsSourceMap[scriptId] ||= (0, utils_1.fetchTextWithBrowserFallback)(sourceMapResolvedUrl, this._cdp.runtime, this._sessionId).catch((err) => err);
+        }
+        else {
+            this._scriptIdToSourceMapUrl[scriptId] = sourceMapResolvedUrl;
+            this._scriptsSourceMap[scriptId] ||= (0, fs_cache_1.hasCachedSelectivityFile)(fs_cache_1.CacheType.Asset, sourceMapResolvedUrl).then(isCached => {
+                return isCached
+                    ? true
+                    : (0, utils_1.fetchTextWithBrowserFallback)(sourceMapResolvedUrl, this._cdp.runtime, this._sessionId)
+                        .then(data => (0, fs_cache_1.setCachedSelectivityFile)(fs_cache_1.CacheType.Asset, sourceMapResolvedUrl, data)
+                        .then(() => true)
+                        .catch(err => {
+                        (0, debug_1.debugSelectivity)(`Couldn't offload asset from "${sourceMapResolvedUrl}" to fs-cache: %O`, err);
+                        return data;
+                    }))
+                        .catch((err) => err);
+            });
+        }
     }
-    async start() {
-        const debuggerOnPaused = (this._debuggerOnPausedFn = async () => {
-            return this._cdp.debugger.resume(this._sessionId).catch(() => { });
+    // If we haven't got "scriptParsed" event for the script, pull up source code + source map manually
+    _ensureScriptsAreLoading(coverage) {
+        coverage.forEach(({ scriptId, url }) => {
+            const fixedUrl = url || this._scriptIdToSourceUrl[scriptId];
+            const wasProcessed = Object.hasOwn(this._scriptsSource, scriptId) && Object.hasOwn(this._scriptsSourceMap, scriptId);
+            const isAnonymous = !fixedUrl;
+            const urlWasCorrected = url && url !== this._scriptIdToSourceUrl[scriptId];
+            const shouldRecalculateSource = Boolean(urlWasCorrected && this._mapSourceMapUrl);
+            if ((wasProcessed && !shouldRecalculateSource) || isAnonymous) {
+                return;
+            }
+            // Not dropping sources to fs the end of test (when "stop" is called) because we use it immediately
+            const scriptSourcePromise = (async () => {
+                const currentValue = await this._scriptsSource[scriptId];
+                const cacheResolvedValue = (0, utils_1.isCachedOnFs)(currentValue)
+                    ? await (0, fs_cache_1.getCachedSelectivityFile)(fs_cache_1.CacheType.Asset, fixedUrl)
+                    : currentValue;
+                if (cacheResolvedValue && typeof cacheResolvedValue === "string") {
+                    return cacheResolvedValue;
+                }
+                return this._cdp.debugger
+                    .getScriptSource(this._sessionId, scriptId)
+                    .then(({ scriptSource }) => {
+                    (0, fs_cache_1.setCachedSelectivityFile)(fs_cache_1.CacheType.Asset, fixedUrl, scriptSource).catch(() => { });
+                    return scriptSource;
+                })
+                    .catch((err) => err);
+            })();
+            this._scriptIdToSourceUrl[scriptId] ||= url;
+            this._scriptsSource[scriptId] ||= scriptSourcePromise;
+            if (!this._scriptsSourceMap[scriptId] || shouldRecalculateSource) {
+                this._scriptsSourceMap[scriptId] = scriptSourcePromise.then(async (sourceCode) => {
+                    if (sourceCode instanceof Error) {
+                        return sourceCode;
+                    }
+                    const sourceMapsStartIndex = sourceCode.lastIndexOf(constants_1.JS_SOURCE_MAP_URL_COMMENT);
+                    const sourceMapsEndIndex = sourceCode.indexOf("\n", sourceMapsStartIndex);
+                    // Source maps are not generated for this source file
+                    if (sourceMapsStartIndex === -1) {
+                        return null;
+                    }
+                    const sourceMapURL = sourceMapsEndIndex === -1
+                        ? sourceCode.slice(sourceMapsStartIndex + constants_1.JS_SOURCE_MAP_URL_COMMENT.length)
+                        : sourceCode.slice(sourceMapsStartIndex + constants_1.JS_SOURCE_MAP_URL_COMMENT.length, sourceMapsEndIndex);
+                    let resolvedSourceMapUrl = (0, node_url_1.resolve)(fixedUrl, sourceMapURL);
+                    const mappedResult = this._mapSourceMapUrl
+                        ? this._mapSourceMapUrl({ type: "js", sourceUrl: fixedUrl, sourceMapUrl: resolvedSourceMapUrl })
+                        : true;
+                    if (!mappedResult) {
+                        this._scriptsSource[scriptId] = null;
+                        return null;
+                    }
+                    if (mappedResult !== true) {
+                        this._scriptIdToSourceMapUrl[scriptId] = resolvedSourceMapUrl = mappedResult;
+                    }
+                    else {
+                        this._scriptIdToSourceMapUrl[scriptId] ||= resolvedSourceMapUrl;
+                    }
+                    if ((0, utils_1.isDataProtocol)(resolvedSourceMapUrl)) {
+                        return (0, utils_1.fetchTextWithBrowserFallback)(resolvedSourceMapUrl, this._cdp.runtime, this._sessionId).catch((err) => err);
+                    }
+                    try {
+                        const cachedSourceMaps = await (0, fs_cache_1.getCachedSelectivityFile)(fs_cache_1.CacheType.Asset, resolvedSourceMapUrl);
+                        if (cachedSourceMaps) {
+                            return cachedSourceMaps;
+                        }
+                        const sourceMap = await (0, utils_1.fetchTextWithBrowserFallback)(resolvedSourceMapUrl, this._cdp.runtime, this._sessionId);
+                        (0, fs_cache_1.setCachedSelectivityFile)(fs_cache_1.CacheType.Asset, resolvedSourceMapUrl, sourceMap).catch(() => { });
+                        return sourceMap;
+                    }
+                    catch (err) {
+                        return err;
+                    }
+                });
+            }
         });
+    }
+    async _waitForLoadingScripts() {
+        await Promise.all([
+            Promise.allSettled(Object.values(this._scriptsSource)),
+            Promise.allSettled(Object.values(this._scriptsSourceMap)),
+        ]);
+    }
+    async start() {
         const debuggerOnScriptParsedFn = (this._debuggerOnScriptParsedFn = this._processScript.bind(this));
         const sessionId = this._sessionId;
-        this._cdp.debugger.on("paused", debuggerOnPaused);
         this._cdp.debugger.on("scriptParsed", debuggerOnScriptParsedFn);
-        await Promise.all([
-            this._cdp.target.setAutoAttach(sessionId, { autoAttach: true, waitForDebuggerOnStart: false }),
-            this._cdp.debugger.enable(sessionId),
-            this._cdp.profiler.enable(sessionId).then(() => this._cdp.profiler.startPreciseCoverage(sessionId, {
-                callCount: false,
-                detailed: false,
-                allowTriggeredUpdates: false,
-            })),
-        ]);
+        await this._cdp.profiler.startPreciseCoverage(sessionId, {
+            callCount: false,
+            detailed: false,
+            allowTriggeredUpdates: false,
+        });
+    }
+    async takeCoverageSnapshot() {
+        const coveragePart = await this._cdp.profiler.takePreciseCoverage(this._sessionId);
+        this._ensureScriptsAreLoading(coveragePart.result);
+        await this._waitForLoadingScripts();
+        this._coverageResult.push(...coveragePart.result);
     }
     /** @param drop only performs cleanup without providing actual deps. Should be "true" if test is failed */
     async stop(drop) {
-        if (drop) {
-            this._debuggerOnPausedFn && this._cdp.debugger.off("paused", this._debuggerOnPausedFn);
-            this._debuggerOnScriptParsedFn && this._cdp.debugger.off("scriptParsed", this._debuggerOnScriptParsedFn);
-            return null;
-        }
-        const coverage = await this._cdp.profiler.takePreciseCoverage(this._sessionId);
-        const scriptsWithUrl = coverage.result.filter(script => script.url);
-        // If we haven't got "scriptParsed" event for the script, pull up source code + source map manually
-        scriptsWithUrl.forEach(({ scriptId, url }) => {
-            if (Object.hasOwn(this._scriptsSource, scriptId) && Object.hasOwn(this._scriptsSourceMap, scriptId)) {
-                return;
+        try {
+            if (drop) {
+                return null;
             }
-            const scriptSourcePromise = this._cdp.debugger
-                .getScriptSource(this._sessionId, scriptId)
-                .then(res => res.scriptSource)
-                .catch((err) => err);
-            this._scriptsSource[scriptId] ||= scriptSourcePromise;
-            this._scriptsSourceMap[scriptId] ||= scriptSourcePromise.then(sourceCode => {
-                if (sourceCode instanceof Error) {
-                    return sourceCode;
+            const coverageLastPart = await this._cdp.profiler.takePreciseCoverage(this._sessionId);
+            const coverageScripts = [...this._coverageResult, ...coverageLastPart.result];
+            this._ensureScriptsAreLoading(coverageLastPart.result);
+            const totalDependingSourceFiles = new Set();
+            const grouppedByScriptCoverage = (0, lodash_1.groupBy)(coverageScripts, "scriptId");
+            const scriptIds = Object.keys(grouppedByScriptCoverage);
+            await Promise.all(scriptIds.map(async (scriptId) => {
+                const [source, sourceMaps] = await Promise.all([
+                    this._scriptsSource[scriptId],
+                    this._scriptsSourceMap[scriptId],
+                ]);
+                const sourceMapUrl = this._scriptIdToSourceMapUrl[scriptId];
+                // Every "scriptId" has only one uniq "url"
+                const sourceUrl = this._scriptIdToSourceUrl[scriptId] || grouppedByScriptCoverage[scriptId][0].url;
+                // Function was called, but source maps were not generated for the file
+                // Or its anonymous script, without source url
+                if (!source || !sourceMaps || !sourceUrl) {
+                    return;
                 }
-                const sourceMapsStartIndex = sourceCode.lastIndexOf(constants_1.JS_SOURCE_MAP_URL_COMMENT);
-                const sourceMapsEndIndex = sourceCode.indexOf("\n", sourceMapsStartIndex);
-                if (sourceMapsStartIndex === -1) {
-                    return new Error("Source maping url comment is missing");
+                if (source instanceof Error) {
+                    throw new Error([`JS Selectivity: Couldn't load source code from ${sourceUrl}:`, String(source)].join("\n"));
                 }
-                const sourceMapURL = sourceMapsEndIndex === -1
-                    ? sourceCode.slice(sourceMapsStartIndex + constants_1.JS_SOURCE_MAP_URL_COMMENT.length)
-                    : sourceCode.slice(sourceMapsStartIndex + constants_1.JS_SOURCE_MAP_URL_COMMENT.length, sourceMapsEndIndex);
-                return (0, utils_1.fetchTextWithBrowserFallback)((0, node_url_1.resolve)(url, sourceMapURL), this._cdp.runtime, this._sessionId);
-            });
-        });
-        const totalDependingSourceFiles = new Set();
-        const grouppedByScriptCoverage = (0, lodash_1.groupBy)(coverage.result, "scriptId");
-        const scriptIds = Object.keys(grouppedByScriptCoverage);
-        await Promise.all(scriptIds.map(async (scriptId) => {
-            // Every "scriptId" has only one uniq "url"
-            const url = grouppedByScriptCoverage[scriptId][0].url;
-            const [source, sourceMaps] = await Promise.all([
-                this._scriptsSource[scriptId],
-                this._scriptsSourceMap[scriptId],
-            ]);
-            if (!source || !sourceMaps) {
-                return;
-            }
-            if (source instanceof Error) {
-                throw new Error(`JS Selectivity: Couldn't load source code at ${url}: ${source}`);
-            }
-            if (sourceMaps instanceof Error) {
-                throw new Error(`JS Selectivity: Couldn't load source maps of ${url}: ${sourceMaps}`);
-            }
-            const startOffsetsSet = new Set();
-            grouppedByScriptCoverage[scriptId].forEach(entry => {
-                entry.functions.forEach(fn => {
-                    fn.ranges.forEach(range => {
-                        startOffsetsSet.add(range.startOffset);
-                    });
-                });
-            });
-            const dependingSourceFiles = await (0, utils_1.extractSourceFilesDeps)(source, sourceMaps, Array.from(startOffsetsSet), this._sourceRoot);
-            for (const sourceFile of dependingSourceFiles.values()) {
-                if (isSourceCodeFile(sourceFile)) {
+                if (sourceMaps instanceof Error) {
+                    throw new Error([`JS Selectivity: Couldn't load source maps from ${sourceMapUrl}`, String(sourceMaps)].join("\n"));
+                }
+                if ((0, utils_1.isCachedOnFs)(sourceMaps) && !sourceMapUrl) {
+                    throw new Error("Assertation failed: souce map url has to present if source maps are fs-cached");
+                }
+                const [sourceString, sourceMapsString] = await Promise.all([
+                    (0, utils_1.isCachedOnFs)(source) ? (0, fs_cache_1.getCachedSelectivityFile)(fs_cache_1.CacheType.Asset, sourceUrl) : source,
+                    (0, utils_1.isCachedOnFs)(sourceMaps)
+                        ? (0, fs_cache_1.getCachedSelectivityFile)(fs_cache_1.CacheType.Asset, sourceMapUrl)
+                        : sourceMaps,
+                ]);
+                if (!sourceString || !sourceMapsString) {
+                    throw new Error(`JS Selectivity: fs-cache is broken for ${sourceUrl}`);
+                }
+                const dependingSourceFiles = (0, utils_1.extractSourceFilesDeps)(sourceString, sourceMapsString, grouppedByScriptCoverage[scriptId], this._sourceRoot, isSourceCodeFile);
+                for (const sourceFile of dependingSourceFiles.values()) {
                     totalDependingSourceFiles.add(sourceFile);
                 }
-            }
-        }));
-        this._debuggerOnPausedFn && this._cdp.debugger.off("paused", this._debuggerOnPausedFn);
-        this._debuggerOnScriptParsedFn && this._cdp.debugger.off("scriptParsed", this._debuggerOnScriptParsedFn);
-        return totalDependingSourceFiles;
+            }));
+            return totalDependingSourceFiles;
+        }
+        finally {
+            this._debuggerOnScriptParsedFn && this._cdp.debugger.off("scriptParsed", this._debuggerOnScriptParsedFn);
+        }
     }
 }
 exports.JSSelectivity = JSSelectivity;