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();

　　}

　　}

　　}

　　}

　　}

　　主要看 onException 这个方法的实现，很明显， 我们捕获了这个异常， 然后通过强制转换将图片提取出来，写入硬盘。

　　然后是使用这个监听器， 通常会在 setup 方法里面将这个监听器注册到 WebDriver 中去， 看代码：

　　@Test

　　public void setup(){

　　String remote_driver_url = "http://localhost:4444/wd/hub";

　　DesiredCapabilities capability = null;

　　capability = DesiredCapabilities.firefox();

　　WebDriverEventListener eventListener = new CustomWebDriverEventListener ();

　　WebDriver driver = new EventFiringWebDriver(new RemoteWebDriver(new URL(

　　remote_driver_url)， capability)).register(eventListener);

　　}

　　在这之后，如果运行出错， WebDriver 抛出异常会在相应的 classpath 下面生成 png 的截图。

　　自定义 TestRule

　　和自定义 WebDriver 监听器不同， 自定义 TestRule 只有在这个 Rule 被执行的时候， 才去做一些我们预设的 CallBack。 所以这个截图动作，对于 WebDriver 而言， 是主动的。 那么，我们需要自定义一个 RemoteWebDriver 来实现截图功能。 WebDriver 自身提供了 TakesScreenshot 这个接口， 我们只要实现它可以了， 看代码：

　　import java.net.URL;

　　import org.openqa.selenium.OutputType;

　　import org.openqa.selenium.TakesScreenshot;

　　import org.openqa.selenium.WebDriverException;

　　import org.openqa.selenium.remote.CapabilityType;

　　import org.openqa.selenium.remote.DesiredCapabilities;

　　import org.openqa.selenium.remote.DriverCommand;

　　import org.openqa.selenium.remote.RemoteWebDriver;

　　public class CustomRemoteWebDriver extends RemoteWebDriver implements

　　TakesScreenshot {

　　public CustomRemoteWebDriver(URL url， DesiredCapabilities dc) {

　　super(url， dc);

　　}

　　@Override

　　public X getScreenshotAs(OutputType target)

　　throws WebDriverException {

　　if ((Boolean) getCapabilities().getCapability(

　　CapabilityType.TAKES_SCREENSHOT)) {

　　return target

　　.convertFromBase64Png(execute(DriverCommand.SCREENSHOT)

　　.getValue().toString());

　　}

　　return null;

　　}

　　}