Selenium网络拦截：请求Mock与响应修改全攻略

最新推荐文章于 2025-10-30 17:21:45 发布
原创 最新推荐文章于 2025-10-30 17:21:45 发布 · 1k 阅读 ·
CC 4.0 BY-SA版权
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

Selenium网络拦截：请求Mock与响应修改全攻略

【免费下载链接】selenium SeleniumHQ/selenium: Selenium是一个开源自动化测试工具套件，支持多种浏览器和语言环境。它可以模拟真实用户的行为来驱动浏览器自动执行各种操作，广泛应用于Web应用程序的功能测试、回归测试以及端到端测试场景。 【免费下载链接】selenium 项目地址: https://gitcode.com/GitHub_Trending/se/selenium

引言：为什么需要网络拦截？

在Web应用测试过程中，你是否遇到过以下痛点：

  • 依赖第三方API导致测试不稳定
  • 支付接口等敏感操作无法在测试环境执行
  • 需要模拟各种异常响应（如500错误、超时等）
  • 测试环境数据不一致影响测试结果

Selenium的网络拦截功能（Network Interception）通过BiDi（双向通信协议）提供了强大的请求控制能力，让你能够完全掌控浏览器与服务器之间的通信。本文将系统介绍如何使用Selenium进行请求Mock和响应修改，解决上述测试难题。

读完本文后，你将能够：

  • 拦截并修改HTTP请求参数、 headers和请求体
  • 模拟各种响应状态码和响应内容
  • 处理认证请求和敏感操作
  • 构建稳定、可重复的自动化测试环境

Selenium网络拦截原理

Selenium通过BiDi协议实现网络拦截，其核心组件包括：

mermaid

Selenium网络拦截主要涉及以下关键类和方法：

类/方法作用
driver.network.add_request_handler()添加请求拦截处理器
Request.continue_request()继续处理拦截的请求
Request.fail_request()使拦截的请求失败
Request封装拦截到的请求对象

环境准备与基础配置

安装依赖

确保你的Selenium版本支持BiDi网络拦截功能（Selenium 4.0+）：

pip install selenium>=4.0.0

启用BiDi支持

在初始化浏览器驱动时，需要启用BiDi协议支持：

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

chrome_options = Options()
chrome_options.add_argument("--enable-bidi")  # 启用BiDi支持

driver = webdriver.Chrome(options=chrome_options)
driver.implicitly_wait(10)

基础架构搭建

以下是一个基础的网络拦截测试框架：

class NetworkInterceptor:
    def __init__(self, driver):
        self.driver = driver
        self.callbacks = {}
        
    def register_request_handler(self, event, callback, url_patterns=None):
        """注册请求处理器"""
        callback_id = self.driver.network.add_request_handler(
            event, callback, url_patterns=url_patterns
        )
        self.callbacks[event] = callback_id
        return callback_id
        
    def unregister_all_handlers(self):
        """移除所有已注册的处理器"""
        for event, callback_id in self.callbacks.items():
            self.driver.network.remove_request_handler(event, callback_id)
        self.callbacks.clear()

# 使用示例
interceptor = NetworkInterceptor(driver)

请求拦截与修改实战

1. 基础请求拦截

拦截所有请求并打印请求信息：

def log_request(request):
    print(f"拦截到请求: {request.method} {request.url}")
    print(f"请求头: {request.headers}")
    # 继续处理请求
    request.continue_request()

# 注册请求拦截器
interceptor.register_request_handler("before_request", log_request)

# 访问目标页面
driver.get("https://example.com")

# 清理
interceptor.unregister_all_handlers()

2. 按URL模式拦截请求

只拦截特定URL模式的请求：

def mock_api_request(request):
    print(f"拦截到API请求: {request.url}")
    # 修改请求URL
    if "api.example.com/users" in request.url:
        request.continue_request(url="https://mock.example.com/test-users")
    else:
        request.continue_request()

# 只拦截API请求
url_patterns = [{"type": "string", "pattern": "api.example.com"}]
interceptor.register_request_handler(
    "before_request", 
    mock_api_request,
    url_patterns=url_patterns
)

driver.get("https://example.com")

3. 修改请求方法和请求体

