// Selenium Client: selenium-java-2.34.0
在 Selenium WebDriver 测试脚本中添加截图功能,可以使用以下2种方法:
(1) TakesScreenshot (org.openqa.selenium.TakesScreenshot)
File srcFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
FileUtils.copyFile(srcFile, new File(filePath));
此方法可以截取整个页面,且不包含任何浏览器元素。目前已在 Firefox 18, IE 9 及 Chrome 23 等浏览器执行。其中,Firefox 和 IE 截图正常,但 Chrome 截图得到的图片全黑色,于是找到了下面一种解决方案。
(2) Robot (java.awt.Robot)
final Rectangle captureSize = new Rectangle(Toolkit
.getDefaultToolkit().getScreenSize());
final Robot robot = new Robot();
BufferedImage bufferedImage = robot
.createScreenCapture(captureSize);
ImageIO.write(bufferedImage, "png", new File(filePath));
此方法所截取的是整个计算机屏幕,即只包含页面的可见部分,但在上述的浏览器中,均能截图到正常的图片。