WebDriver进行屏幕截图

转自 http://blog.csdn.net/iceryan/article/details/8566628

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

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 {
    @Test
    public 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 {
    @Test
    public 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();
        }