拦截POST请求并修改请求体：

def modify_post_request(request):
    if request.method == "POST" and "/submit-form" in request.url:
        # 修改请求方法为GET
        # 修改请求体内容
        new_body = "username=test&email=test@example.com"
        request.continue_request(method="GET", body=new_body)
    else:
        request.continue_request()

interceptor.register_request_handler("before_request", modify_post_request)
driver.get("https://example.com/form")
# 执行表单提交操作...

4. 模拟请求失败

模拟特定请求失败场景：

def fail_sensitive_requests(request):
    # 模拟支付接口请求失败
    if "payment-gateway.com/charge" in request.url:
        print(f"阻止敏感请求: {request.url}")
        request.fail_request()  # 使请求失败
    else:
        request.continue_request()  # 其他请求正常继续

interceptor.register_request_handler("before_request", fail_sensitive_requests)
driver.get("https://example.com/checkout")
# 触发支付流程...

高级应用：响应修改与Mock服务

1. 模拟JSON响应

拦截API请求并返回自定义JSON响应：

def mock_api_response(request):
    if "api.example.com/products" in request.url:
        # 构造自定义JSON响应
        mock_response = {
            "status": 200,
            "headers": {"Content-Type": "application/json"},
            "body": '[{"id": 1, "name": "测试产品", "price": 99.99}]'
        }
        
        # 修改请求以返回自定义响应
        request.continue_request(
            url="data:application/json," + mock_response["body"],
            headers=mock_response["headers"]
        )
    else:
        request.continue_request()

interceptor.register_request_handler("before_request", mock_api_response)
driver.get("https://example.com/products")

2. 模拟错误响应

测试应用在遇到错误响应时的表现：

def simulate_server_errors(request):
    # 随机模拟500错误
    import random
    if "api.example.com/random-data" in request.url and random.random() < 0.3:
        # 使用data URL返回500错误页面
        error_html = """
        <html>
            <head><title>500 Internal Server Error</title></head>
            <body>Server Error</body>
        </html>
        """
        request.continue_request(
            url=f"data:text/html,{error_html}",
            headers={"Status": "500 Internal Server Error"}
        )
    else:
        request.continue_request()

interceptor.register_request_handler("before_request", simulate_server_errors)
driver.get("https://example.com/dashboard")

3. 处理认证请求

自动处理需要认证的请求：

def handle_authentication_requests(request):
    # 检测需要认证的请求
    if "WWW-Authenticate" in (request.headers or {}):
        # 提供认证凭据
        request._continue_with_auth(username="testuser", password="testpass")
    else:
        request.continue_request()

# 注册认证处理器
auth_callback_id = driver.network.add_auth_handler("testuser", "testpass")
driver.get("https://protected.example.com")
# 完成后移除认证处理器
driver.network.remove_auth_handler(auth_callback_id)

实际测试场景应用

1. 前端性能测试：模拟慢网络

def throttle_network(request):
    # 为图片资源添加延迟
    if request.resource_type == "image":
        # 通过修改URL指向延迟服务
        delayed_url = f"https://slow-server.example.com/delay/2000?url={request.url}"
        request.continue_request(url=delayed_url)
    else:
        request.continue_request()

interceptor.register_request_handler("before_request", throttle_network)
# 测量页面加载性能...

2. 测试缓存机制

def control_cache_headers(request):
    if "static/css" in request.url or "static/js" in request.url:
        # 修改缓存头，强制不缓存
        new_headers = request.headers.copy()
        new_headers["Cache-Control"] = "no-cache, no-store, must-revalidate"
        new_headers["Pragma"] = "no-cache"
        new_headers["Expires"] = "0"
        request.continue_request(headers=new_headers)
    else:
        request.continue_request()

interceptor.register_request_handler("before_request", control_cache_headers)
# 测试缓存机制...

3. A/B测试内容切换

def ab_test_content_switch(request):
    if "example.com/main-content" in request.url:
        # 根据用户ID分配不同版本
        user_id = "test-user-123"
        version = "variant-b" if hash(user_id) % 2 == 0 else "variant-a"
        
        # 请求不同版本的内容
        request.continue_request(
            url=f"https://example.com/content/{version}"
        )
    else:
        request.continue_request()

