【转帖】WebDriver拾级而上(12) – 截图selenium-webdriver

悠悠小仙仙

好的测试人员都会截得一手好图，就跟骨灰级宅男定会吟得一手好诗一般。
截取页面全图，不管页面有多长。
Java代码:
CODE:

1. <font size="4">package com.test;
   
2. 
   
3. import java.io.File;
   
4. import java.io.IOException;
   
5. import java.text.SimpleDateFormat;
   
6. import java.util.Date;
   
7. import org.apache.commons.io.FileUtils;
   
8. import org.openqa.selenium.OutputType;
   
9. import org.openqa.selenium.TakesScreenshot;
   
10. import org.openqa.selenium.WebDriver;
    
11. import org.openqa.selenium.chrome.ChromeDriver;
    
12. 
    
13. public class Test_ShotScreen {
    
14.     public static void main(String[] args) throws IOException, InterruptedException{
    
15.         String url = "http://www.51.com";
    
16.         //打开chrome
    
17.         WebDriver dr = new ChromeDriver();
    
18.         dr.get(url);
    
19. 
    
20.         //这里等待页面加载完成
    
21.         Thread.sleep(5000);
    
22.          
    
23.         //下面代码是得到截图并保存在D盘下
    
24.         File screenShotFile = ((TakesScreenshot)dr).getScreenshotAs(OutputType.FILE);
    
25.         //FileUtils.copyFile(screenShotFile, new File("D:/test.png"));  
    
26.         FileUtils.copyFile(screenShotFile, new File("D:\\AutoScreenCapture\\" + getCurrentDateTime()+ ".jpg"));  
    
27. 
    
28.         dr.quit();
    
29.     }
    
30. 
    
31.     public static String getCurrentDateTime(){
    
32.         SimpleDateFormat df = new SimpleDateFormat("yyyyMMdd_HHmmss");//设置日期格式
    
33.         //System.out.println(df.format(new Date()));
    
34.         return df.format(new Date());
    
35.     }
    
36. }</font>

复制代码

方法:
CODE:

1. <font size="4">import java.io.File;  
   
2. import java.io.IOException;  
   
3. import org.apache.commons.io.FileUtils;  
   
4.       
   
5. public void takeScreenShot(String name){  
   
6.     File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);  
   
7.     try {  
   
8.         FileUtils.copyFile(scrFile, new File("c:\\Learning\\"+ name));  
   
9.     } catch (IOException e) {  
   
10.         e.printStackTrace();  
    
11.     }  
    
12. }</font>

复制代码

巴黎的灯光下

辛苦了，我的楼