Selenium Webdriver实现截图功能的示例

2021-05-03 11:20Monkey大圣 Java教程

今天小编就为大家分享一篇Selenium Webdriver实现截图功能的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

前几天在研究中自动化的时候突发奇想，想着能不能来截个图，以便之后查看，实现的方法其实也不难，毕竟selenium webdriver已经提供了截图额功能，TakesScreenshot接口函数（英文意思就是获取屏幕截图takes-screenshot）。

废话不多说了，直接上代码

									package com.wch;

									import java.io.File;

									import java.io.IOException;

									import org.junit.After;

									import org.junit.Before;

									import org.junit.Test;

									import org.openqa.selenium.OutputType;

									import org.openqa.selenium.TakesScreenshot;

									import org.openqa.selenium.WebDriver;

									import org.openqa.selenium.firefox.FirefoxDriver;

									import org.openqa.selenium.support.ui.WebDriverWait;

									import com.sun.jna.platform.FileUtils;

									public class TestTakesScreenshot {

									    public static void main(String[] args) {

									        System.setProperty("webdriver.firefox.bin", "D:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");

									        WebDriver driver = new FirefoxDriver();

									        driver.get("http://www.baidu.com");

									        File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); //讲截取的图片以文件的形式返回

									        try {

									            org.apache.commons.io.FileUtils.copyFile(srcFile, new File("d:\\screenshot.png")); //使用copyFile()方法保存获取到的截图文件

									        } catch (IOException e) {

									            // TODO Auto-generated catch block

									            e.printStackTrace();

									        }

									        driver.quit();

									    }

									}