Playwright进阶指南 (5)：拦截与模拟网络请求

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")

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

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_()

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

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

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)$"), lambda r: r.abort())  # 阻止图片加载

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