selenium webdriver学习（十二）------------如何利用selenium-webdriver截图

在自动化测试中常常会用到截图功能。最近用了一下selenium-webdriver的截图功能还算不错，可以截取页面全图，不管页面有多长。

下面的代码演示了如何使用webdriver进行截图：

 

1. import java.io.File; 
2. import java.io.IOException; 
3.  
4. import org.apache.commons.io.FileUtils; 
5. import org.openqa.selenium.OutputType; 
6. import org.openqa.selenium.TakesScreenshot; 
7. import org.openqa.selenium.WebDriver; 
8. import org.openqa.selenium.firefox.FirefoxDriver; 
9. public class ShotScreen { 
10.  
11.     /**
12.      * @author gongjf
13.      * @throws IOException
14.      * @throws InterruptedException
15.      */ 
16.     publicstaticvoid main(String[] args)throws IOException, InterruptedException { 
17.          
18.         System.setProperty("webdriver.firefox.bin","D:\\Program Files\\Mozilla Firefox\\firefox.exe");   
19.         WebDriver dr = new FirefoxDriver(); 
20.         dr.get("http://www.51.com"); 
21.                  
22.         //这里等待页面加载完成 
23.         Thread.sleep(5000); 
24.         //下面代码是得到截图并保存在D盘下 
25.         File screenShotFile = ((TakesScreenshot)dr).getScreenshotAs(OutputType.FILE); 
26.         FileUtils.copyFile(screenShotFile,new File("D:/test.png"));}} 

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;
public class ShotScreen {

	/**
	 * @author gongjf
	 * @throws IOException 
	 * @throws InterruptedException 
	 */
	public static void main(String[] args) throws IOException, InterruptedException {
		
		System.setProperty("webdriver.firefox.bin","D:\\Program Files\\Mozilla Firefox\\firefox.exe");  
		WebDriver dr = new FirefoxDriver();
		dr.get("http://www.51.com");
				
		//这里等待页面加载完成
		Thread.sleep(5000);
		//下面代码是得到截图并保存在D盘下
		File screenShotFile = ((TakesScreenshot)dr).getScreenshotAs(OutputType.FILE);
		FileUtils.copyFile(screenShotFile, new File("D:/test.png"));}}

看了一下OutputType接口和TakesScreenshot接口，吐槽一下，貌似这两个接口不是同一个开发写的或者注释没有更新怎么的。在OutputType里面的注释说：

 

1. /** 
2. * Defines the output type for a screenshot. See org.openqa.selenium.Screenshot for usage and 
3. * examples. 
4. ... 

/**
 * Defines the output type for a screenshot. See org.openqa.selenium.Screenshot for usage and
 * examples.
...

然后在那找了半天的org.openqa.selenium.Screenshot 接口，晕，后来想应该是org.openqa.selenium.TakesScreenshot。

在TakesScreenshot里有如下注释：

 

1. /**
2. * Capture the screenshot and store it in the specified location.
3. *
4. * <p>For WebDriver extending TakesScreenshot, this makes a best effort
5. * depending on the browser to return the following in order of preference:
6. * <ul>
7. *   <li>Entire page</li>
8. *   <li>Current window</li>
9. *   <li>Visible portion of the current frame</li>
10. *   <li>The screenshot of the entire display containing the browser</li>
11. * </ul>
12. *
13. * <p><span style="color: rgb(255, 0, 0);">For WebElement extending TakesScreenshot, this makes a best effort
14. * depending on the browser to return the following in order of preference:
15. *   - The entire content of the HTML element
16. *   - The visisble portion of the HTML element</span>
17. *
18. * @param <X> Return type for getScreenshotAs.
19. * @param target target type, @see OutputType
20. * @return Object in which is stored information about the screenshot.
21. * @throws WebDriverException on failure.
22. */ 

  /**
   * Capture the screenshot and store it in the specified location.
   *
   * <p>For WebDriver extending TakesScreenshot, this makes a best effort
   * depending on the browser to return the following in order of preference:
   * <ul>
   *   <li>Entire page</li>
   *   <li>Current window</li>
   *   <li>Visible portion of the current frame</li>
   *   <li>The screenshot of the entire display containing the browser</li>
   * </ul>
   *
   * <p>For WebElement extending TakesScreenshot, this makes a best effort
   * depending on the browser to return the following in order of preference:
   *   - The entire content of the HTML element
   *   - The visisble portion of the HTML element
   *
   * @param <X> Return type for getScreenshotAs.
   * @param target target type, @see OutputType
   * @return Object in which is stored information about the screenshot.
   * @throws WebDriverException on failure.
   */

试了一下截取Webelement最终发现WebElement接口没有实现这个类。搞了半天也只是会了截取页面的全图。截取当前的frame也截取的页面全图。难道这个功能没有完善，好吧，这样说自我安慰一下。

selenium-webdriver 面向接口编程，找一个需要的功能还真是挺难的。