WebDriver

WebDriver失败截图可以通过两种方式实现：

1. Use WebdriverEventListener

第一步：创建自己的WebDriverEventListener

创建自己的WebDriverEventListener 重写Onexception 方法， 当webdriver 遇到异常的时候执行截图动作。

import java.io.File;
import java.io.IOException;
import java.io.FileOutputStream;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.apache.commons.io.FileUtils;
import org.apache.log4j.LogManager;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.internal.Base64Encoder;
import org.openqa.selenium.remote.ScreenshotException;
//import org.openqa.selenium.support.events.AbstractWebDriverEventListener;
import org.openqa.selenium.support.events.WebDriverEventListener;

public class CustomWebDriverEventListener implements WebDriverEventListener  {
    
    @Override
    public void onException(Throwable paramThrowable, WebDriver paramWebDriver) {
        // TODO Auto-generated method stub
        Throwable cause = paramThrowable.getCause();
        if (cause instanceof ScreenshotException) {
            SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss");
            String dateString = formatter.format(new Date());
            File of = new File(dateString + "-exception.png");
            FileOutputStream out = null;
            try {
                out = new FileOutputStream(of);
                out.write(new Base64Encoder().decode(((ScreenshotException) cause)
                        .getBase64EncodedScreenshot()));
            } catch (Exception e) {
                e.printStackTrace();
            } finally {
                try {
                    out.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

}

第二步：通过EventFiringWebDriver给webdriver注册自己的listener

WebDriver driver = new RemoteWebDriver(new URL(remoteUrl), capabilities);

WebDriverEventListener eventListener = new CustomWebDriverEventListener();

driver = new EventFiringWebDriver(driver).register(eventListener);

注册完就OK了，driver遇到异常的时候就执行异常监控中的截图操作，当然其他事件中的操作也会执行。

2. Use TestNG/JUnit listener

第一步：创建自己的TestListenerAdapter 或者custom rule

TestNG就是创建TestListenerAdapter，重写Onfailure方法， 如果用的是Junit就创建对应的test rule，此处只举例TestNG，Junit的可参照（http://www.ltesting.net/ceshi/open/kygncsgj/selenium/2012/0824/205449_2.html）

Onfailure方法添加截图功能

import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;

import org.apache.log4j.LogManager;
import org.openqa.selenium.WebDriver;
import org.testng.IAnnotationTransformer;
import org.testng.IResultMap;
import org.testng.IRetryAnalyzer;
import org.testng.ITestContext;
import org.testng.ITestNGMethod;
import org.testng.ITestResult;
import org.testng.Reporter;
import org.testng.TestListenerAdapter;
import org.testng.annotations.ITestAnnotation;

public class MyListener extends TestListenerAdapter {
    
    @Override
    public synchronized void onTestFailure(ITestResult result) {
        LogManager.getLogger(this.getClass()).info("In failed method");
        Object currentClass = result.getInstance();  
        ((TestBase) currentClass).takeScreenShot(true);
        LogManager.getLogger(this.getClass()).info("Out failed method");
    }

}

两者比较：

WebdriverEventListener：

1.  可以截图所有webdriver异常的图片，但是不能截取Assert.fail 失败的图片

2. 如果使用while等待某种条件是的报错也会截图，可能造成截图过多

3. 只能用于Webdriver，如果使用remote driver可能不行

4. 坚挺的事件很多，方便很多操作

测试框架TestNG，Junit

1. 用例失败就截图，和Webdriver异常关联不大