@vitest/browser 3.1.0-beta.2 → 3.1.1

package/dist/{webdriver-BP2w_ajA.js → webdriver-2iYWIzBv.js}

@@ -1,3 +1,5 @@
+import { createManualModuleSource } from '@vitest/mocker/node';
+
 const playwrightBrowsers = [
 	"firefox",
 	"webkit",
@@ -13,6 +15,7 @@ class PlaywrightBrowserProvider {
 	contexts = new Map();
 	pages = new Map();
 	browserPromise = null;
+	mocker;
 	getSupportedBrowsers() {
 		return playwrightBrowsers;
 	}
@@ -20,6 +23,7 @@ class PlaywrightBrowserProvider {
 		this.project = project;
 		this.browserName = browser;
 		this.options = options;
+		this.mocker = this.createMocker();
 	}
 	async openBrowser() {
 		if (this.browserPromise) {
@@ -58,6 +62,109 @@ class PlaywrightBrowserProvider {
 		})();
 		return this.browserPromise;
 	}
+	createMocker() {
+		const idPreficates = new Map();
+		const sessionIds = new Map();
+		function createPredicate(sessionId, url) {
+			const moduleUrl = new URL(url, "http://localhost");
+			const predicate = (url) => {
+				if (url.searchParams.has("_vitest_original")) {
+					return false;
+				}
+				if (url.pathname !== moduleUrl.pathname) {
+					return false;
+				}
+				url.searchParams.delete("t");
+				url.searchParams.delete("v");
+				url.searchParams.delete("import");
+				if (url.searchParams.size !== moduleUrl.searchParams.size) {
+					return false;
+				}
+				for (const [param, value] of url.searchParams.entries()) {
+					if (moduleUrl.searchParams.get(param) !== value) {
+						return false;
+					}
+				}
+				return true;
+			};
+			const ids = sessionIds.get(sessionId) || [];
+			ids.push(moduleUrl.href);
+			sessionIds.set(sessionId, ids);
+			idPreficates.set(moduleUrl.href, predicate);
+			return predicate;
+		}
+		return {
+			register: async (sessionId, module) => {
+				const page = this.getPage(sessionId);
+				await page.route(createPredicate(sessionId, module.url), async (route) => {
+					if (module.type === "manual") {
+						const exports = Object.keys(await module.resolve());
+						const body = createManualModuleSource(module.url, exports);
+						return route.fulfill({
+							body,
+							headers: getHeaders(this.project.browser.vite.config)
+						});
+					}
+					const isWebkit = this.browserName === "webkit";
+					if (isWebkit) {
+						const url = module.type === "redirect" ? (() => {
+							const url = new URL(module.redirect);
+							return url.href.slice(url.origin.length);
+						})() : (() => {
+							const url = new URL(route.request().url());
+							url.searchParams.set("mock", module.type);
+							return url.href.slice(url.origin.length);
+						})();
+						const result = await this.project.browser.vite.transformRequest(url).catch(() => null);
+						if (!result) {
+							return route.continue();
+						}
+						let content = result.code;
+						if (result.map && "version" in result.map && result.map.mappings) {
+							const type = isDirectCSSRequest(url) ? "css" : "js";
+							content = getCodeWithSourcemap(type, content.toString(), result.map);
+						}
+						return route.fulfill({
+							body: content,
+							headers: getHeaders(this.project.browser.vite.config)
+						});
+					}
+					if (module.type === "redirect") {
+						return route.fulfill({
+							status: 302,
+							headers: { Location: module.redirect }
+						});
+					} else if (module.type === "automock" || module.type === "autospy") {
+						const url = new URL(route.request().url());
+						url.searchParams.set("mock", module.type);
+						return route.fulfill({
+							status: 302,
+							headers: { Location: url.href }
+						});
+					} else ;
+				});
+			},
+			delete: async (sessionId, id) => {
+				const page = this.getPage(sessionId);
+				const predicate = idPreficates.get(id);
+				if (predicate) {
+					await page.unroute(predicate).finally(() => idPreficates.delete(id));
+				}
+			},
+			clear: async (sessionId) => {
+				const page = this.getPage(sessionId);
+				const ids = sessionIds.get(sessionId) || [];
+				const promises = ids.map((id) => {
+					const predicate = idPreficates.get(id);
+					if (predicate) {
+						return page.unroute(predicate).finally(() => idPreficates.delete(id));
+					}
+					return null;
+				});
+				await Promise.all(promises).finally(() => sessionIds.delete(sessionId));
+			}
+		};
+	}
 	async createContext(sessionId) {
 		if (this.contexts.has(sessionId)) {
 			return this.contexts.get(sessionId);
@@ -66,8 +173,7 @@ class PlaywrightBrowserProvider {
 		const { actionTimeout,...contextOptions } = this.options?.context ?? {};
 		const options = {
 			...contextOptions,
-			ignoreHTTPSErrors: true,
-			serviceWorkers: "allow"
+			ignoreHTTPSErrors: true
 		};
 		if (this.project.config.browser.ui) {
 			options.viewport = null;
@@ -100,7 +206,7 @@ class PlaywrightBrowserProvider {
 					const timeout = setTimeout(() => {
 						const err = new Error(`Cannot find "vitest-iframe" on the page. This is a bug in Vitest, please report it.`);
 						reject(err);
-					}, 1e3);
+					}, 1e3).unref();
 					page.on("frameattached", (frame) => {
 						clearTimeout(timeout);
 						resolve(frame);
@@ -166,6 +272,32 @@ class PlaywrightBrowserProvider {
 		await browser?.close();
 	}
 }
+function getHeaders(config) {
+	const headers = { "Content-Type": "application/javascript" };
+	for (const name in config.server.headers) {
+		headers[name] = String(config.server.headers[name]);
+	}
+	return headers;
+}
+function getCodeWithSourcemap(type, code, map) {
+	if (type === "js") {
+		code += `\n//# sourceMappingURL=${genSourceMapUrl(map)}`;
+	} else if (type === "css") {
+		code += `\n/*# sourceMappingURL=${genSourceMapUrl(map)} */`;
+	}
+	return code;
+}
+function genSourceMapUrl(map) {
+	if (typeof map !== "string") {
+		map = JSON.stringify(map);
+	}
+	return `data:application/json;base64,${Buffer.from(map).toString("base64")}`;
+}
+const CSS_LANGS_RE = /\.(?:css|less|sass|scss|styl|stylus|pcss|postcss|sss)(?:$|\?)/;
+const directRequestRE = /[?&]direct\b/;
+function isDirectCSSRequest(request) {
+	return CSS_LANGS_RE.test(request) && directRequestRE.test(request);
+}
 
 const webdriverBrowsers = [
 	"firefox",