怎么用Selenium webdriver進行截圖操作

1.IE瀏覽器下截圖：

public static void takeScreenShotIE(WebDriver driver, String name){
  WebDriver augmentedDriver = new Augmenter().augment(driver);
  File output = ((TakesScreenshot) augmentedDriver).getScreenshotAs(OutputType.FILE);
  File file = new File(screenshootDir, name + ".png");
  try {
      FileUtils.copyFile(output, file);
  } catch (IOException e) {
      e.printStackTrace();
  }
} 

 

2.Chrome&FireFox瀏覽器下截圖：

public static void takeScreenShotChromeFirefox(WebDriver driver, String name){
  File output = null;
  File file;
  output = ((TakesScreenshot) driver)
                  .getScreenshotAs(OutputType.FILE);
  file = new File(screenshootDir, name + ".png");
  try {
      FileUtils.copyFile(output, file);
  } catch (IOException e) {
      e.printStackTrace();
  }
}