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 @Override4 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 @BeforeClass10 public void setUp(){11 driver = new FirefoxDriver();12 driver.manage().window().maximize();13 driver.navigate().to("http://www.baidu.com"); 14 }15 16 @AfterClass17 public void tearDown(){18 driver.close();19 driver.quit();20 }21 }
1 public class Test10 extends TestBase{ 2 3 @Test4 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 @Test5 public void testInput(){6 System.out.println(5/0);7 } 8 9 }
感谢分享:http://www.cnblogs.com/zhangfei