WebDriver進行屏幕截圖

例： 打開百度首頁 ，進行截圖

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