Java中用Maven与Selenium测试，如何在测试失败时自动调用截图函数？

完全不需要给每个测试用例套冗余的try-catch块，依托你当前用的Java测试框架自带的失败事件拦截机制就能实现，全程不用改动已写好的业务用例逻辑，只需要改一次公共配置就能全局生效。

核心实现逻辑 #

不管你用TestNG还是JUnit作为测试执行框架，框架本身会自动捕获所有未处理的用例异常（比如你遇到的NoSuchElementException抛出场景）、断言失败的状态，并对外抛出对应的失败事件。你只需要监听这个失败事件，确认用例处于失败状态时，直接调用你已经写好的自定义截图函数即可，不会和现有用例的业务逻辑产生耦合。
相比手动在每个用例里写try-catch捕获异常再调截图的方式，这种方法的代码量减少90%，也不会因为漏写某个用例的捕获块，导致漏掉失败场景的截图。

分框架的具体落地步骤 #

TestNG 实现（Selenium项目最常用的生态搭配） #

• 自定义一个继承TestListenerAdapter的监听器类，重写onTestFailure(ITestResult result)方法
• 方法内先校验当前用例的执行状态确实为失败，再触发截图，避免对跳过执行、执行成功的用例做多余操作
• 直接在方法内调用你已经写好的截图工具类，传入当前失败用例的名称作为截图文件名，加上时间戳避免重名覆盖
  示例代码参考：

public class FailureScreenshotListener extends TestListenerAdapter {
    // 直接注入你已经开发完成的截图工具类即可
    private final ScreenshotUtil screenshotUtil = new ScreenshotUtil();

    @Override
    public void onTestFailure(ITestResult result) {
        // 仅对真正执行失败的用例触发截图
        if (result.getStatus() == ITestResult.FAILURE) {
            String failedCaseName = result.getMethod().getMethodName();
            // 调用现成的截图函数，不需要额外写截图逻辑
            screenshotUtil.captureBrowserPage(failedCaseName + "_" + System.currentTimeMillis() + ".png");
        }
    }
}

监听器的挂载完全无侵入：

• 如果你所有用例都继承了统一的测试基类，只需要在基类上加@Listeners(FailureScreenshotListener.class)注解，所有子类用例都会自动生效
• 如果不想改基类，也可以在testng.xml配置文件的监听器节点添加这个类的全路径，实现全局范围生效

JUnit5 实现 #

• 直接实现JUnit5内置的TestExecutionListener接口，重写testExecutionFailed(ExtensionContext context, Throwable cause)方法即可
• 同样在方法内判断执行状态后，调用已有的截图函数，不需要修改单个用例的代码
  示例代码参考：

public class FailureScreenshotListener implements TestExecutionListener {
    private final ScreenshotUtil screenshotUtil = new ScreenshotUtil();

    @Override
    public void testExecutionFailed(ExtensionContext context, Throwable cause) {
        String failedCaseName = context.getDisplayName();
        screenshotUtil.captureBrowserPage(failedCaseName + "_" + System.currentTimeMillis() + ".png");
    }
}

这个监听器的挂载不需要改任何用例代码，只需要在测试资源目录下新建META-INF/services/org.junit.platform.launcher.TestExecutionListener文件，写入你这个监听器的全类名，测试启动时会自动加载生效。

实操注意事项 #

• 触发截图的时机要匹配Selenium的页面加载状态，避免截到还在加载的空白页面
• 不要把截图逻辑硬塞进@After的公共方法里，时间久了不同用例的后置逻辑耦合，维护成本很高；用监听器的方式是框架层面解耦，后续调整截图逻辑只需要改监听器一处就行
• 建议把截图的保存路径绑定到Maven的测试输出目录（比如target/test-screenshots），后续跑CI流程的时候可以直接把这个目录作为测试附件归档，不会污染源码目录

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