@vitest/browser-playwright 5.0.0-beta.2 → 5.0.0-beta.4

package/dist/index.js

@@ -527,7 +527,7 @@ async function takeScreenshot(context, name, options) {
 	const mask = options.mask?.map((selector) => getDescribedLocator(context, selector));
 	const style = context.project.config.browser.ui ? options.style === undefined ? SCREENSHOT_STYLES : SCREENSHOT_STYLES + options.style : options.style;
 	if (options.element) {
-		const { element: selector, ...config } = options;
+		const { element: selector, target: _target, ...config } = options;
 		const element = getDescribedLocator(context, selector);
 		const buffer = await element.screenshot({
 			...config,
@@ -540,11 +540,17 @@ async function takeScreenshot(context, name, options) {
 			path
 		};
 	}
-	const buffer = await getDescribedLocator(context, {
+	const { target, ...config } = options;
+	const buffer = target === "page" ? await context.page.screenshot({
+		...config,
+		mask,
+		path: savePath,
+		style
+	}) : await getDescribedLocator(context, {
 		selector: "body",
 		locator: "locator('body')"
 	}).screenshot({
-		...options,
+		...config,
 		mask,
 		path: savePath,
 		style
@@ -955,7 +961,7 @@ class PlaywrightBrowserProvider {
 	createMocker() {
 		const idPredicates = new Map();
 		const sessionIds = new Map();
-		function createPredicate(sessionId, url) {
+		function createPredicate(url) {
 			const moduleUrl = new URL(url, "http://localhost");
 			const predicate = (url) => {
 				if (url.searchParams.has("_vitest_original")) {
@@ -980,11 +986,10 @@ class PlaywrightBrowserProvider {
 				}
 				return true;
 			};
-			const ids = sessionIds.get(sessionId) || [];
-			ids.push(moduleUrl.href);
-			sessionIds.set(sessionId, ids);
-			idPredicates.set(predicateKey(sessionId, moduleUrl.href), predicate);
-			return predicate;
+			return {
+				url: moduleUrl.href,
+				predicate
+			};
 		}
 		function predicateKey(sessionId, url) {
 			return `${sessionId}:${url}`;
@@ -992,7 +997,17 @@ class PlaywrightBrowserProvider {
 		return {
 			register: async (sessionId, module) => {
 				const page = this.getPage(sessionId);
-				await page.context().route(createPredicate(sessionId, module.url), async (route) => {
+				const { url: moduleUrl, predicate } = createPredicate(module.url);
+				const key = predicateKey(sessionId, moduleUrl);
+				const existingPredicate = idPredicates.get(key);
+				if (existingPredicate) {
+					await page.context().unroute(existingPredicate);
+				}
+				const ids = sessionIds.get(sessionId) ?? new Set();
+				ids.add(moduleUrl);
+				sessionIds.set(sessionId, ids);
+				idPredicates.set(key, predicate);
+				await page.context().route(predicate, async (route) => {
 					if (module.type === "manual") {
 						const exports$1 = Object.keys(await module.resolve());
 						const body = createManualModuleSource(module.url, exports$1);
@@ -1053,8 +1068,8 @@ class PlaywrightBrowserProvider {
 			},
 			clear: async (sessionId) => {
 				const page = this.getPage(sessionId);
-				const ids = sessionIds.get(sessionId) || [];
-				const promises = ids.map((id) => {
+				const ids = sessionIds.get(sessionId) ?? new Set();
+				const promises = [...ids].map((id) => {
 					const key = predicateKey(sessionId, id);
 					const predicate = idPredicates.get(key);
 					if (predicate) {
@@ -1176,18 +1191,10 @@ class PlaywrightBrowserProvider {
 		const page = this.getPage(sessionid);
 		const cdp = await page.context().newCDPSession(page);
 		return {
-			send(method, params) {
-				return cdp.send(method, params);
-			},
-			on(event, listener) {
-				return cdp.on(event, listener);
-			},
-			off(event, listener) {
-				return cdp.off(event, listener);
-			},
-			once(event, listener) {
-				return cdp.once(event, listener);
-			}
+			send: cdp.send.bind(cdp),
+			on: cdp.on.bind(cdp),
+			off: cdp.off.bind(cdp),
+			once: cdp.once.bind(cdp)
 		};
 	}
 	async close() {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@vitest/browser-playwright",
   "type": "module",
-  "version": "5.0.0-beta.2",
+  "version": "5.0.0-beta.4",
   "description": "Browser running for Vitest using playwright",
   "license": "MIT",
   "funding": "https://opencollective.com/vitest",
@@ -42,7 +42,7 @@
   ],
   "peerDependencies": {
     "playwright": "*",
-    "vitest": "5.0.0-beta.2"
+    "vitest": "5.0.0-beta.4"
   },
   "peerDependenciesMeta": {
     "playwright": {
@@ -51,12 +51,12 @@
   },
   "dependencies": {
     "tinyrainbow": "^3.1.0",
-    "@vitest/browser": "5.0.0-beta.2",
-    "@vitest/mocker": "5.0.0-beta.2"
+    "@vitest/mocker": "5.0.0-beta.4",
+    "@vitest/browser": "5.0.0-beta.4"
   },
   "devDependencies": {
-    "playwright": "^1.59.0",
-    "vitest": "5.0.0-beta.2"
+    "playwright": "^1.60.0",
+    "vitest": "5.0.0-beta.4"
   },
   "scripts": {
     "build": "premove dist && pnpm rollup -c",