跳至主要内容

可扩展性

自定义选择器引擎

Playwright 支持自定义选择器引擎,并通过 selectors.register() 注册。

选择器引擎应具有以下属性

  • query 函数用于查询相对于 root 的第一个与 selector 匹配的元素。
  • queryAll 函数用于查询相对于 root 的所有与 selector 匹配的元素。

默认情况下,引擎直接在框架的 JavaScript 上下文中运行,例如,可以调用应用程序定义的函数。为了将引擎与框架中的任何 JavaScript 隔离,但保留对 DOM 的访问权限,请使用 {contentScript: true} 选项注册引擎。内容脚本引擎更安全,因为它受到保护,不会被任何篡改全局对象的操作影响,例如更改 Node.prototype 方法。所有内置选择器引擎都作为内容脚本运行。请注意,当引擎与其他自定义引擎一起使用时,无法保证以内容脚本的形式运行。

必须在创建页面之前注册选择器。

一个根据标签名称查询元素的注册选择器引擎示例

baseTest.ts
import { test as base } from '@playwright/test';

export { expect } from '@playwright/test';

// Must be a function that evaluates to a selector engine instance.
const createTagNameEngine = () => ({
// Returns the first element matching given selector in the root's subtree.
query(root, selector) {
return root.querySelector(selector);
},

// Returns all elements matching given selector in the root's subtree.
queryAll(root, selector) {
return Array.from(root.querySelectorAll(selector));
}
});

export const test = base.extend<{}, { selectorRegistration: void }>({
// Register selectors once per worker.
selectorRegistration: [async ({ playwright }, use) => {
// Register the engine. Selectors will be prefixed with "tag=".
await playwright.selectors.register('tag', createTagNameEngine);
await use();
}, { scope: 'worker', auto: true }],
});
example.spec.ts
import { test, expect } from './baseTest';

test('selector engine test', async ({ page }) => {
// Now we can use 'tag=' selectors.
const button = page.locator('tag=button');
await button.click();

// We can combine it with built-in locators.
await page.locator('tag=div').getByText('Click me').click();

// We can use it in any methods supporting selectors.
await expect(page.locator('tag=button')).toHaveCount(3);
});