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脚本中去调用相关截图方法。我打算这部分放到框架设计之前讲，不过，还是建议你试试。