Playwright进阶篇 (5) | 网络请求拦截与Mock

在现代Web应用测试中，精准控制网络请求是提升测试效率和可靠性的关键。Playwright 提供了强大的网络请求拦截与Mock能力，让你能够：

拦截并修改 请求头、参数、URL
模拟API响应 无需依赖真实后端
构造异常场景 测试边缘情况
加速测试执行 避免真实网络延迟

一、核心概念：路由与请求处理

from playwright.sync_api import sync_playwright

with sync_playwright() as p:
    browser = p.chromium.launch()
    context = browser.new_context()
    page = context.new_page()
    
    # 核心路由拦截方法
    def handle_route(route):
        if"api/v1/user"in route.request.url:
            # 在此处处理特定请求
            ...
        route.continue_()
    
    # 注册全局路由处理器
    page.route("**/*", handle_route)
    
    page.goto("https://your-app.com")

二、实战技巧：4种典型应用场景

场景1：修改请求参数（如添加认证Token）

def handle_route(route):
    headers = route.request.headers
    headers["Authorization"] = "Bearer mock_token"  # 注入模拟Token
    route.continue_(headers=headers)  # 修改后继续请求

场景2：拦截特定请求返回Mock数据

def mock_user_api(route):
    if route.request.url.endswith("/api/user"):
        # 构造JSON响应
        route.fulfill(
            status=200,
            content_type="application/json",
            body=json.dumps({"name": "测试用户", "id": 1001})
        )
    else:
        route.continue_()

场景3：模拟网络异常（测试错误处理）

def simulate_failure(route):
    if "payment-api" in route.request.url:
        # 返回500错误
        route.fulfill(status=500, body="Service Unavailable")
    else:
        route.continue_()

场景4：替换资源文件（如修改CSS/JS）

def replace_styles(route):
    if route.request.url.endswith(".css"):
        route.fulfill(
            body="* { color: red !important; }"  # 强制所有文字变红
        )

三、高级技巧：动态Mock服务

结合 pytest 实现自动化Mock：

import pytest
from playwright.sync_api import Page

@pytest.fixture
def setup_mock(page: Page):
    def mock_handler(route):
        if"/data"in route.request.url:
            # 根据请求参数动态响应
            params = route.request.url.split("?")[1]
            route.fulfill(body=f"Response to {params}")
    
    page.route("**/api/*", mock_handler)
    yield

def test_dynamic_mock(setup_mock, page: Page):
    page.goto("https://app.com")
    # 测试将自动获得动态Mock响应

四、调试技巧：实时监控网络请求

# 监听所有请求
page.on("request", lambda req: print(f">> {req.method} {req.url}"))

# 监听所有响应
page.on("response", lambda res: print(f"<< {res.status} {res.url}"))

# 捕获请求失败
page.on("requestfailed", lambda req: print(f"!! {req.url} {req.failure}"))

五、性能优化：选择性拦截

避免不必要的处理开销：

# 精确匹配URL
page.route("https://api.example.com/v1/users", handler)

# 使用正则表达式
page.route(re.compile(r".*\.(jpg|png)#34;), lambda r: r.abort())  # 阻止图片加载

# 按请求类型过滤
def handle_post(route):
    if route.request.method == "POST":
        ...

六、Mock数据管理建议

分层管理：
基础Mock：放置在 conftest.py
场景Mock：封装为独立模块
数据工厂：def generate_user_data(role="user"):
return {
"name": f"test_{role}",
"permissions": ["read"] if role=="user" else ["read","write"]
}
使用模板引擎：from jinja2 import Template

template = Template('{"user": "{{ name }}", "role": "{{ role }}"}')
route.fulfill(body=template.render(name="admin", role="superuser"))

七、注意事项

作用域控制：
使用 browser_context.route() 实现上下文隔离
通过 page.unroute() 及时清理避免污染
真实请求验证：# 临时禁用Mock进行验证
with page.expect_request("**/api/data") as req_info:
page.click("#refresh-btn")
request = req_info.value
assert request.post_data == '{"valid":true}'
资源释放：# 测试结束后移除路由
def teardown():
page.remove_route(handler)