跳到主要内容

模拟 API

简介

Web API 通常实现为 HTTP 端点。Playwright 提供了用于模拟修改网络流量(包括 HTTP 和 HTTPS)的 API。页面发出的任何请求,包括 XHRfetch 请求,都可以被跟踪、修改和模拟。使用 Playwright,你还可以使用包含页面发出的多个网络请求的 HAR 文件进行模拟。

模拟 API 请求

以下代码将拦截对 */**/api/v1/fruits 的所有调用,并返回自定义响应。不会向 API 发出任何请求。测试将访问使用模拟路由的 URL,并断言页面上存在模拟数据。

// 拦截到水果 API 的路由
await page.RouteAsync("*/**/api/v1/fruits", async route => {
var json = new[] { new { name = "Strawberry", id = 21 } };
// 使用模拟数据完成路由
await route.FulfillAsync(new()
{
Json = json
});
});

// 访问页面
await page.GotoAsync("https://demo.playwright.dev/api-mocking");

// 断言草莓水果可见
await Expect(page.GetByTextAsync("Strawberry")).ToBeVisibleAsync();

从示例测试的跟踪中可以看到,API 从未被调用,但它确实使用模拟数据完成了请求。api mocking trace

阅读更多关于 高级网络 的内容。

修改 API 响应

有时,进行 API 请求至关重要,但响应需要进行修补,以便进行可重现的测试。在这种情况下,无需模拟请求,而是可以执行请求并用修改后的响应来完成它。

在下面的示例中,我们拦截对水果 API 的调用,并向数据中添加一种名为 “枇杷” 的新水果。然后,我们访问该 URL 并断言数据已存在:

await page.RouteAsync("*/**/api/v1/fruits", async (route) => {
var response = await route.FetchAsync();
var fruits = await response.JsonAsync<Fruit[]>();
fruits.Add(new Fruit() { Name = "Loquat", Id = 100 });
// 使用原始响应完成请求,同时用给定的 JSON 对象修补响应正文。
await route.FulfillAsync(new ()
{
Response = response,
Json = fruits
});
}
);
// 访问页面
await page.GotoAsync("https://demo.playwright.dev/api-mocking");

// 断言枇杷水果可见
await Expect(page.GetByTextAsync("Loquat", new () { Exact = true })).ToBeVisibleAsync();

在测试的跟踪信息中,我们可以看到 API 被调用且响应被修改。显示 API 被调用并完成的测试跟踪

通过检查响应,我们可以看到新水果已添加到列表中。显示模拟响应的测试跟踪

阅读更多关于 高级网络 的内容。

使用 HAR 文件进行模拟

HAR 文件是一种 HTTP 存档 文件,其中包含页面加载时发出的所有网络请求记录。它包含有关请求和响应头、Cookie、内容、时间等信息。你可以在测试中使用 HAR 文件来模拟网络请求。你需要:

  1. 记录 HAR 文件。
  2. 将 HAR 文件与测试一起提交。
  3. 在测试中使用保存的 HAR 文件来路由请求。

记录 HAR 文件

要记录 HAR 文件,我们使用 Page.RouteFromHARAsync()BrowserContext.RouteFromHARAsync() 方法。此方法接受 HAR 文件的路径和一个可选的选项对象。选项对象可以包含 URL,这样只有 URL 与指定通配符模式匹配的请求才会从 HAR 文件提供服务。如果未指定,则所有请求都将从 HAR 文件提供服务。

update 选项设置为 true 将使用实际网络信息创建或更新 HAR 文件,而不是从 HAR 文件提供请求服务。在创建测试以使用真实数据填充 HAR 时使用此选项。

// 从 HAR 文件获取响应
await page.RouteFromHARAsync("./hars/fruit.har", new () {
Url = "*/**/api/v1/fruits",
Update = true,
});

// 导航到页面
await page.GotoAsync("https://demo.playwright.dev/api-mocking");

// 断言水果可见
await Expect(page.GetByText("Strawberry")).ToBeVisibleAsync();

修改 HAR 文件

记录 HAR 文件后,你可以通过打开“hars”文件夹内的哈希 .txt 文件并编辑 JSON 来修改它。此文件应提交到你的源代码控制中。每当你使用 update: true 运行此测试时,它将使用来自 API 的请求更新你的 HAR 文件。

[
{
"name": "Playwright",
"id": 100
},
// ... 其他水果
]

从 HAR 重放

现在你已经录制了 HAR 文件并修改了模拟数据,可以在测试中使用它来提供匹配的响应。为此,只需关闭或直接移除 update 选项。这样就会针对 HAR 文件运行测试,而不是访问 API。

// 从 HAR 重放 API 请求。
// 要么使用 HAR 中匹配的响应,
// 要么在无匹配项时中止请求。
await page.RouteFromHARAsync("./hars/fruit.har", new ()
{
Url = "*/**/api/v1/fruits",
Update = false,
}
);

// 导航到页面
await page.GotoAsync("https://demo.playwright.dev/api-mocking");

// 断言 Playwright 水果可见
await page.ExpectByTextAsync("Playwright", new() { Exact = true }).ToBeVisibleAsync();

在测试的跟踪信息中,我们可以看到路由是从 HAR 文件中获取的,API 并未被调用。显示正在使用 HAR 文件的跟踪信息

如果检查响应,我们可以看到新水果已添加到 JSON 中,这是通过手动更新 hars 文件夹内经过哈希处理的 .txt 文件完成的。显示来自 HAR 文件的响应的跟踪信息

HAR 重放会严格匹配 URL 和 HTTP 方法。对于 POST 请求,它还会严格匹配 POST 负载。如果有多个记录与请求匹配,将选择匹配头信息最多的记录。导致重定向的条目将自动跟进。

与录制时类似,如果给定的 HAR 文件名以 .zip 结尾,则会将其视为一个存档,其中包含 HAR 文件以及作为单独条目存储的网络负载。你也可以解压此存档,手动编辑负载或 HAR 日志,并指向解压后的 har 文件。所有负载将相对于文件系统上解压后的 har 文件进行解析。

使用命令行界面(CLI)录制 HAR

我们建议使用 update 选项为测试录制 HAR 文件。不过,你也可以使用 Playwright CLI 录制 HAR。

使用 Playwright CLI 打开浏览器,并传递 --save-har 选项以生成 HAR 文件。还可以选择使用 --save-har-glob 选项,仅保存你感兴趣的请求,例如 API 端点。如果 HAR 文件名以 .zip 结尾,相关文件将被分别写入,并全部压缩到一个 zip 文件中。

# 将来自 example.com 的 API 请求保存为 "example.har" 存档。
pwsh bin/Debug/netX/playwright.ps1 open --save-har=example.har --save-har-glob="**/api/**" https://example.com

了解更多关于 高级网络功能 的信息。

模拟 WebSocket

以下代码将拦截 WebSocket 连接,并模拟通过 WebSocket 的整个通信过程,而不是连接到服务器。此示例用 "response" 响应 "request"

await page.RouteWebSocketAsync("wss://example.com/ws", ws => {
ws.OnMessage(frame => {
if (frame.Text == "request")
ws.Send("response");
});
});

或者,你可能希望连接到实际服务器,但在通信过程中拦截消息并对其进行修改或阻止。以下示例修改了页面发送到服务器的部分消息,其余消息保持不变。

await page.RouteWebSocketAsync("wss://example.com/ws", ws => {
var server = ws.ConnectToServer();
ws.OnMessage(frame => {
if (frame.Text == "request")
server.Send("request2");
else
server.Send(frame.Text);
});
});

更多详细信息,请参阅 WebSocketRoute