soga! selenium在执行用例失败时如何有效截取失败画面

 1 package com.daanhealth.tnb.util;
 2 
 3 import java.io.File;
 4 
 5 import org.apache.commons.io.FileUtils;
 6 import org.openqa.selenium.OutputType;
 7 import org.openqa.selenium.TakesScreenshot;
 8 import org.openqa.selenium.WebDriver;
 9 
10 public class ScreenShot {
11     
12     public WebDriver driver;
13     
14     public ScreenShot(WebDriver driver){
15         this.driver=driver;
16     }
17     
18     private void takeScreenshot(String screenPath){
19         try {
20             File srcFile=((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
21             FileUtils.copyFile(srcFile, new File(screenPath));
22         } catch (Exception e) {
23             // TODO: handle exception
24             System.out.println("Screen shot error: " + screenPath);
25         }
26     }
27     
28     public void takeScreenshot(){
29         String screenName=new DateFormat().formatDateToString(System.currentTimeMillis())+".jpg";
30         File dir=new File("test-output/snapshot");
31         if(!dir.exists()){
32             dir.mkdirs();
33         }
34         String screenPath=dir.getAbsolutePath()+File.separator+screenName;
35         takeScreenshot(screenPath);
36     }
37 
38 }