WebDriver高级应用实例（7）
7.1在测试中断言失败的步骤进行屏幕截图
目的：在测试过程中，在断言语句执行失败时，对当前的浏览器进行截屏，并在磁盘上新建一个yyyy-mm-dd格式的目录，并在断言失败时新建一个已hh-mm-ss格式的截图文件
被测网页的网址：
http://www.baidu.com
Java语言版本的API实例代码
需要新建一个Package，将FileUtil类、DateUtil类和TestFailCaptureScreen测试类3个文件都放到这个Package下
此例子借助了两个工具类来实现测试目的，此种方式将代码封装，便于提高代码的复用度，提高测试脚本编写的效率
DateUtil类:
import java.util.Date; //DataUtil类主要用于生成年、月、日、小时、分钟和秒的信息，用于生成保存截图文件目录名和文件名 public class DateUtil { /* * 格式化输出日期 * @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类:
import java.io.File;
import java.io.IOException;
public class FileUtil {
/*
* FileUtil类用于创目录和文件，此例子中只使用此类的创建目录方法，创建文件的方法仅供大家参考，将来用于根据测试需要创建指定的数据文件
* */
public static boolean createFile(String destFileName){
File file = new File(destFileName);
if(file.exists()){
System.out.println("创建单个文件"+destFileName+"失败，目标文件已存在！");
return false;
}
if(destFileName.endsWith(File.separator)){
System.out.println("创建单个文件"+destFileName+"失败，目标文件不能为目录");
return false;
}
//判断目标目录是否存在
if(!file.getParentFile().exists()){
System.out.println("目标文件所在目录不存在，准备创建它！");
if(!file.getParentFile().mkdirs()){
System.out.println("创建目标文件所在目录失败");
return false;
}
}
//创建目标文件
try {
if(file.createNewFile()){
System.out.println("创建单个文件"+destFileName+"成功");
return true;
}else{
System.out.println("创建单个文件"+destFileName+"失败");
return false;
}
} catch (IOException e) {
e.printStackTrace();
System.out.println("创建单个文件"+destFileName+"失败"+e.getMessage());
return false;
}
}
public static boolean createDir(String destDirName){
File dir = new File(destDirName);
if(dir.exists()){
System.out.println("创建目录"+destDirName+"失败目标目录已经存在");
return false;
}
//创建目录
if(dir.mkdirs()){
System.out.println("创建目录"+destDirName+"成功！");
return true;
}else{
System.out.println("创建目录"+destDirName+"失败!");
return false;
}
}
}
TestFailCaptureScreen测试类:
import org.testng.annotations.Test;
import org.testng.annotations.BeforeMethod;
import java.io.File;
import java.util.Date;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.Assert;
import org.testng.annotations.AfterMethod;
public class TestFailCaptureScreen {
WebDriver driver;
String url = "http://www.baidu.com";
@Test
public void testSearch() {
driver.findElement(By.id("kw")).sendKeys("seleium");
driver.findElement(By.id("su")).click();
try {
/*
* 断言页面的代码是否存在‘事在人为’这4个关键字，因页面中没有这4个字所以触发catch语句执行并触发截图操作
* */
Assert.assertTrue(driver.getPageSource().contains("事在人为"));
System.out.println("assert后续执行了");
} catch (AssertionError e) {
System.out.println("catch中的代码被执行了");
takeTakesScreenshot(driver);
}
}
/*
* 在测试类中声明截图的方法，截图方法调用了时间类和文件操作类的静态方法，用时间格式生成目录名称和截图文件名称
* */
public void takeTakesScreenshot(WebDriver driver) {
try {
//生成日期对象
Date date = new Date();
//调用DateUtil类中的方法，生成截图文件夹日期名称
String picDir = "e:\\"+String.valueOf(DateUtil.getYear(date))+"-"+String.valueOf(DateUtil.getMonth(date))+"-"+String.valueOf(DateUtil.getDay(date));
if(!new File(picDir).exists()){
FileUtil.createDir(picDir);
}
//调用DateUtil类中的方法，生成截图文件的时间名称
String filePath = picDir+"\\"+String.valueOf(DateUtil.getHour(new Date()))
+"-"+String.valueOf(DateUtil.getMinute(new Date()))
+"-"+String.valueOf(DateUtil.getSecond(new Date()))+".png";
//进行截图时将文件保存在srcFile对象中
File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
//将截图文件内容写到磁盘中生成截图文件
FileUtils.copyFile(srcFile,new File(filePath));
} catch (Exception e) {
e.printStackTrace();
}
}
@BeforeMethod
public void beforeMethod() {
System.setProperty("webdriver.chrome.driver", "D:\\WebDriver\\chromedriver_win32\\chromedriver.exe");
driver = new ChromeDriver();
driver.manage().window().maximize();
driver.get(url);
}
@AfterMethod
public void afterMethod() {
driver.quit();
}
}
