Selenium的UI自动化测试屏幕截图功能实例代码

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

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);

   }