Selenium Webdriver——实现截图功能

截图方法

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.搜索，截图

具体代码如下：

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.WebElement;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

public class TestSaveScreen {
    
    private WebDriver driver;
        @BeforeClass
          public void beforeClass() {
            driver = new FirefoxDriver();
            driver.get("https://www.baidu.com/");
            snapshot((TakesScreenshot)driver,"open_baidu.png");
          }
        @Test
        public void test() {
            WebElement searchInput= driver.findElement(By.id("kw"));
            searchInput.sendKeys("selenium");29             WebElement searchButton= driver.findElement(By.id("su"));
            searchButton.click();
             snapshot((TakesScreenshot)driver,"search_output.png");
          }

        @AfterClass
        public void afterClass() {
            driver.quit();
        }
        
        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");
                }
          }
}

 截图效果如下：