我需要使用Selenium Webdriver添加在一个单词文件中执行的所有步骤的屏幕截图

Question

请帮帮我。

我想针对特定的测试用例，使用带有Java的Selenium Webdriver将所有步骤的屏幕快照添加到一个文档（单词）文件中，并且该文件应与该特定的测试用例名称一起存储。

public static void main(String[] args) {
    try {

        XWPFDocument docx = new XWPFDocument();
        XWPFRun run = docx.createParagraph().createRun();
        FileOutputStream out = new FileOutputStream(System.getProperty("user.dir")+"\\Result\\Screenshot");

        for (int counter = 1; counter <= 5; counter++) {
            captureScreenShot(docx, run, out);
            TimeUnit.SECONDS.sleep(1);
        }

        docx.write(out);
        out.flush();
        out.close();
        docx.close();

    } catch (Exception e) {
        e.printStackTrace();
    }
}

public static void captureScreenShot(XWPFDocument docx, XWPFRun run, FileOutputStream out) throws Exception {

    String screenshot_name = System.currentTimeMillis() + ".png";
    BufferedImage image = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
    File file = new File(System.getProperty("user.dir")+"\\Result\\Screenshot" + screenshot_name);
    ImageIO.write(image, "png", file);
    InputStream pic = new FileInputStream(System.getProperty("user.dir")+"\\Result\\Screenshot" + screenshot_name);
    run.addBreak();
    run.addPicture(pic, XWPFDocument.PICTURE_TYPE_PNG, screenshot_name, Units.toEMU(350), Units.toEMU(350));
    pic.close();
    file.delete();
}