Selenium v>=3.4.0 的整页屏幕截图？

Question

Selenium开发人员决定停止整页截图功能： https://github.com/SeleniumHQ/selenium/issues/3912

在应用程序测试脚本中模拟此功能所需的代码是什么？

我试图降级到 Selenium 3.3.1，但对驱动程序和/或浏览器版本有一些依赖，这不向前兼容。

Answer 1

试试这个。

import unittest
from selenium import webdriver
import datetime
class Tests(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome(desired_capabilities={'browserName': 'chrome', 'chromeOptions': {'args': ['--headless'], 'binary': '/Applications/Google Chrome Canary.app/Contents/MacOS/Google Chrome Canary'}})
self.driver.get('http://google.com/')
def tearDown(self):
self.driver.quit()
def test_screenshot(self):
width = self.driver.execute_script(
"return Math.max(document.body.scrollWidth, document.body.offsetWidth, document.documentElement.clientWidth, document.documentElement.scrollWidth, document.documentElement.offsetWidth);")
height = self.driver.execute_script(
"return Math.max(document.body.scrollHeight, document.body.offsetHeight, document.documentElement.clientHeight, document.documentElement.scrollHeight, document.documentElement.offsetHeight);")
self.driver.set_window_size(width+100, height+100)
self.driver.save_screenshot('screenshot ' + str(datetime.datetime.now().date()) + ' ' + str(datetime.datetime.now().time()) + '.png')
if __name__ == '__main__':
unittest.main()

Answer 2

在一个晦涩的开源库中找到的解决方案：

绒球.xml

<dependency>
<groupId>ru.yandex.qatools.ashot</groupId>
<artifactId>ashot</artifactId>
<version>1.5.4</version>
</dependency>

SomeJavaTest.java

private void screenshot(File out) throws IOException, InterruptedException, AWTException {
final Screenshot screenshot = new AShot()
.shootingStrategy(ShootingStrategies.viewportPasting(100))
.takeScreenshot(driver);
final BufferedImage image = screenshot.getImage();
ImageIO.write(image, "PNG", out);
}