【教程】browsermob-proxy 基于Java的代理服务配合selenium使用

2024 年 03 月 12 日

171 次浏览

3290字数

BrowserMob Proxy是一个基于Java的代理工具，用于捕获浏览器的HTTP流量。当与Selenium WebDriver结合使用时，它可以很好地用于监控和操作Web自动化测试中的网络流量。下面是如何配置BrowserMob Proxy，并将其与Selenium WebDriver配合使用的教程。

安装BrowserMob Proxy

首先，需要下载BrowserMob Proxy。可以从其官方GitHub存储库或其他可靠的源获取最新版本的BrowserMob Proxy。

启动BrowserMob Proxy

一旦获得了BrowserMob Proxy，您需要启动一个BrowserMob Proxy实例。以下是在Java代码中启动它的示例：

import net.lightbody.bmp.BrowserMobProxy;
import net.lightbody.bmp.BrowserMobProxyServer;

public class ProxyExample {
    public static void main(String[] args) {
        // 创建一个BrowserMob Proxy实例
        BrowserMobProxy proxy = new BrowserMobProxyServer();
    
        // 启动代理
        proxy.start(0);
    
        // 获取代理服务器的端口号
        int port = proxy.getPort();
        System.out.println("Proxy is running on port: " + port);
    }
}

配置Selenium以使用BrowserMob Proxy

要让Selenium WebDriver通过BrowserMob Proxy运行，需要将代理配置到WebDriver中。这里以Firefox为例进行配置：

import org.openqa.selenium.Proxy;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxOptions;

public class SeleniumWithProxyExample {
    public static void main(String[] args) {
        // 假设BrowserMob Proxy运行在8080端口
        int proxyPort = 8080;
    
        // 设置代理
        Proxy seleniumProxy = new Proxy();
        seleniumProxy.setHttpProxy("localhost:" + proxyPort);
        seleniumProxy.setSslProxy("localhost:" + proxyPort);

        // 配置Firefox以使用代理
        FirefoxOptions options = new FirefoxOptions();
        options.setProxy(seleniumProxy);
    
        // 创建WebDriver实例
        WebDriver driver = new FirefoxDriver(options);
    
        // ... 执行你的Selenium测试步骤
    }
}

使用BrowserMob Proxy进行网络监控和操作

BrowserMob Proxy允许你截获和操纵HTTP请求和响应。例如，您可以添加自定义HTTP请求头，断言响应，甚至模拟网络条件。

// 添加自定义请求头
proxy.addRequestFilter((request, contents, messageInfo) -> {
    request.headers().add("Custom-Header", "Test-Value");
    return null;
});

// 添加断言验证响应状态码
proxy.addResponseFilter((response, contents, messageInfo) -> {
    assert response.getStatus().code() == 200;
});

// 设置网络延迟模拟
proxy.setLatency(100, TimeUnit.MILLISECONDS); // 增加100ms的延迟

停止BrowserMob Proxy

测试结束后，应当关闭BrowserMob Proxy实例以释放资源。

proxy.stop();

整合到测试框架

将上面的步骤整合到您的测试框架中，例如JUnit或TestNG，即可实现代理服务的启动、配置及关闭。可根据测试框架提供的@Before和@After（或者其它的生命周期钩子）进行对应的设置和清理。

小结

BrowserMob Proxy与Selenium WebDriver的结合使用能够为Web自动化测试提供了强大的网络流量捕获和处理能力。通过上述指南，你可以设置代理服务，进而监控，断言和操作HTTP请求和响应，以及模拟不同的网络环境对应用进行全面的测试。这些技术支持工程师更细致地验证应用在各种网络状态下的表现，提高测试的准确性和效果。