关于Selenium v​​＆gt; = 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

在一个不起眼的开源库中找到的解决方案：

<强>的pom.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);

}