selenium2入门 断言失败自动截图 （四）

一般web应用程序出错过后，会抛出异常。这个时候能截个图下来，当然是极好的。

selenium自带了截图功能。

            //获取截图file
            File scrFile= ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
            //将图片移动到指定位置
            FileUtils.moveFile(scrFile, new File(newFilePath));

当然在截图之前 需要判断新路径是否合法。下面贴出整体的代码。

package common;
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;
public class ScreenShot {
    public WebDriver driver;
    public ScreenShot(WebDriver _driver){
        driver=_driver;
    }
    private void takeScreenshot(String screenPath){
        try {
            //获取截图file
            File scrFile= ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
            //将图片移动到指定位置
            FileUtils.moveFile(scrFile, new File(screenPath));
        }  catch (IOException e) {
            e.printStackTrace();
        }
    }
    public void takeScreenshot(){
        SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd(hh_mm_ss)SSS");
        String fileName=dateFormat.format( new Date());
        //创建新的文件名
        String screenName=fileName+".jpg";
        //创建文件夹
        File dir = new File("test-output/snapshot");
        if (!dir.exists()){
            dir.mkdirs();
        }
        //获取新文件名的绝对路径
        String screenPath= dir.getAbsolutePath()+"/"+screenName;
        //截图
        this.takeScreenshot(screenPath);
    }
}

调用

ScreenShot ss = new ScreenShot(Driver);