selenium 验证关键步骤截图功能

package com.property;

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

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


public  class Test {
public void screenShot() {
	/**
	 * 实现截图功能：按照日期时间命名方式
	 * */
    // driver is your WebDriver
	WebDriver driver = null;
	String dateString = getDateFormat();
    File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
    try {
		FileUtils.copyFile(screenshot, new File(".\\ScreenShots\\" + dateString + ".png"));
	} catch (IOException e) {
		// TODO Auto-generated catch block
		e.printStackTrace();
	}
}

private String getDateFormat() {
	Date date = new Date();
	SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddhhmmss");
	String dateString = sdf.format(date);
	return dateString;
}
}