WebDriver进行屏幕截图

来源：互联网发布：c语言判断奇偶编辑：程序博客网时间：2024/05/16 03:53

例：打开百度首页，进行截图

package com.example.tests;import java.io.File;import org.apache.commons.io.FileUtils;import org.junit.*;import org.openqa.selenium.*;import org.openqa.selenium.ie.InternetExplorerDriver;public class Selenium2 {@Testpublic void testTakesScreenshot() {WebDriver driver = new InternetExplorerDriver();driver.get("http://www.baidu.com");try {File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);FileUtils.copyFile(srcFile,new File("d:\\screenshot.png"));} catch (Exception e) {e.printStackTrace();}   driver.close();}}

TakesScreenshot接口提供了getScreenshotAs()方法来捕捉屏幕。上面的例子中，我们指定了OutputType.FILE作为参数传递给getScreenshoAs()方法，告诉它将截取的屏幕以文件形式返回。

如果使用的是RemoteWebDriver() ，则方法应该如下

首先启动selenium java -jar selenium-server-standalone-2.25.0.jar

package com.example.tests;import java.io.File;import java.io.IOException;import java.net.MalformedURLException;import java.net.URL;import org.apache.commons.io.FileUtils;import org.junit.*;import org.openqa.selenium.*;import org.openqa.selenium.remote.*;public class Selenium2 {@Testpublic void testRemoteWebDriverScreenShot() {//指定使用的浏览器DesiredCapabilities capability = DesiredCapabilities.internetExplorer();WebDriver driver = null;try {driver = new RemoteWebDriver( //我使用localhost来测试new URL("http://localhost:4444/wd/hub"), capability);} catch (MalformedURLException e) {e.printStackTrace();}driver.get("http://www.sina.com.cn");//对远程系统进行截图driver = new Augmenter().augment(driver); File scrFile =    ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);try {FileUtils.copyFile(scrFile, new File("D:\\screenshot.png"));} catch (IOException e) {e.printStackTrace();}}}