Selenium WebDriver 4.7.2（Java17）下如何捕获弹窗截图

问题背景 #

在Selenium测试中遇到弹窗出现时无法捕获截图的问题，已设置Chrome期望能力：

chromeOptions.setCapability(CapabilityType.UNHANDLED_PROMPT_BEHAVIOUR, "ignore");

但执行以下代码时，第二行截图操作失败，尝试显式等待也无效：

driverManager.getWebDriver().switchTo().alert();
byte[] screenshot = ((TakesScreenshot) driverManager.getWebDriver()).getScreenshotAs(OutputType.BYTES);
scenario.attach(screenshot, "image/png", "attachment-"+LocalTime.now().toSecondOfDay());
driverManager.getWebDriver().switchTo().defaultContent();

核心原因 #

当Driver切换到Alert（原生弹窗）上下文时，TakesScreenshot接口无法正常工作——Alert属于浏览器原生组件，不在页面DOM上下文范围内，此时调用截图方法会直接报错。

解决方案 #

方案1：先截图再处理Alert（推荐，捕获带弹窗的页面） #

在切换到Alert上下文之前完成截图操作，此时Driver仍处于页面上下文，能捕获到包含弹窗的完整页面：

// 先执行截图，此时弹窗已显示但未切换到Alert上下文
byte[] screenshot = ((TakesScreenshot) driverManager.getWebDriver()).getScreenshotAs(OutputType.BYTES);
scenario.attach(screenshot, "image/png", "attachment-"+LocalTime.now().toSecondOfDay());

// 再处理Alert
Alert alert = driverManager.getWebDriver().switchTo().alert();
alert.accept(); // 或alert.dismiss()根据需求选择
driverManager.getWebDriver().switchTo().defaultContent();

方案2：先处理Alert再截图（捕获处理后的页面） #

如果需要的是处理弹窗后的页面截图，先完成Alert操作并切回页面上下文，再执行截图：

// 先处理Alert
Alert alert = driverManager.getWebDriver().switchTo().alert();
alert.accept();
driverManager.getWebDriver().switchTo().defaultContent();

// 回到页面上下文后执行截图
byte[] screenshot = ((TakesScreenshot) driverManager.getWebDriver()).getScreenshotAs(OutputType.BYTES);
scenario.attach(screenshot, "image/png", "attachment-"+LocalTime.now().toSecondOfDay());

方案3：使用第三方库Ashot处理复杂截图场景 #

如果需要更灵活的截图（比如单独捕获弹窗、处理滚动页面等），可以使用Ashot库，它能直接捕获包含原生弹窗的页面，无需切换Alert上下文：

1. 添加Maven依赖：

<dependency>
    <groupId>ru.yandex.qatools.ashot</groupId>
    <artifactId>ashot</artifactId>
    <version>1.5.4</version>
</dependency>

2. 示例代码：

// 直接用Ashot捕获带弹窗的页面
Screenshot screenshot = new AShot().takeScreenshot(driverManager.getWebDriver());

// 将截图转换为byte数组用于附件上传
ByteArrayOutputStream baos = new ByteArrayOutputStream();
ImageIO.write(screenshot.getImage(), "png", baos);
byte[] screenshotBytes = baos.toByteArray();

scenario.attach(screenshotBytes, "image/png", "attachment-"+LocalTime.now().toSecondOfDay());

// 最后处理Alert
Alert alert = driverManager.getWebDriver().switchTo().alert();
alert.accept();

关于UNHANDLED_PROMPT_BEHAVIOUR配置说明 #

该配置仅控制Driver遇到未处理弹窗时的默认行为（设为ignore表示不自动处理弹窗），但无法解决切换Alert上下文后无法截图的问题，核心问题仍在于上下文切换导致的截图接口失效。

内容的提问来源于stack exchange，提问作者sritej_kr