如何在Selenium WebDriver中为截图添加文本？​​

Question

我的问题可能有点鲁莽，但我想知道是否有人有过使用Selenium WebDriver或任何其他Java库在截图中添加文本的经验？目前我正在使用：

File screenShotFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
            FileUtils.copyFile(screenShotFile,  new File("C:\\XXX\\XXX\\SeleniumScreenshots\\" + "png"));

为了拍摄截图并且它正常工作，但有时我需要在屏幕截图中添加一个descritive文字，如：＆＃34;这是因为这个失败...＆＃34;我需要的是页面的某些位置（例如全局页脚，标题，汉堡菜单，某些图像），可以使用xpath表达式找到，获取该位置的屏幕截图并添加带有问题描述的文本。如果有人知道如何实现这一点，请回答示例代码。

Answer 1

没什么！从@andrucz

提供的链接调整代码

WebElement failedElement = driver.findElement(<locate your element>);
File screenShotFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
final BufferedImage image = ImageIO.read(screenShotFile);
Graphics g = image.getGraphics();
g.setFont(g.getFont().deriveFont(30f));
g.drawString("Failed because of this!!", failedElement.getSize().getX(), failedElement.getSize().getY()); //Top-left coordinates of your failed element
g.dispose();
ImageIO.write(image, "png", new File("test.png"));

Answer 2

直接绘制驱动程序返回的屏幕截图：

WebDriver driver = new FirefoxDriver();
driver.get("http://www.google.com/");

// take the screenshot
byte[] img_bytes = ((TakesScreenshot)driver).getScreenshotAs(OutputType.BYTES);
BufferedImage img = ImageIO.read(new ByteArrayInputStream(img_bytes));

// add some text and draw a rectangle
Graphics g = img.getGraphics();
g.setColor(Color.red);
g.setFont(new Font( "SansSerif", Font.BOLD, 14));
g.drawString("My text", 10, 10);
g.drawRect(5, 5, 50, 50);
g.dispose();

// save the image
ImageIO.write(img, "png", new File("screenshot.png"));

如果目标元素在屏幕外，那么您可能需要事先将其滚动到窗口中：

((JavascriptExecutor)driver).executeScript("arguments[0].scrollIntoView(true);", element);