selenium web driver 实现截图功能

在验证某些关键步骤时，需要截个图来记录一下当时的情况

Webdriver截图时，需要引入

import java.io.File;
import java.io.IOException;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;

截图方法

public static void snapshot(TakesScreenshot drivername, String filename)
  {
      // this method will take screen shot ,require two parameters ,one is driver name, another is file name
      
    String currentPath = System.getProperty("user.dir"); //get current work folder
    System.out.println(currentPath);
    File scrFile = drivername.getScreenshotAs(OutputType.FILE);
        // Now you can do whatever you need to do with it, for example copy somewhere
        try {
            System.out.println("save snapshot path is:"+currentPath+"/"+filename);
            FileUtils.copyFile(scrFile, new File(currentPath+"\"+filename));
        } catch (IOException e) {
            // TODO Auto-generated catch block
            System.out.println("Can't save screenshot");
            e.printStackTrace();
        } 
        finally
        {
           
            System.out.println("screen shot finished");
        }
  }

以下任务：

1.使用selenium打开百度，截图；

2.输入selenium关键字，截图；

3.搜索 并打开 selenium的百度百科，截图；

具体代码如下：

package baidu;

 

import java.io.File;
import java.io.IOException;

 

import org.apache.commons.io.FileUtils;
 
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
//import org.openqa.selenium.WebDriver.Navigation;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
 

 

public class selenium  {
    
    

  public static void snapshot(TakesScreenshot drivername, String filename)
  {
      // this method will take screen shot ,require two parameters ,one is driver name, another is file name
      
      
    File scrFile = drivername.getScreenshotAs(OutputType.FILE);
        // Now you can do whatever you need to do with it, for example copy somewhere
        try {
            System.out.println("save snapshot path is:E:/"+filename);
            FileUtils.copyFile(scrFile, new File("E:\"+filename));
        } catch (IOException e) {
            // TODO Auto-generated catch block
            System.out.println("Can't save screenshot");
            e.printStackTrace();
        } 
        finally
        {
            System.out.println("screen shot finished");
        }
  }

    public static void main (String [] args) throws InterruptedException
    {
         
     
        String URL="http://www.baidu.com";
        System.setProperty("webdriver.chrome.driver", "E:\chromedriver.exe"); 
        WebDriver driver = new ChromeDriver();
        driver.get(URL);
        //max size the browser
        driver.manage().window().maximize(); 
 /*
        Navigation navigation = driver.navigate();
         navigation.to(URL);*/
         Thread.sleep(2000);
         snapshot((TakesScreenshot)driver,"open_baidu.png");
         //WebElement reg=driver.findElement(By.name("tj_reg"));
         //reg.click();
         //    WebElement keyWord = driver.findElement(By.id("kw1"));
         
         //find the element
         WebElement keyWord = driver.findElement(By.xpath("//input[@id='kw1']"));
        keyWord.clear();
        //send key words 
         keyWord.sendKeys("Selenium");
         Thread.sleep(3000);
          snapshot((TakesScreenshot)driver,"input_keyWord.png");
         
 
         
          WebElement submit = driver.findElement(By.id("su1"));
          
          System.out.println(submit.getLocation());
          submit.click();
          //System.out.println(driver.getWindowHandle());
         Thread.sleep(5000);
          
        // System.out.println(driver.getPageSource());
        
        String pageSource=driver.getPageSource();
       //  System.out.println(pageSource);
        //WebElement link =driver.findElement(By.xpath(SELENIUM_LINK));
        WebElement link =driver.findElement(By.xpath("//*[@id="1"]/h3/a"));     //*[@id="1"]/h3/a
        link.click();
        Thread.sleep(5000); 
        driver.switchTo().window(driver.getWindowHandles().toArray(new String[0])[1]);
     
        //get page title
        System.out.println(driver.getTitle());
         Thread.sleep(5000);
    //     navigation.back();
          snapshot((TakesScreenshot)driver,"open_bake.png");
         System.out.println(driver.getTitle()+"
"+driver.getCurrentUrl());
         
     
         
          driver.quit();
           
       
    }
    
}

在百度搜索结果中，拿到你想要的elements，可以使用浏览器的查看元素，通过xpath方法获取

运行此代码后截图效果如下：

console输出：

Starting ChromeDriver (v2.9.248315) on port 33834
save snapshot path is:E:/open_baidu.png
screen shot finished
save snapshot path is:E:/input_keyWord.png
screen shot finished
(858, 179)
Selenium_百度百科
save snapshot path is:E:/open_bake.png
screen shot finished
Selenium_百度百科
http://baike.baidu.com/subview/478050/6464537.htm?fr=aladdin