interceptor.register_request_handler("before_request", ab_test_content_switch)
driver.get("https://example.com")
# 验证A/B测试内容是否正确加载...

最佳实践与注意事项

1. 性能优化

网络拦截会对性能产生一定影响，建议：

# 只拦截需要的请求
specific_patterns = [
    {"type": "string", "pattern": "api.example.com"},
    {"type": "string", "pattern": "/api/"}
]

# 避免在非必要时使用通配符
# 及时清理不再需要的拦截器
def test_specific_interception():
    callback_id = driver.network.add_request_handler(
        "before_request", 
        callback,
        url_patterns=specific_patterns
    )
    
    # 执行测试操作...
    
    # 测试完成后立即移除拦截器
    driver.network.remove_request_handler("before_request", callback_id)

2. 调试技巧

使用详细日志追踪拦截过程：

def debug_interceptor(request):
    print(f"拦截到请求: {request.method} {request.url}")
    print(f"请求头: {request.headers}")
    print(f"资源类型: {request.resource_type}")
    
    # 记录请求信息到文件
    with open("network_logs.txt", "a") as f:
        f.write(f"URL: {request.url}\n")
        f.write(f"方法: {request.method}\n")
        f.write(f"时间戳: {datetime.now()}\n\n")
    
    request.continue_request()

3. 跨浏览器兼容性

不同浏览器对BiDi协议的支持程度不同：

def create_driver_with_bidi():
    """创建支持BiDi的浏览器驱动，处理不同浏览器兼容性"""
    browser = os.getenv("BROWSER", "chrome")
    
    if browser == "chrome":
        options = webdriver.ChromeOptions()
        options.add_argument("--enable-bidi")
        return webdriver.Chrome(options=options)
    elif browser == "firefox":
        options = webdriver.FirefoxOptions()
        options.set_capability("webSocketUrl", True)  # Firefox启用BiDi的方式
        return webdriver.Firefox(options=options)
    else:
        raise ValueError(f"不支持的浏览器: {browser}")

常见问题解决

1. 拦截器不生效

检查以下几点：

  • 确认Selenium版本 >= 4.0
  • 确保正确启用了BiDi支持
  • 验证URL模式是否正确匹配
  • 检查是否有其他拦截器干扰

2. 页面加载超时

网络拦截可能导致页面加载延迟，适当增加超时时间：

driver.set_page_load_timeout(30)  # 设置页面加载超时为30秒

3. 复杂请求处理顺序

多个拦截器的执行顺序可能影响结果：

# 先注册通用拦截器，再注册特定拦截器
# 特定拦截器会覆盖通用拦截器的设置
def general_handler(request):
    # 通用处理逻辑...
    request.continue_request()

def specific_handler(request):
    # 特定处理逻辑...
    request.continue_request()

# 注册顺序很重要
driver.network.add_request_handler("before_request", general_handler)
driver.network.add_request_handler("before_request", specific_handler)

总结与展望

Selenium的网络拦截功能为Web测试带来了革命性的变化，使测试工程师能够完全控制网络请求流程，解决了传统测试中依赖外部服务和环境不稳定的痛点。

本文介绍了Selenium网络拦截的核心概念、基础操作和高级应用，包括：

  • 请求拦截与修改
  • 响应Mock与错误模拟
  • 认证处理与安全测试
  • 实际测试场景应用

随着Web技术的发展，Selenium的BiDi协议支持将不断完善，未来可能会提供更多高级功能，如WebSocket拦截、流式响应处理等。掌握网络拦截技术，将使你能够构建更强大、更稳定的自动化测试框架。

建议结合官方文档和实际项目需求，进一步探索Selenium网络拦截的潜力，为你的测试工作带来更多可能性。

【免费下载链接】selenium SeleniumHQ/selenium: Selenium是一个开源自动化测试工具套件，支持多种浏览器和语言环境。它可以模拟真实用户的行为来驱动浏览器自动执行各种操作，广泛应用于Web应用程序的功能测试、回归测试以及端到端测试场景。 【免费下载链接】selenium 项目地址: https://gitcode.com/GitHub_Trending/se/selenium

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值