如何實現Selenium的UI自動化測試屏幕截圖功能

小編給大家分享一下如何實現Selenium的UI自動化測試屏幕截圖功能，相信大部分人都還不怎么了解，因此分享這篇文章給大家參考一下，希望大家閱讀完這篇文章后大有收獲，下面讓我們一起去了解一下吧！

UI自動化測試執行過程中，當遇到檢查失敗的情況，往往會發現打印的log并不能有效地幫助我們定位問題。我們需要失敗時刻的屏幕截圖來重現當時的失敗場景，進而排查出錯原因。

基于這種需求可以使用Selenium的屏幕截圖功能。

實現代碼如下：

import java.io.File;
import java.io.IOException;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.time.DateUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.interactions.Actions;

public static void takeScreeshot(String screenPath, WebDriver chrome){
   try {
     //指定了OutputType.FILE做為參數傳遞給getScreenshotAs()方法，其含義是將截取的屏幕以文件形式返回。
    File scrFile = ((TakesScreenshot) chrome)
       .getScreenshotAs(OutputType.FILE); // 關鍵代碼，執行屏幕截圖，默認會把截圖保存到temp目錄
    FileUtils.copyFile(scrFile, new File(screenPath)); //利用FileUtils工具類的copyFile()方法保存getScreenshotAs()返回的文件對象。 
    } catch (IOException e) {
     System.out.println("Screen shot error: " + screenPath);
     System.out.println("該錯誤可以查看截圖："+screenPath);
   } catch (Exception e) {
    // TODO: handle exception
   }
  }
  
   public static void takeScreenshot(WebDriver chrome,String imgName) {
    String screenName=imgName+DateUtils.MILLIS_PER_DAY+".jpg";
    String fileString= "D:\\selenium\\SchoolpalERP_QTP\\image";      
    if (!(new File(fileString).isDirectory())) { // 判斷是否存在該目錄
     new File(fileString).mkdir(); // 如果不存在則新建一個目錄
    }
    File dir = new File(fileString);
    if (!dir.exists())
     dir.mkdirs();
    String screenPath = dir.getAbsolutePath() + "\\" + screenName;
    takeScreeshot(screenPath, chrome);
   }

以上是“如何實現Selenium的UI自動化測試屏幕截圖功能”這篇文章的所有內容，感謝各位的閱讀！相信大家都有了一定的了解，希望分享的內容對大家有所幫助，如果還想學習更多知識，歡迎關注億速云行業資訊頻道！