WebDriver进行屏幕截图

jopen 11年前

 例: 打开百度首页 ,进行截图

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