testng对失败时截图处理

1.截图类：

 1 public class ScreenShot {
 2     public WebDriver driver;
 3  
 4     public ScreenShot(WebDriver driver) {
 5         this.driver = driver;
 6     }
 7  
 8     private void takeScreenshot(String screenPath) {
 9         try {
10             File scrFile = ((TakesScreenshot) driver)
11                     .getScreenshotAs(OutputType.FILE);
12             FileUtils.copyFile(scrFile, new File(screenPath));
13         } catch (IOException e) {
14             System.out.println("Screen shot error: " + screenPath);
15         }
16     }
17  
18     public void takeScreenshot() {
19         String screenName = String.valueOf(new Date().getTime()) + ".jpg";
20         File dir = new File("test-output/snapshot");
21         if (!dir.exists())
22             dir.mkdirs();
23         String screenPath = dir.getAbsolutePath() + "/" + screenName;
24         this.takeScreenshot(screenPath);       
25     }
26 }

2.我们可以用testng的一个监听器来监听错误时截图：

1 public class DotTestListener extends TestListenerAdapter {
2  
3     @Override
4     public void onTestFailure(ITestResult tr) {    
5          
6     }
7  
8 }

3.也就是说我们只需要在onTestFailure方法里面调用ScreenShot类里面的takeScreenshot方法即可，但是我们注意到ScreenShot类里需要传一个driver进去。

现在问题来了，对于driver的处理，各式各样，有的用到了单子模式，即把driver当成一个全局的静态变量，在哪都可以用，所以ScreenShot类里可以访问得到driver对象，但这样也就有一个问题，即全局只有一个driver,如果想多线程运行时，启多个driver实例时，用这种方式就做不到了，于是出现了另外一种处理方式，即每一个类或者每一个测试方法是，启一个新的driver对象，这样，driver对象就不是全局的了，就是类对象属性了，比如：

 1 public class TestBase {
 2      
 3     public WebDriver driver;   
 4      
 5     public WebDriver getDriver() {
 6         return driver;
 7     }
 8  
 9     @BeforeClass
10     public void setUp(){
11         driver = new FirefoxDriver();
12         driver.manage().window().maximize();
13         driver.navigate().to("http://www.baidu.com");      
14     }
15      
16     @AfterClass
17     public void tearDown(){
18         driver.close();
19         driver.quit();
20     }
21 }

1 public class Test10 extends TestBase{  
2      
3     @Test
4     public void testInput(){
5         System.out.println(5/0);
6     }  
7      
8 }

那如何把这个类对象的driver属性给传到onTestFailure方法里去？

 1 public class DotTestListener extends TestListenerAdapter {
 2  
 3     @Override
 4     public void onTestFailure(ITestResult tr) {    
 5         try {
 6             TestBase tb = (TestBase) tr.getInstance();
 7             WebDriver driver = tb.getDriver();         
 8             System.out.println(driver.getTitle());
 9         } catch (SecurityException e) {
10             e.printStackTrace();
11         } catch (IllegalArgumentException e) {         
12             e.printStackTrace();
13         }
14     }
15  
16 }

最后再加上监听即可：

1 @Listeners({ DotTestListener.class })
2 public class Test10 extends TestBase{  
3      
4     @Test
5     public void testInput(){
6         System.out.println(5/0);
7     }  
8      
9 }

感谢分享：http://www.cnblogs.com/zhangfei