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

Vitest visual testing plugin

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

export function webdriverio(context: BrowserCommandContext): BrowserApi {
	const page = (context.provider as any).browser!
	return {
		async takeScreenshot(filePath, selector) {
			const element = await page.$(`${selector}`)
			return element.saveScreenshot(filePath)
		},
	}
}