使用Selenium WebDriver截取屏幕截图

Question

有谁知道是否可以使用Selenium WebDriver截取屏幕截图？(注:不是Selenium RC)

Answer 1

Java的

对的,这是可能的.以下示例使用Java:

WebDriver driver = new FirefoxDriver();
driver.get("http://www.google.com/");
File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
// Now you can do whatever you need to do with it, for example copy somewhere
FileUtils.copyFile(scrFile, new File("c:\\tmp\\screenshot.png"));

Answer 2

蟒蛇

每个WebDriver都有一个.save_screenshot(filename)方法.所以对于Firefox,它可以像这样使用:

from selenium import webdriver

browser = webdriver.Firefox()
browser.get('http://www.google.com/')
browser.save_screenshot('screenie.png')

令人困惑的是,.get_screenshot_as_file(filename)也存在同样的方法.

还有以下方法:( .get_screenshot_as_base64()用于嵌入html)和.get_screenshot_as_png()(用于检索二进制数据).

并注意WebElements有一个.screenshot()类似的方法,但只捕获所选元素.

Answer 3

C#

public void TakeScreenshot()
{
    try
    {            
        Screenshot ss = ((ITakesScreenshot)driver).GetScreenshot();
        ss.SaveAsFile(@"D:\Screenshots\SeleniumTestingScreenshot.jpg", System.Drawing.Imaging.ImageFormat.Jpeg);
    }
    catch (Exception e)
    {
        Console.WriteLine(e.Message);
        throw;
    }
}

Answer 4

JavaScript(Selenium-Webdriver)

driver.takeScreenshot().then(function(data){
   var base64Data = data.replace(/^data:image\/png;base64,/,"")
   fs.writeFile("out.png", base64Data, 'base64', function(err) {
        if(err) console.log(err);
   });
});

Answer 5

红宝石

require 'rubygems'
require 'selenium-webdriver'

driver = Selenium::WebDriver.for :ie 
driver.get "https://www.google.com"   
driver.save_screenshot("./screen.png")

更多文件类型和选项可用,您可以在takes_screenshot.rb中看到它们

Answer 6

Java的

我解决了这个问题.您可以扩充RemoteWebDriver以为其代理驱动程序实现的所有接口提供:

WebDriver augmentedDriver = new Augmenter().augment(driver); 
((TakesScreenshot)augmentedDriver).getScreenshotAs(...); //works this way

Answer 7

PHP(PHPUnit)

使用PHPUnit_Selenium扩展版本1.2.7:

class MyTestClass extends PHPUnit_Extensions_Selenium2TestCase {
    ...
    public function screenshot($filepath) {
        $filedata = $this->currentScreenshot();
        file_put_contents($filepath, $filedata);
    }

    public function testSomething() {          
        $this->screenshot('/path/to/screenshot.png');
    }
    ...
}

Answer 8

C#

public Bitmap TakeScreenshot(By by) {
    // 1. Make screenshot of all screen
    var screenshotDriver = _selenium as ITakesScreenshot;
    Screenshot screenshot = screenshotDriver.GetScreenshot();
    var bmpScreen = new Bitmap(new MemoryStream(screenshot.AsByteArray));

    // 2. Get screenshot of specific element
    IWebElement element = FindElement(by);
    var cropArea = new Rectangle(element.Location, element.Size);
    return bmpScreen.Clone(cropArea, bmpScreen.PixelFormat);
}

Answer 9

Java的

public String captureScreen() {
    String path;
    try {
        WebDriver augmentedDriver = new Augmenter().augment(driver);
        File source = ((TakesScreenshot)augmentedDriver).getScreenshotAs(OutputType.FILE);
        path = "./target/screenshots/" + source.getName();
        FileUtils.copyFile(source, new File(path)); 
    }
    catch(IOException e) {
        path = "Failed to capture screenshot: " + e.getMessage();
    }
    return path;
}

Answer 10

Jython的

import org.openqa.selenium.OutputType as OutputType
import org.apache.commons.io.FileUtils as FileUtils
import java.io.File as File
import org.openqa.selenium.firefox.FirefoxDriver as FirefoxDriver

self.driver = FirefoxDriver()
tempfile = self.driver.getScreenshotAs(OutputType.FILE)
FileUtils.copyFile(tempfile, File("C:\\screenshot.png"))

Answer 11

Java(机器人框架)

