在测试中对断言失败的步骤进行截图并保存！

1.目的 ：在测试当中，在断言语句执行失败的时候，对当前浏览器显示的内容进行截图操作，并且在磁盘上新建一个名为当天日期（yyyy-mm-dd）的目录，并在目录中新建一个名称以断言执行失败时发生时间表示的hh-mm-dd格式的截图文件；

2.步骤：创建DateUtil类 来获取 年 月 日 时 分 秒  ；

3.创建FileUtil 类 进行创建目录 保存 截图文件 ；

4.代码如下：

package scr.comm;
import java.util.*;
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;
    }
//返回年份 ；
    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);        
    }
//返回月份 ；
    public static int getMonth(java.util.Date date){
        java.util.Calendar calendar =java.util.Calendar.getInstance();
        calendar.setTime(date);
    return     calendar.get(java.util.Calendar.MONTH)+1;
    }
//返回月份中的第几天；
    public  static int getDay(java.util.Date date){
    java.util.Calendar calendar =java.util.Calendar.getInstance();
    calendar.setTime(date);
    return calendar.get(java.util.Calendar.DAY_OF_MONTH);
    }
    //返回小时；
    public  static int getHour(Date date){
    Calendar calendar=Calendar.getInstance();
    calendar.setTime(date);
    return calendar.get(Calendar.HOUR);
    }
//返回分钟 ；
    public static int getMinute(Date date ){
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(date);
        return calendar.get(Calendar.MINUTE);
    }
//返回秒    
    public static int getSecond(Date date ){
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(date);
        return calendar.get(Calendar.SECOND);
    }
}

 

 

package scr.comm;

import java.io.File;
import java.io.IOException;

/*
 * FileUtil 用于创建目录和文件，此例子中指使用了该类创建目录的方法，创建文件仅供大家参考；
 */
public class 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()) {
            // r如果目标文件不存在创建父目录 ；
            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("c创建 " + destFileName + " 失败！");
                return false;
            }

        } catch (IOException e) {
            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;
        }

    }

}

 

测试类：

package testNGPractice;

import java.io.File;

import org.apache.commons.io.FileUtils;
import org.junit.Assert;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import scr.comm.DateUtil;
import scr.comm.FileUtil;
import scr.comm.Log;
import scr.comm.OpenBrowserInfo;

public class TestFailCaptureScreen {
 public  WebDriver driver  ;
    @Test
  public void f() {
Log.startTextCase("@Test 开始执行！");
        String url ="http://www.sogou.com/";
    driver.navigate().to(url);
    driver.findElement(By.id("query")).sendKeys("光荣之路自动化测试！");
    driver.findElement(By.id("stb")).click();
    try{
        Assert.assertTrue(driver.getPageSource().contains("事在人为"));
        System.out.println("assert 执行了！");
    }catch(AssertionError e){
        System.out.println("catch  中的代码 被执行了！");
        takeTakesScreenShot(driver);
        
    }
Log.endTestCase("@Test 执行结束 ！");    
        
  }
  @BeforeMethod
  public void beforeMethod() {
      OpenBrowserInfo.Log4jInitialize();
      OpenBrowserInfo.SystemSetProperty();
      driver =new FirefoxDriver() ;  
  }

  @AfterMethod
  public void afterMethod() {
      driver.quit();
  }
  public void takeTakesScreenShot(WebDriver driver){
      try{
          //生成日期
          java.util.Date date =new java.util.Date();
          String picDir="D:\\"+
                  String.valueOf(DateUtil.getYear(date)+"-"+
                  String.valueOf(DateUtil.getMonth(date))+"-"+
                  String.valueOf(DateUtil.getDay(date))/*+" "+
                  String.valueOf(DateUtil.getHour(date))+"_"+
                  String.valueOf(DateUtil.getMinute(date))+"_"+
                  String.valueOf(DateUtil.getSecond(date))*/
                  );
         //创建目录；
          if(!new File(picDir).exists()){
              FileUtil.createDir(picDir);
          }
          //调用DateUtil类生成 截图文件的名称；
          String filePath=picDir + "\\" +
                  String.valueOf(DateUtil.getHour(date))+"_"+
                  String.valueOf(DateUtil.getMinute(date))+"_"+
                  String.valueOf(DateUtil.getSecond(date))+".png" ;
          //进行截图，并将文件内容保存到srcFile对象当中
          File srcFile= ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
          //将截图文件内容写入到磁盘当中去；
          FileUtils.copyFile(srcFile,new File(filePath));
      }catch(Exception e ){
          e.printStackTrace();
          
          
      }
      
      
  }

}

 

转载于:https://www.cnblogs.com/linbo3168/p/6125889.html