java - HtmlUnitDriver 浏览器模拟模式

Question

我创建了一个简单的工厂方法，可以在指定模式下创建 HtmlUnitDriver。例如

public static HtmlUnitDriver createHtmlUnitDriver(String browserMode) {
    switch(browserMode.toLowerCase()) {
        case "chrome":
            return new HtmlUnitDriver(DesiredCapabilities.chrome());
        case "firefox":
            return new HtmlUnitDriver(DesiredCapabilities.firefox());
        case "ie":
        case "internet explorer":
            return new HtmlUnitDriver(DesiredCapabilities.internetExplorer());
        case default:
            return new HtmlUnitDriver(true);
    }
}

我想为这个方法写一个快速的单元测试，但我看不出如何判断返回的“模式”HtmlUnitDriver是什么。我该怎么做？

Answer 1

使用反射，可以得到私有字段webClient

@Test
public void test() throws Exception {
    WebDriver driver = new HtmlUnitDriver(BrowserVersion.CHROME);
    WebClient webClient = (WebClient) get(driver, "webClient");
    System.out.println(webClient.getBrowserVersion());
    System.out.println(webClient.getBrowserVersion().isIE());
    driver.close();
}

private static Object get(Object object, String field) throws Exception {
    Field f = object.getClass().getDeclaredField(field);
    f.setAccessible(true);
    return f.get(object);
}