Java+Selenium3方法篇31-利用Robot类截图【转载】

前面我们介绍了Selenium中TakeScreenshot类来截图，得到的图片是浏览器窗口内的截图。有时候，只截浏览器窗口内的图是不够的，而且TakeScreenshot截图只针对浏览器的web事件，假如你在运行脚本过程，windows上有一个其他软件弹出了一个购物的弹窗页面，这个时候就会干扰你截图。所以，有时候我们需要整个屏幕截图，这个时候我们就需要用到Robot这个类。原理大概是，从电脑屏幕左上角画一个长方形，一直画到屏幕右下角，然后得到File对象，在把这个File是用png还是jpg保存，复制到一个具体路径，这个就是截图的整个流程。下面我们用脚本来实现这个过程。

package lessons;

import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.concurrent.TimeUnit;

import javax.imageio.ImageIO;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;

public class TakeScreenshot {

public static void main(String[] args) throws Exception {

        System.setProperty("webdriver.chrome.driver", ".\\Tools\\chromedriver.exe");

        WebDriver driver = new ChromeDriver();

        driver.manage().window().maximize();

        driver.manage().timeouts().implicitlyWait(5, TimeUnit.SECONDS);

        driver.get("https://www.baidu.com");

        Thread.sleep(1000);

        // 调用截图方法
        BufferedImage image = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));

        try{

            ImageIO.write(image, "png", new File(".\\Screenshots\\screen_robot.png"));

        }catch(IOException e){

            System.out.println(e.getMessage());
        }

      }



}

      这里建议你试试，把这篇和前面一篇文章提到的截图方法，摘取出来，然后单独写一个类或者方法，在你的selenium脚本中去调用相关截图方法。我打算这部分放到框架设计之前讲，不过，还是建议你试试。