fix: screenshots for sessions

Author: kobenguyent

lib/helper/WebDriver.js

@@ -1788,14 +1788,14 @@ class WebDriver extends Helper {
 
         if (browser) {
           this.debug(`Screenshot of ${sessionName} session has been saved to ${outputFile}`)
-          return browser.saveScreenshot(outputFile)
+          await browser.saveScreenshot(outputFile)
         }
       }
     }
 
     if (!fullPage) {
       this.debug(`Screenshot has been saved to ${outputFile}`)
-      return this.browser.saveScreenshot(outputFile)
+      await this.browser.saveScreenshot(outputFile)
     }
 
     const originalWindowSize = await this.browser.getWindowSize()

test/acceptance/session_test.js

@@ -77,7 +77,7 @@ Scenario('Different cookies for different sessions @Playwright @Puppeteer', asyn
   I.expectNotEqual(cookies.john, cookies.mary)
 })
 
-Scenario('should save screenshot for sessions @Puppeteer @Playwright', async function ({ I }) {
+Scenario('should save screenshot for sessions @Puppeteer @Playwright @WebDriver', async function ({ I }) {
   await I.amOnPage('/form/bug1467')
   await I.saveScreenshot('original.png')
   await I.amOnPage('/')