用過 WebDriver 的同學(xué)都知道，WebDriver 可以對瀏覽器中的頁面進行截圖。例如：
public byte[] takeScreenshot() throws IOException { TakesScreenshot takesScreenshot = (TakesScreenshot) driver; return takesScreenshot.getScreenshotAs(OutputType.BYTES);}這樣產(chǎn)生的圖片是整個網(wǎng)頁。但有時候我們并不需要整個網(wǎng)頁，只需要某些特定的 WebElement，以避免一些干擾。
雖然 WebDriver 本身沒有提供這樣的 API，不過我們可以自己來，就是在全屏的截圖中再進行裁剪，如下：
public BufferedImage createElementImage(WebElement webElement) throws IOException { // 獲得webElement的位置和大小。 Point location = webElement.getLocation(); Dimension size = webElement.getSize(); // 創(chuàng)建全屏截圖。 BufferedImage originalImage = ImageIO.read(new ByteArrayInputStream(takeScreenshot())); // 截取webElement所在位置的子圖。 BufferedImage croppedImage = originalImage.getSubimage( location.getX(), localtion.getY(), size.getWidth(), size.getHeight()); return croppedImage;}其中 ImageIO 和 BufferedImage 分別來自 javax.imageio 和 java.awt.image。
這種方法可以很好的削減截圖的大小，也避免了一些不確定因素（比如日期時間等），是測試中保存截圖的很好的方式。
注：如果測試的網(wǎng)頁中含有 iframe，則需要使用更復(fù)雜的方式計算 WebElement 的位置，詳見：計算網(wǎng)頁中某個元素的位置。
新聞熱點
疑難解答
圖片精選
