谷歌浏览器自动化测试--截图

        我想很多朋友都遇见过对一个网页截图的需求，我来说下我上次遇见的需求；我上次的需求是做一个爬虫，爬取一些网站的敏感信息，然后要把这个网页敏感信息的证据保存下来，我们这里会保存两种，第一种就是网页内容，第二种就是我们现在说的截图，把这个网页保存为一张图片。

开始

下面实现一个简单的保存网页为图片的代码：

public static void main(String[] args) throws Exception {
    System.setProperty("webdriver.chrome.driver",
            "C:\\Users\\Administrator\\AppData\\Local\\Google\\Chrome\\Application\\chromedriver.exe");
      
    WebDriver driver = new ChromeDriver();
  
    driver.manage().window().maximize();
     
    driver.get("http://www.baidu.com/");
  
    //找到百度上面的输入框、放入输入内容‘鹿晗人妖’
    driver.findElement(By.id("kw")).sendKeys("鹿晗人妖");
    //点击百度旁边的搜索按钮
    driver.findElement(By.id("su")).click();
    //暂停两秒，让他加载搜索出来的数据
    Thread.sleep(2000);
  
    //对整个网页截图
    File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); 
     
    //把截图保存到桌面
    FileUtils.copyFile(srcFile, new File("C:\\Users\\Administrator\\Desktop\\1233.png")); 
    driver.quit();
}

有些朋友不想截全屏，就只想要截这个页面的一小块，代码如下：

public static void main(String[] args) throws Exception {
    System.setProperty("webdriver.chrome.driver",
            "C:\\Users\\Administrator\\AppData\\Local\\Google\\Chrome\\Application\\chromedriver.exe");
     
    WebDriver driver = new ChromeDriver();
 
    driver.manage().window().maximize();
 
    driver.get("http://tool.oschina.net/highlight");
    Thread.sleep(2000);
 
    //找到class为wrapper的节点
    WebElement webElement = driver.findElement(By.className("wrapper"));
    Point point = webElement.getLocation();
    int eleWidth = webElement.getSize().getWidth();
    int eleHeight = webElement.getSize().getHeight();
     
    //对整个网页截图
    File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); 
     
    //在上面的网页截图中，把根据class找到的节点截取出来、并覆盖上面的网页截图
    BufferedImage  fullImg = ImageIO.read(srcFile);
    BufferedImage eleScreenshot= fullImg.getSubimage(point.getX(), point.getY(),
            eleWidth, eleHeight);
    ImageIO.write(eleScreenshot, "png", srcFile);
 
    //把根据class找到的节点截图保存到桌面
    FileUtils.copyFile(srcFile, new File("C:\\Users\\Administrator\\Desktop\\1233.png")); 
    driver.quit();
}

还有的朋友要说，比如我爬取了百度新闻网站，我发现百度新闻网站上面有敏感信息，比如有'强奸',我这边需要把'强奸'两个字标记下来，在进行截图，这样别人一看截图，就知道这里有敏感信息，代码如下：

public static void main(String[] args) throws Exception {
    System.setProperty("webdriver.chrome.driver",
            "C:\\Users\\Administrator\\AppData\\Local\\Google\\Chrome\\Application\\chromedriver.exe");
      
    WebDriver driver = new ChromeDriver();
  
    driver.manage().window().maximize();
     
    driver.get("http://news.baidu.com");
     
    //获取百度新闻中html
    String htmlContent = driver.getPageSource();
  
    //解析html字符串(引入了jsoup-1.8.1.jar)
    Document document = Jsoup.parse(htmlContent);
     
    //删除html下面标签中的onclick属性、href属性(我这里只是截图、点击事件对我没用)
    for (Element element : document.getAllElements()) {
        element.removeAttr("onclick").removeAttr("href");
    }
     
    //删除html下面所有的script标签(我这里只是截图、不需要动态页面)
    for (Element element : document.getElementsByTag("script")) {
        element.remove();
    }
     
    //替换html中的双引号为单引号、删除换行
    String reHtmlContent = document.body().html().replace("\"", "'").replaceAll("\r|\n", "");;
     
    //标记'网页'为敏感字、用红色框给他框住
    reHtmlContent = reHtmlContent.replace("网页", "<span style='border:2px solid red;'>网页</span>");
     
    reHtmlContent = "\"" + reHtmlContent + "\"";
     
    //通过js把转换完的html替换到页面的body上面
    JavascriptExecutor js = (JavascriptExecutor) driver;
    js.executeScript("document.body.innerHTML=" + reHtmlContent);
     
    //对整个网页截图
    File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); 
      
    //把截图保存到桌面
    FileUtils.copyFile(srcFile, new File("C:\\Users\\Administrator\\Desktop\\1233.png")); 
    driver.quit();
}

试了上面方法的朋友都会发现，如果你截图的网站比较长，是截图不全的，这种情况我们可以采取下拉滚动条截图，然后再拼接，不过这种方案遇见有些瀑布流布局的网站也是有点问题的；我记得火狐浏览器是没有这个问题的，火狐浏览器对页面截图，会直接截整个网页。

import java.io.File;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import com.xjxcc.util.ImageUtils;
 
public class Test1 {
    /**
     * @param args
     * @throws Exception
     */
    public static void main(String[] args) throws Exception {
        System.setProperty("webdriver.chrome.driver",
                "C:\\Users\\Administrator\\AppData\\Local\\Google\\Chrome\\Application\\chromedriver.exe");
           
        WebDriver driver = new ChromeDriver();
       
        driver.manage().window().maximize();
        driver.get("https://zhidao.baidu.com");
       
        /* 通过js获取浏览器的各种高度 */
        JavascriptExecutor js = (JavascriptExecutor) driver;
        String heightStrs = (String) js.executeScript("return document.body.scrollHeight.toString()+','+document.body.scrollTop.toString() + ',' + window.screen.height.toString()");
        String[] heights = heightStrs.split(",");
        int htmlHeight = Integer.parseInt(heights[0]);//整个页面的高度
        int scrollTop = Integer.parseInt(heights[1]);//滚动条现在所处的高度
        int screenHeight = Integer.parseInt(heights[2]);//电脑屏幕的高度
        screenHeight = screenHeight - 140;
         
        //开始滚动截图
        int count = 0;
        while(scrollTop < htmlHeight){
            scrollTop += screenHeight;
            System.out.println("document.body.scrollTop = " + screenHeight * count);
            ((JavascriptExecutor) driver).executeScript("window.scrollTo(0, "+ (screenHeight * count) +")");
             
            //对整个网页截图
            File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); 
               
            //把截图保存到桌面
            FileUtils.copyFile(srcFile, new File("C:\\Users\\Administrator\\Desktop\\allImg\\"+ (++count) +".png")); 
        }
         
        //拼接图片
        File imgsFile = new File("C:\\Users\\Administrator\\Desktop\\allImg");
        if(!imgsFile.isDirectory()){
            throw new RuntimeException("地址不是一个正确的目录...");
        }
        File[] imgsFiles = imgsFile.listFiles();
        ImageUtils.mergeImg(imgsFiles, ImageUtils.IMG_TYPE_PNG, ImageUtils.MERGE_IMG_TYPE_Y, "C:\\Users\\Administrator\\Desktop\\111.png");
         
        driver.quit();
    }
}

结束