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

Vitest visual testing plugin

import type { BrowserCommandContext } from 'vitest/node'
import type { BrowserApi } from './types.ts'

export function playwright(context: BrowserCommandContext): BrowserApi {
	return {
		async takeScreenshot(filePath, selector, options) {
			// The `Locator` type from `vitest` has less props than the `Locator` in `playwright`
			const subject = context.iframe.locator(selector)
			return subject.screenshot({
				timeout: options?.timeout,
				animations: 'disabled',
				path: filePath,
			})
		},
	}
}