我用这种方法进行屏幕截图.

void takeScreenShotMethod(){
    try{
        Thread.sleep(10000)
        BufferedImage image = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
        ImageIO.write(image, "jpg", new File("./target/surefire-reports/screenshot.jpg"));
    }
    catch(Exception e){
        e.printStackTrace();
    }
}

您可以在需要的地方使用此方法.

Answer 12

Java的

这里似乎缺少 - 在Java中截取特定元素的屏幕截图:

public void takeScreenshotElement(WebElement element) throws IOException {
    WrapsDriver wrapsDriver = (WrapsDriver) element;
    File screenshot = ((TakesScreenshot) wrapsDriver.getWrappedDriver()).getScreenshotAs(OutputType.FILE);
    Rectangle rectangle = new Rectangle(element.getSize().width, element.getSize().height);
    Point location = element.getLocation();
    BufferedImage bufferedImage = ImageIO.read(screenshot);
    BufferedImage destImage = bufferedImage.getSubimage(location.x, location.y, rectangle.width, rectangle.height);
    ImageIO.write(destImage, "png", screenshot);
    File file = new File("//path//to");
    FileUtils.copyFile(screenshot, file);
}

Answer 13

C#

using System;
using OpenQA.Selenium.PhantomJS;
using System.Drawing.Imaging;

namespace example.com
{
    class Program
    {
        public static PhantomJSDriver driver;

        public static void Main(string[] args)
        {
            driver = new PhantomJSDriver();
            driver.Manage().Window.Size = new System.Drawing.Size(1280, 1024);
            driver.Navigate().GoToUrl("http://www.example.com/");
            driver.GetScreenshot().SaveAsFile("screenshot.png", ImageFormat.Png);
            driver.Quit();
        }
    }
}

需要NuGetPackages:

1. PhantomJS 2.0.0
2. Selenium.Support 2.48.2
3. Selenium.WebDriver 2.48.2

使用.NETFramework v4.5.2进行测试

Answer 14

Java的

我无法得到已接受的工作答案,但根据当前的WebDriver文档,在OS X 10.9上使用Java 7时,以下工作正常:

import java.io.File;
import java.net.URL;

import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.Augmenter;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

public class Testing {

   public void myTest() throws Exception {
       WebDriver driver = new RemoteWebDriver(
               new URL("http://localhost:4444/wd/hub"),
               DesiredCapabilities.firefox());

       driver.get("http://www.google.com");

       // RemoteWebDriver does not implement the TakesScreenshot class
       // if the driver does have the Capabilities to take a screenshot
       // then Augmenter will add the TakesScreenshot methods to the instance
       WebDriver augmentedDriver = new Augmenter().augment(driver);
       File screenshot = ((TakesScreenshot)augmentedDriver).
               getScreenshotAs(OutputType.FILE);
   }
}

Answer 15

电源外壳

Set-Location PATH:\to\selenium

Add-Type -Path "Selenium.WebDriverBackedSelenium.dll"
Add-Type -Path "ThoughtWorks.Selenium.Core.dll"
Add-Type -Path "WebDriver.dll"
Add-Type -Path "WebDriver.Support.dll"

$driver = New-Object OpenQA.Selenium.PhantomJS.PhantomJSDriver

$driver.Navigate().GoToUrl("https://www.google.co.uk/")

# Take a screenshot and save it to filename
$filename = Join-Path (Get-Location).Path "01_GoogleLandingPage.png"
$screenshot = $driver.GetScreenshot()
$screenshot.SaveAsFile($filename, [System.Drawing.Imaging.ImageFormat]::Png)

其他司机...

$driver = New-Object OpenQA.Selenium.Chrome.ChromeDriver
$driver = New-Object OpenQA.Selenium.Firefox.FirefoxDriver
$driver = New-Object OpenQA.Selenium.IE.InternetExplorerDriver
$driver = New-Object OpenQA.Selenium.Opera.OperaDriver

Answer 16

有多种方法可以通过Selenium的Java和Python客户端使用Selenium WebDriver进行屏幕截图。

Java 方法

以下是不同的Java截图方法：

• getScreenshotAs()从TakesScreenshot界面使用：

