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