HtmlUnitDriver 浏览器模拟模式

HtmlUnitDriver browser emulate mode

我创建了一个简单的工厂方法,它以指定的模式创建一个 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 在哪个 "mode" 中。我该怎么做?

利用反射,可以得到私有字段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);
}