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();
}
}
<dependency>
<groupId>ru.yandex.qatools.ashot</groupId>
<artifactId>ashot</artifactId>
<version>1.5.2</version>
</dependency>