selenium之如何截图，留下程序不听话的证据！

一.概述

      知道吗，selenium在执行测试用例的时候，肯定会有执行失败的用例，那么在用例执行失败的那一刻，你能

怎么办呢，就像跟警察破案一样，要得到当时的监控画面，selenium也可以直接截图留下用例执行失败的有效依据，

那么如何截图呢，请看下文分解，你一定会有小小的收获。

二.编写一个截图类ScreenShot

package com.daanhealth.tnb.util;

import java.io.File;

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){
        this.driver=driver;
    }
    
    private void takeScreenshot(String screenPath){
        try {
            File srcFile=((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
            FileUtils.copyFile(srcFile, new File(screenPath));
        } catch (Exception e) {
            // TODO: handle exception
            System.out.println("Screen shot error: " + screenPath);
        }
    }
    
    public void takeScreenshot(){
        String screenName=new DateFormat().formatDateToString(System.currentTimeMillis())+".jpg";
        File dir=new File("test-output/snapshot");
        if(!dir.exists()){
            dir.mkdirs();
        }
        String screenPath=dir.getAbsolutePath()+File.separator+screenName;
        takeScreenshot(screenPath);
    }

}