webdriver 截图的方法实现工具类

package util;

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;
/**
 * 该类用于实现截图功能
 * sceenshot工具方法
 *
 */
public class Sceenshot {
	//定义截图的文件名，去当前时间
	 public static SimpleDateFormat formatter=new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss");	 
	 public static  String name=formatter.format(new Date());
	 //定义图片保存的路径
	 public static  String Dir="D:\\p_w_picpaths"; 
	 
       public static void sceenshot(WebDriver dr) {  
    	   //判定目录是否存在，不存在则创建
    	   File dirpath=new File(Dir);
    	   if (! dirpath.isDirectory()){
    		   dirpath.mkdirs();
    	   }
  
    	//将截图以文件的形式返回
   		File srcFile=((TakesScreenshot)dr).getScreenshotAs(OutputType.FILE);
   		//创建截图的位置和文件名
   		File filepath=new File(Dir,name+".png");
   		try {
   			//使用copyFile()方法保存截图到指定位置
   			FileUtils.copyFile(srcFile, filepath);
   		} catch (IOException e) {
   			// TODO Auto-generated catch block
   			System.out.println("sceenshot p_w_picpaths fail!!!!");
   		}
   	}
}