Selenium自动化之截图保存现场

前提：
自动化中截图功能是直接呈现问题，帮助定位的一种很好的手段。
下面总结一下，我用Selenium做自动化是如何保留截图的。

环境和工具：
eclipse、Selenium、Testng、Reporter

场景：
使用Selenium提供的方法来截取屏幕，使用Reporter监听器，当用例执行失败时截图，截图以出错时系统时间和出错方法的拼接命名，截图保存到项目目录下snapshot

1、编写获取屏幕截图类
ScreenShot.java

package Tool;

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

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.testng.Reporter;

public class ScreenShot {

    public static void takeScreenshot(String screenPath, WebDriver driver) {
        try {
            File scrFile = ((TakesScreenshot) driver)
                    .getScreenshotAs(OutputType.FILE);
            FileUtils.copyFile(scrFile, new File(screenPath));
        } catch (IOException e) {
            System.out.println("Screen shot error: " + screenPath);
            Reporter.log("该错误可以查看截图："+screenPath);
        }

    }

    public static void takeScreenshot(WebDriver driver) {
        String screenName=CommonFunction.getFormatDate()+".jpg";
        File dir = new File("snapshot");
        if (!dir.exists())
            dir.mkdirs();
        String screenPath = dir.getAbsolutePath() + "/" + screenName;
        takeScreenshot(screenPath, driver);
    }

}

2、修改监听器，当用例执行失败时，添加代码调用上面方法保留截图。
我是自定义了一个监听器，继承了TestListenerAdapter ，然后重写了onTestFailure
ScrennFailtureListener.java

package Tool;

import org.testng.ITestResult;
import org.testng.TestListenerAdapter;

import TestCase.FirefoxInitPre;

public class ScrennFailtureListener extends TestListenerAdapter {
    @Override
    public void onTestFailure(ITestResult tr) {
        System.out.println("执行onTestFailure");
        ScreenShot.takeScreenshot(FirefoxInitPre.driver);
        super.onTestFailure(tr);
    }
}

3、用例中添加监听器
我所有的用例都会继承基类TestCase.java，所以我在基类中添加了监听器

完整代码如下：

package TestCase;
import org.openqa.selenium.support.PageFactory;
import org.testng.Reporter;
import org.testng.annotations.AfterSuite;
import org.testng.annotations.BeforeSuite;
import org.testng.annotations.Listeners;
import org.uncommons.reportng.HTMLReporter;
import org.uncommons.reportng.JUnitXMLReporter;

import Pages.BaiduPage;
import Pages.LoginPage;
import Tool.ScrennFailtureListener;

@Listeners({ScrennFailtureListener.class,HTMLReporter.class,JUnitXMLReporter.class})
public class TestBase extends FirefoxInitPre{
    protected BaiduPage baiduPage;
    protected LoginPage loginPage;
    
    @BeforeSuite
    public void initPage(){
        Reporter.log("初始化Page页面开始");
        baiduPage=PageFactory.initElements(FirefoxInitPre.driver, BaiduPage.class);
        loginPage=PageFactory.initElements(FirefoxInitPre.driver, LoginPage.class);
        Reporter.log("初始化Page页面完成");
    }
    
    
    @AfterSuite
    public void quit(){
        Reporter.log("用例执行完毕，关闭浏览器");
        System.out.println("该方法为TestBase:quit");
        FirefoxInitPre.driver.quit();
    }
    
}

上面的弄好后，修改用例使其执行失败，然后运行用例。
查看目录snapshot是否会生成图片