Java + Cucumber失败时的屏幕截图

import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import javax.imageio.ImageIO;

public class ScreenshotUtil {
    public static void captureScreenshot(String filePath) {
        try {
            // 创建一个Robot对象
            Robot robot = new Robot();
            
            // 获取屏幕尺寸
            Dimension screenSize = Toolkit.getDefaultToolkit().getScreenSize();
            
            // 创建一个矩形，表示屏幕区域
            Rectangle screenRect = new Rectangle(screenSize);
            
            // 截图
            BufferedImage screenshot = robot.createScreenCapture(screenRect);
            
            // 保存截图到指定文件路径
            ImageIO.write(screenshot, "png", new File(filePath));
            
            System.out.println("屏幕截图保存成功：" + filePath);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

import io.cucumber.java.After;
import io.cucumber.java.Scenario;

public class StepDefinitions {
    private Scenario scenario;
    
    // 注入Scenario对象
    public StepDefinitions(Scenario scenario) {
        this.scenario = scenario;
    }
    
    // 在测试失败时进行屏幕截图
    @After
    public void captureScreenshotOnFailure() {
        if (scenario.isFailed()) {
            String screenshotFilePath = "path/to/save/screenshot.png";
            ScreenshotUtil.captureScreenshot(screenshotFilePath);
            
            // 将截图文件路径添加到测试报告中
            scenario.attach(screenshotFilePath, "image/png", "失败截图");
        }
    }
    
    // 其他测试步骤的定义...
}