目的：
在测试过程中，断言失败的时候，对当前浏览器显示的内容进行截屏操作，把截图文件保存到磁盘。
步骤：新建名为 cn.ErrorScreenshot的package，并在此包下新建DataUtil类、FileUtil类、Screenshot类和TestCase类。
DataUtil类的代码如下：
package cn.ErrorScreenshot; import java.util.Date; //DataUtil类主要用于生成年、月、日、时、分、秒的信息，用于生成保存截图文件目录名和文件名 public class DataUtil{ //格式化输入日期，@return 返回字符型日期 public static String format(java.util.Date date,String format){ String result=""; try{ if(date != null){ java.text.DateFormat df=new java.text.SimpleDateFormat(format); result=df.format(date); } }catch(Exception e){ e.printStackTrace(); } return result; } //返回年份 @return返回年份 public static int getYear(java.util.Date date){ java.util.Calendar c=java.util.Calendar.getInstance(); c.setTime(date); return c.get(java.util.Calendar.YEAR); } //返回月份 @return返回月份 public static int getMonth(java.util.Date date){ java.util.Calendar c= java.util.Calendar.getInstance(); c.setTime(date); return c.get(java.util.Calendar.MONTH)+1; } //返回在月份中的第几天 @return返回月份中的第几天 public static int getDay(java.util.Date date){ java.util.Calendar c= java.util.Calendar.getInstance(); c.setTime(date); return c.get(java.util.Calendar.DAY_OF_MONTH); } /* * 返回小时 * @param date * 日期 * @return返回小时 */ public static int getHour(java.util.Date date){ java.util.Calendar c=java.util.Calendar.getInstance(); c.setTime(date); return c.get(java.util.Calendar.HOUR_OF_DAY); } /* * 返回分钟 * @param date * 日期 * @return返回分钟 */ public static int getMinute(java.util.Date date){ java.util.Calendar c=java.util.Calendar.getInstance(); c.setTime(date); return c.get(java.util.Calendar.MINUTE); } /* * 返回秒 * @param date * 日期 * @return返回秒 */ public static int getSecond(java.util.Date date){ java.util.Calendar c=java.util.Calendar.getInstance(); c.setTime(date); return c.get(java.util.Calendar.SECOND); } }
FileUtil类代码如下：
本类调用Log4j方法，log4j相关代码可参考：java-selenium 使用log4j在测试过程中打印执行日志 如不需要去掉本类中相关代码即可
package cn.ErrorScreenshot; import java.io.File; import java.io.IOException; //自己工程目录下的log4j代码 import cn.Log4j.Log; //FileUtil 类用于创建目录和文件，此例子只是用此类的创建目录的方法 public class FileUtil{ final static Log log=Log.getlogger(FileUtil.class); public static boolean createFile(String destFileName){ File file=new File(destFileName); if(file.exists()){ Log.info("创建单个文件"+destFileName+"失败，目标文件已存在！"); return false; } if(destFileName.endsWith(file.separator)){ Log.info("创建单个文件"+destFileName+"失败，目标文件不能为目录！"); return false; } //判断目标文件所在目录是否存在 if(!file.getParentFile().exists()){ //如果目标文件所在的目录不存在，则创建父目录 Log.info("目录文件所在目录不存在，准备创建它！"); if(!file.getParentFile().mkdirs()){ Log.info("创建目标文件所在目录失败！"); return false; } } //创建目标问价 try{ if(file.createNewFile()){ Log.info("创建单个文件"+destFileName+"成功！"); return true; }else{ Log.info("创建单个文件"+destFileName+"失败！"); return false; } }catch(IOException e){ e.printStackTrace(); Log.info("创建单个文件"+destFileName+"失败！"+e.getMessage()); return false; } } public static boolean createDir(String destDirName){ File dir=new File(destDirName); if(dir.exists()){ Log.info("创建目录"+destDirName+"失败，目标目录已存在"); return false; } //创建目录 if(dir.mkdirs()){ Log.info("创建目录"+destDirName+"成功！"); return true; }else{ Log.info("创建目录"+destDirName+"失败！"); return false; } } }
Screenshot类代码如下：
package cn.ErrorScreenshot; import java.io.File; 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; import cn.ErrorScreenshot.DataUtil; import cn.ErrorScreenshot.FileUtil; public class Screenshot{ /* *截图方法，调用了时间类和文件操作类的静态方法并传入一个String类型的CaseName参数 *用来以yyyy-MM-dd的格式生成目录，HH时mm分ss秒加CaseName的格式命名文件名。如果命 *名不需要CaseName去掉代码中的两处CaseName即可。 */ public static void DN(WebDriver driver,String CaseName) { try{ //生成日期对象 Date date=new Date(); //调用DateUtil类中的方法，生成截图所在的文件夹日期类型 String picDir="C:\"+"errorScreenshot"+"\"+String.valueOf(DataUtil.getYear(date))+"-" +String.valueOf(DataUtil.getMonth(date))+"-"+String.valueOf(DataUtil.getDay(date)); if(!new File(picDir).exists()){ FileUtil.createDir(picDir); } //调用DataUtil类中的方法，生成截图文件的时间名称 String filePath=picDir+"//"+String.valueOf(DataUtil.getHour(new Date()))+"时"+ String.valueOf(DataUtil.getMinute(new Date()))+"分"+ String.valueOf(DataUtil.getSecond(new Date()))+"秒"+CaseName+".png"; //进行截图，并将文件内容保存在srcFile对象中 File srcFile=((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); // File scrFile = new File(""); // scrFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE); //将截图文件内容写入磁盘中，生成截图文件 FileUtils.copyFile(srcFile, new File(filePath)); }catch(Exception e){ e.printStackTrace(); } } }
TestCase类代码如下：
package cn.ErrorScreenshot; import org.openqa.selenium.By; import org.openqa.selenium.WebDriver; import org.openqa.selenium.chrome.ChromeDriver; import org.openqa.selenium.support.ui.ExpectedConditions; import org.openqa.selenium.support.ui.WebDriverWait; import org.testng.Assert; import org.testng.annotations.AfterMethod; import org.testng.annotations.BeforeMethod; import org.testng.annotations.Test; public class TestCase { WebDriver driver; @BeforeMethod public void Bfore(){ System.setProperty("webdriver.chrome.driver","E://chromedriver.exe"); driver=new ChromeDriver(); driver.get("http://www.baidu.com"); } @Test public void test(){ driver.findElement(By.id("kw")).sendKeys("java"); driver.findElement(By.id("su")).click(); //创建一个WebDriverWait对象，设置等待时长5秒 WebDriverWait w=new WebDriverWait(driver,5); //等待百度搜索结果页面中class属性值为 nums_text的元素 w.until(ExpectedConditions.presenceOfAllElementsLocatedBy(By.className("nums_text"))); try{ //断言页面源码包含 selenium Assert.assertTrue(driver.getPageSource().contains("selenium")); }catch(AssertionError e){ //断言失败则调用截图方法进行截图 //传入用例名字 搜索java，在截图文件命名中的时间后面加上用例名 Screenshot.DN(driver,"搜索java"); } } @AfterMethod public void After(){ driver.quit(); } }
运行TestCase测试类结束后，电脑C盘会生成errorScreenshot文件夹，errorScreenshot文件里面还有一个当前时间的文件夹及截图，如下图所示：
