（转）testng对失败时截图处理

写这篇微博之前，自动化测试代码中有对于失败测试截图的功能，但是却散落在各个catch语句块中。不便于以后的扩展和维护，AOP思想里说明是面向切面编程，把公共的组件提取出来，可以单独修改维护。

但是直到我看了http://www.cnblogs.com/zhangfei/p/4271923.html文章后，觉得应该尝试下了。在此感谢博主张飞的倾情分享！

步骤一，在公共utility包中加入截图方法

几个重点 a. 明确截图文件存放的位置和名字   b. 调用webdriver 下的getScreenshotAs方法截图，并将截图文件copy到指定目录下

直接上张飞的code！

 public class ScreenShot {
     public WebDriver driver;

     public ScreenShot(WebDriver driver) {
         this.driver = driver;
     }

     private void takeScreenshot(String screenPath) {
         try {
             File scrFile = ((TakesScreenshot) driver)
                     .getScreenshotAs(OutputType.FILE);
             FileUtils.copyFile(scrFile, new File(screenPath));
         } catch (IOException e) {
             System.out.println("Screen shot error: " + screenPath);
         }
     }

     public void takeScreenshot() {
         String screenName = String.valueOf(new Date().getTime()) + ".jpg";
         File dir = new File("test-output/snapshot");
         if (!dir.exists())
             dir.mkdirs();
         String screenPath = dir.getAbsolutePath() + "/" + screenName;
         this.takeScreenshot(screenPath);
     }
 }

步骤二，定义一个类，实现TestNG的TestListenerAdapter接口

几个重点： a.得到当前要监听接口的webdriver对象 b.进而去调用utility类的截图方法

public class DotTestListener extends TestListenerAdapter{

	@Override
    public void onTestFailure(ITestResult tr) {
		try {
			loginTest tb = (loginTest) tr.getInstance();
			WebDriver driver=tb.getDriver();
            System.out.println("listener is running");

            ScreenShot ssh = new ScreenShot(driver);
            ssh.takeScreenshot();
            System.out.println("take screenshot ok!");

        } catch (SecurityException e) {
            e.printStackTrace();
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
        }

    }

	}

步骤三，使用监听器

TestNG有两种方式使用监听器

方式一   在 testng.xml 中使用 TestNG 监听器

TestNG 通过 testng.xml 配置所有的测试方法。Testng.xml 提供了 listeners 和 listener 标签用来添加自定义的监听器。下面示范的是本文示例代码中包含的 testng.xml 文件。

 <suite name="TestNGSample">
     <listeners>
         <listener class-name="listeners.OSFilter" />
         <listener class-name="listeners.ProgressTracker" />
     </listeners>
     <test name="ProgressTracker Demo">
         <classes>
             <class name="tests.SampleTest" />
         </classes>
     </test>
 </suite>

方式二  在源代码中使用 TestNG 监听器

通过 @Listeners 注释，可以直接在 Java 源代码中添加 TestNG 监听器。下面示范的是本文示例代码中如何使用 @Listeners 注释。

 @Listeners({ OSFilter.class, ProgressTracker.class })
 public class SampleTest {

     @Test(groups = { OSNames.OS_LINUX })
     public void test1() {
         sleep(5000);
         System.out.println(">>>test1");
     }

我用的是第二种方式。

参考文章：

http://www.cnblogs.com/zhangfei/p/4271923.html