Selenium自动截图的方法（getScreenshotAs）

天行健，君子以自强不息；地势坤，君子以厚德载物。

---

每个人都有惰性，但不断学习是好好生活的根本，共勉！

---

文章均为学习整理笔记，分享记录为主，如有错误请指正，共同学习进步。

---

执行程序前请先配置驱动：
关于Java selenium使用前浏览器驱动的下载和环境变量的配置
关于Selenium自动化测试工具的Java实现详情请参考文章：
如何查看页面对应的Selenium定位参数
Java实现 selenium Web自动化测试（简单篇）
Java实现 selenium Web自动化测试（详细篇）

---

1. Selenium配置

在执行代码前先配置连接参数，这里配置后并打开百度首页进行内容搜索

        //指定驱动，第一个参数为驱动名称，不同浏览器的参数名称不一样，请根据浏览器查询到对应的浏览器参数名，第二个参数为驱动文件路径，即驱动完整文件路径
        System.setProperty("webdriver.chrome.driver", "D:\\JavaSoftWares\\Google\\driver\\chromedriver-win64\\chromedriver.exe");
        // 谷歌驱动
        ChromeOptions cops = new ChromeOptions();
        // 允许所有请求
        cops.addArguments("--remote-allow-origins=*");
        //默认设置开始打开网页时窗口最大化
        cops.addArguments("--start-maximized");
        //创建驱动对象
        WebDriver webDriver = new ChromeDriver(cops);
        // 启动需要打开的网页
        webDriver.get("https://www.baidu.com");
        //指定窗口大小--最大化
//        webDriver.manage().window().maximize();

        //定位百度输入框并填充收缩内容
        webDriver.findElement(By.id("kw")).sendKeys("李白");
        //搜索
        webDriver.findElement(By.id("kw")).submit();
        //此时会提交搜索，等待一会搜索内容出现后再开始截图
        Thread.sleep(5000);

2. 元素截图

截图使用getScreenshotAs(OutputType.FILE)方法

        //存储截图文件的文件夹位置
        String ssp = "E:\\PROJECTS\\MYSELF\\libai-multifunctional-platform\\src\\main\\java\\com\\libai\\test\\selenium\\chrome\\screenshot\\";

        /*
            元素截图部分---------------------------------------------
         */
        //文件名
        String eleLocalFileName = "ssp_ele_001.png";
        //文件完整路径
        String eleLocalFileFullName = ssp+eleLocalFileName;

        //本地文件转为文件类型
        File eleLocalFile = new File(eleLocalFileFullName);

        //定位元素
        WebElement ele = webDriver.findElement(By.id("kw"));
        //将元素对应的部分截图并转为文件类型
        File eleScreenShotFile = ele.getScreenshotAs(OutputType.FILE);
        //将截图文件复制到本地文件
        FileUtils.copyFile(eleScreenShotFile, eleLocalFile);

        Thread.sleep(5000);

3. 网页截图

        /*
            网页截图部分---------------------------------------------
         */
        //文件名
        String localFileName = "ssp_001.png";
        //文件完整路径
        String localFileFullName = ssp+localFileName;

        //本地文件转为文件类型
        File localFile = new File(localFileFullName);

        //开始截图，并将截图转为文件类型
        File fullScreenShootFile = ((TakesScreenshot)webDriver).getScreenshotAs(OutputType.FILE);

        //将截图文件复制到本地文件
        FileUtils.copyFile(fullScreenShootFile, localFile);

        Thread.sleep(5000);

4. 完整代码

ScreenShotTest.java

package com.libai.test.selenium.chrome;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.*;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;

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

/**
 * @ClassDescription: 截图操作
 * @JdkVersion: 1.8
 * @Author: 李白
 * @Created: 2024/5/8 20:15
 */
public class ScreenShotTest {
    public static void main(String[] args) throws IOException, InterruptedException {
        //指定驱动，第一个参数为驱动名称，不同浏览器的参数名称不一样，请根据浏览器查询到对应的浏览器参数名，第二个参数为驱动文件路径，即驱动完整文件路径
        System.setProperty("webdriver.chrome.driver", "D:\\JavaSoftWares\\Google\\driver\\chromedriver-win64\\chromedriver.exe");
        // 谷歌驱动
        ChromeOptions cops = new ChromeOptions();
        // 允许所有请求
        cops.addArguments("--remote-allow-origins=*");
        //默认设置开始打开网页时窗口最大化
        cops.addArguments("--start-maximized");
        //创建驱动对象
        WebDriver webDriver = new ChromeDriver(cops);
        // 启动需要打开的网页
        webDriver.get("https://www.baidu.com");
        //指定窗口大小--最大化
//        webDriver.manage().window().maximize();

        //定位百度输入框并填充收缩内容
        webDriver.findElement(By.id("kw")).sendKeys("李白");
        //搜索
        webDriver.findElement(By.id("kw")).submit();
        //此时会提交搜索，等待一会搜索内容出现后再开始截图
        Thread.sleep(5000);


        //存储截图文件的文件夹位置
        String ssp = "E:\\PROJECTS\\MYSELF\\libai-multifunctional-platform\\src\\main\\java\\com\\libai\\test\\selenium\\chrome\\screenshot\\";

        /*
            元素截图部分---------------------------------------------
         */
        //文件名
        String eleLocalFileName = "ssp_ele_001.png";
        //文件完整路径
        String eleLocalFileFullName = ssp+eleLocalFileName;

        //本地文件转为文件类型
        File eleLocalFile = new File(eleLocalFileFullName);

        //定位元素
        WebElement ele = webDriver.findElement(By.id("kw"));
        //将元素对应的部分截图并转为文件类型
        File eleScreenShotFile = ele.getScreenshotAs(OutputType.FILE);
        //将截图文件复制到本地文件
        FileUtils.copyFile(eleScreenShotFile, eleLocalFile);

        Thread.sleep(5000);

        /*
            网页截图部分---------------------------------------------
         */
        //文件名
        String localFileName = "ssp_001.png";
        //文件完整路径
        String localFileFullName = ssp+localFileName;

        //本地文件转为文件类型
        File localFile = new File(localFileFullName);

        //开始截图，并将截图转为文件类型
        File fullScreenShootFile = ((TakesScreenshot)webDriver).getScreenshotAs(OutputType.FILE);

        //将截图文件复制到本地文件
        FileUtils.copyFile(fullScreenShootFile, localFile);

        Thread.sleep(5000);


        //关闭浏览器
        webDriver.quit();
    }
}

5. 截图文件查看

文件存放位置可以看到文件已生成

截图内容查看：
元素截图

网页截图

---

感谢阅读，祝君暴富！