import org.junit.Test;import org.openqa.selenium.WebDriver;import org.openqa.selenium.chrome.ChromeDriver;import org.openqa.selenium.chrome.ChromeOptions;import ru.yandex.qatools.ashot.AShot;import ru.yandex.qatools.ashot.shooting.ShootingStrategies;import javax.imageio.ImageIO;import java.awt.image.BufferedImage;import java.io.File;import java.io.IOException;public class DemoApplicationTests { @Test public void contextLoads() throws InterruptedException, IOException { System.setProperty("webdriver.chrome.driver", "D:\\chromedriver.exe"); ChromeOptions chromeOptions = new ChromeOptions(); //设置为 headless 模式 (必须) //chromeOptions.addArguments("--headless"); final WebDriver driver = new ChromeDriver(chromeOptions); driver.manage().window().maximize(); driver.get("https://autohome.com.cn/"); // Let the user actually see something! Thread.sleep(5000); BufferedImage bi = new AShot() .shootingStrategy(ShootingStrategies.viewportPasting(100)) .takeScreenshot(driver).getImage(); ImageIO.write(bi, "png", new File("C:/test.png")); driver.quit(); }}
ru.yandex.qatools.ashot ashot 1.5.2