//只需要传入文件名字即可，而这个名字大家可以直接使用测试的方法名

public void captureScreenshot(String fileName) {

String dirName = "screenshot";

if (!(new File(dirName).isDirectory())) {

new File(dirName).mkdir();

}

SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMdd-HHmmss");

String time = sdf.format(new Date());

TakesScreenshot tsDriver = (TakesScreenshot) driver;

File image = new File(dirName+File.separator+time+fileName+".png");

tsDriver.getScreenshotAs(OutputType.FILE).renameTo(image);

}