• 代码块：

       package screenShot;
  
       import java.io.File;
       import java.io.IOException;
  
       import org.apache.commons.io.FileUtils;
       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.ExpectedConditions;
       import org.openqa.selenium.support.ui.WebDriverWait;
  
       public class Firefox_takesScreenshot {
  
           public static void main(String[] args) throws IOException {
  
               System.setProperty("webdriver.gecko.driver", "C:\\Utility\\BrowserDrivers\\geckodriver.exe");
               WebDriver driver =  new FirefoxDriver();
               driver.get("https://login.bws.birst.com/login.html/");
               new WebDriverWait(driver, 20).until(ExpectedConditions.titleContains("Birst"));
               File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
               FileUtils.copyFile(scrFile, new File(".\\Screenshots\\Mads_Cruz_screenshot.png"));
               driver.quit();
           }
       }
  

• 截屏：

  

• 如果网页启用了jQuery，您可以使用pazone/ashot库中的ashot：

• 代码块：

       package screenShot;
  
       import java.io.File;
       import javax.imageio.ImageIO;
       import org.openqa.selenium.WebDriver;
       import org.openqa.selenium.firefox.FirefoxDriver;
       import org.openqa.selenium.support.ui.ExpectedConditions;
       import org.openqa.selenium.support.ui.WebDriverWait;
  
       import ru.yandex.qatools.ashot.AShot;
       import ru.yandex.qatools.ashot.Screenshot;
       import ru.yandex.qatools.ashot.shooting.ShootingStrategies;
  
       public class ashot_CompletePage_Firefox {
  
           public static void main(String[] args) throws Exception {
  
               System.setProperty("webdriver.gecko.driver", "C:\\Utility\\BrowserDrivers\\geckodriver.exe");
               WebDriver driver =  new FirefoxDriver();
               driver.get("https://jquery.com/");
               new WebDriverWait(driver, 20).until(ExpectedConditions.titleContains("jQuery"));
               Screenshot myScreenshot = new AShot().shootingStrategy(ShootingStrategies.viewportPasting(100)).takeScreenshot(driver);
               ImageIO.write(myScreenshot.getImage(),"PNG",new File("./Screenshots/firefoxScreenshot.png"));
               driver.quit();
           }
       }
  

• 截屏：

  

• 使用来自assertthat/selenium-shutterbug库的selenium-shutterbug：

• 代码块：

       package screenShot;
  
       import org.openqa.selenium.WebDriver;
       import org.openqa.selenium.firefox.FirefoxDriver;
       import com.assertthat.selenium_shutterbug.core.Shutterbug;
       import com.assertthat.selenium_shutterbug.utils.web.ScrollStrategy;
  
       public class selenium_shutterbug_fullpage_firefox {
  
           public static void main(String[] args) {
  
               System.setProperty("webdriver.gecko.driver", "C:\\Utility\\BrowserDrivers\\geckodriver.exe");
               WebDriver driver =  new FirefoxDriver();
               driver.get("https://www.google.co.in");
               Shutterbug.shootPage(driver, ScrollStrategy.BOTH_DIRECTIONS).save("./Screenshots/");
               driver.quit();
           }
       }
  

• 截屏：

  

Python 方法

以下是不同的Python截屏方法：

• 使用save_screenshot()方法：

• 代码块：

       from selenium import webdriver
  
       driver = webdriver.Chrome(r'C:\Utility\BrowserDrivers\chromedriver.exe')
       driver.get("http://google.com")
       driver.save_screenshot('./Screenshots/save_screenshot_method.png')
       driver.quit()
  

• 截屏：

  

• 使用get_screenshot_as_file()方法：

• 代码块：

       from selenium import webdriver
  
       driver = webdriver.Chrome(r'C:\Utility\BrowserDrivers\chromedriver.exe')
       driver.get("http://google.com")
       driver.get_screenshot_as_file('./Screenshots/get_screenshot_as_file_method.png')
       driver.quit()
  

• 截屏：

  

• 使用get_screenshot_as_png()方法：

• 代码块：

       from selenium import webdriver
  
       driver = webdriver.Chrome(r'C:\Utility\BrowserDrivers\chromedriver.exe')
       driver.get("http://google.com")
       screenPnG = driver.get_screenshot_as_png()
  
       # Crop it back to the window size (it may be taller)
       box = (0, 0, 1366, 728)
       im = Image.open(BytesIO(screenPnG))
       region = im.crop(box)
       region.save('./Screenshots/get_screenshot_as_png_method.png', 'PNG', optimize=True, quality=95)
       driver.quit()
  

• 截屏：