diff --git a/lib/helper/WebDriver.js b/lib/helper/WebDriver.js index 081570fa5..4472b27ac 100644 --- a/lib/helper/WebDriver.js +++ b/lib/helper/WebDriver.js @@ -1286,6 +1286,8 @@ class WebDriver extends Helper { } /* eslint-disable prefer-arrow-callback, comma-dangle, prefer-const */ + const originalWindowSize = await this.browser.getWindowSize(); + let { width, height } = await this.browser.execute(function () { return { height: document.body.scrollHeight, @@ -1298,7 +1300,9 @@ class WebDriver extends Helper { await this.browser.setWindowSize(width, height); this.debug(`Screenshot has been saved to ${outputFile}, size: ${width}x${height}`); - return this.browser.saveScreenshot(outputFile); + const buffer = await this.browser.saveScreenshot(outputFile); + await this.browser.setWindowSize(originalWindowSize.width, originalWindowSize.height); + return buffer; }