vitest-plugin-vis/src/server/browser_provider/playwright.ts

Vitest visual testing plugin

import { resolve } from 'pathe'
import type { ExtendedBrowserCommandContext } from '../vis_server_context.types.ts'
import type { BrowserApi } from './types.ts'

export function playwright(context: ExtendedBrowserCommandContext): BrowserApi {
	const { page, iframe } = context

	return {
		async takeScreenshot(projectRoot, relativeFilePath, selector, options) {
			const subject = iframe.locator(selector)
			return subject.screenshot({
				path: resolve(projectRoot, relativeFilePath),
				...options,
			})
		},
		async takePageScreenshot(projectRoot, relativeFilePath, options) {
			return page.screenshot({
				path: resolve(projectRoot, relativeFilePath),
				...options,
			})
		},
	}
}