package lessons;

import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.File;  

import javax.imageio.ImageIO;

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

/**
 * @author 北京-宏哥
 * 
 * @公众号:北京宏哥
 * 
 * @《手把手教你》系列技巧篇（六十）-java+ selenium自动化测试 - 截图三剑客 -中篇（详细教程）
 *
 * @2022年01月09日
 */
public class RobotScreenShot {
    
    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.get("https://www.sohu.com/");
        robotSnapshot();
        Thread.sleep(2000);
        System.out.println("browser will be close");
        driver.quit();
         
    }
     
    /**
     * 截屏方法二、Robot实现截屏
     * @throws Exception
     */
    public static void robotSnapshot() throws Exception {
        //调用截图方法
        BufferedImage img = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
        ImageIO.write(img, "png", new File("RobotScreenshots","robot_screen01.png"));
    }

}