testng对失败时截图处理

天天酷派哦

1. 1.截图类：
   
2. 
   
3. 
   
4. public class ScreenShot {
   
5.     public WebDriver driver;
   
6. 
   
7.     public ScreenShot(WebDriver driver) {
   
8.         this.driver = driver;
   
9.     }
   
10. 
    
11.     private void takeScreenshot(String screenPath) {
    
12.         try {
    
13.             File scrFile = ((TakesScreenshot) driver)
    
14.                     .getScreenshotAs(OutputType.FILE);
    
15.             FileUtils.copyFile(scrFile, new File(screenPath));
    
16.         } catch (IOException e) {
    
17.             System.out.println("Screen shot error: " + screenPath);
    
18.         }
    
19.     }
    
20. 
    
21.     public void takeScreenshot() {
    
22.         String screenName = String.valueOf(new Date().getTime()) + ".jpg";
    
23.         File dir = new File("test-output/snapshot");
    
24.         if (!dir.exists())
    
25.             dir.mkdirs();
    
26.         String screenPath = dir.getAbsolutePath() + "/" + screenName;
    
27.         this.takeScreenshot(screenPath);      
    
28.     }
    
29. }
    
30. 2.我们可以用testng的一个监听器来监听错误时截图：
    
31. 
    
32. public class DotTestListener extends TestListenerAdapter {
    
33. 
    
34.     @Override
    
35.     public void onTestFailure(ITestResult tr) {   
    
36.          
    
37.     }
    
38. 
    
39. }
    
40. 3.也就是说我们只需要在onTestFailure方法里面调用ScreenShot类里面的takeScreenshot方法即可，

复制代码