将ssl证书添加到selenium-webdriver

时间:2019-05-07 06:58:51

标签: java selenium selenium-webdriver ssl-certificate browsermob-proxy

我将硒用于chromeDriver进行端到端测试。要测试的网站需要ssl证书。手动打开浏览器时,会弹出一个窗口,供我选择已安装的证书。不同的测试访问不同的URL,并且还需要不同的证书。但是,如果我以无头模式运行测试,则不会弹出窗口。因此,我需要一种以编程方式设置要用于当前测试的证书(例如,设置.pem文件)的方法。

我该如何实现? 我尝试设置一个browserMob代理,然后将其配置为selenium中的代理-但是,这似乎没有任何作用...是否有更好的方法?我究竟做错了什么?这是我尝试过的:

PemFileCertificateSource pemFileCertificateSource = new PemFileCertificateSource(
        new File("myCertificate.pem"),
        new File("myPrivateKey.pem"),
        "myPrivateKeyPassword");

ImpersonatingMitmManager mitmManager = ImpersonatingMitmManager.builder()
        .rootCertificateSource(pemFileCertificateSource)
        .build();

BrowserMobProxy browserMobProxy = new BrowserMobProxyServer();
browserMobProxy.setTrustAllServers(true);
browserMobProxy.setMitmManager(mitmManager);

browserMobProxy.start(8080);


ChromeOptions chromeOptions = new ChromeOptions();
chromeOptions.setProxy(ClientUtil.createSeleniumProxy(browserMobProxy));

WebDriver webDriver = new ChromeDriver(chromeOptions);

// use the webdriver for tests, e.g. assertEquals("foo", webDriver.findElement(...))

1 个答案:

答案 0 :(得分:4)

因此,显然,对于BrowserMob,这是不可能的。因此,我编写了一个代理扩展SeleniumSslProxy,可以将其插入Selenium并添加基于证书的身份验证以创建HTTPS连接。

它是这样工作的:

  • 使用BrowserMob拦截Selenium HTTP请求
  • 设置一个SSLContext给出的证书(.pfx文件)和密码
  • 使用discussion将请求转发到目标URL
  • 将okhttp Response转换为净额FullHttpResponse,以便Selenium可以处理

您可以在okhttp上找到代码。这是一个如何在Selenium端到端测试中使用它的示例(也可以在无头模式下使用):

@Before
public void setup() {
    ClassLoader classLoader = ClassLoader.getSystemClassLoader();
    File clientSslCertificate = new File(
        classLoader.getResource("certificates/some-certificate.pfx").getFile());
    String certificatePassword = "superSecret";

    this.proxy = new SeleniumSslProxy(clientSslCertificate, certificatePassword);
    this.proxy.start();

    ChromeOptions chromeOptions = new ChromeOptions();
    chromeOptions.setProxy(proxy);
    this.webDriver = new ChromeDriver(chromeOptions);
}

@Test
public void pageTitleIsFoo() {
    // given
    String url = "http://myurl.lol";
    // NOTE: do not use https in the URL here. It will be converted to https by the proxy.

    // when
    this.webDriver.get(url);
    this.webDriver.manage().timeouts().implicitlyWait(5, SECONDS);

    // then
    WebElement title = this.webDriver.findElement(By.className("title"));
    assertEquals("Foo", title.getText());
}

@After
public void teardown() {
    this.webDriver.quit();
    this.proxy.stop();
}

请注意,我只使用chromeDriver,从未使用其他驱动程序对其进行测试。与其他驱动程序一起使用时,可能需要对SeleniumSslProxy进行细微调整。