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