fix(vitest): headless mode not working when using webdriverio (fix #3930)

packages/vitest/package.json

@@ -180,6 +180,7 @@
     "@types/micromatch": "^4.0.2",
     "@types/prompts": "^2.4.4",
     "@types/sinonjs__fake-timers": "^8.1.2",
+    "@wdio/types": "^8.16.3",
     "birpc": "0.2.12",
     "chai-subset": "^1.6.0",
     "cli-truncate": "^3.1.0",

packages/vitest/src/node/browser/webdriver.ts

@@ -1,4 +1,5 @@
 import type { Awaitable } from '@vitest/utils'
+import type { Capabilities } from '@wdio/types'
 import type { BrowserProvider, BrowserProviderOptions } from '../../types/browser'
 import { ensurePackageInstalled } from '../pkg'
 import type { WorkspaceProject } from '../workspace'
@@ -10,6 +11,14 @@ export interface WebdriverProviderOptions extends BrowserProviderOptions {
   browser: WebdriverBrowser
 }
 
+type BrowserOptionsKey = 'goog:chromeOptions' | 'moz:firefoxOptions' | 'ms:edgeOptions'
+interface WebdriverBrowserExtensionMapValue {
+  key: BrowserOptionsKey
+  value: {
+    args: string[]
+  }
+}
+
 export class WebdriverBrowserProvider implements BrowserProvider {
   public name = 'webdriverio'
 
@@ -42,6 +51,9 @@ export class WebdriverBrowserProvider implements BrowserProvider {
     const options = this.ctx.config.browser
 
     if (this.browser === 'safari') {
+      if (options.headless)
+        throw new Error('You\'ve enabled headless mode for Safari but it doesn\'t currently support it')
+
       const safaridriver = await import('safaridriver')
       safaridriver.start({ diagnose: true })
       this.stopSafari = () => safaridriver.stop()
@@ -53,13 +65,57 @@ export class WebdriverBrowserProvider implements BrowserProvider {
 
     const { remote } = await import('webdriverio')
 
+    const webdriverBrowserExtensionMap: {
+      chrome: WebdriverBrowserExtensionMapValue
+      firefox: WebdriverBrowserExtensionMapValue
+      edge: WebdriverBrowserExtensionMapValue
+    } = {
+      chrome: {
+        key: 'goog:chromeOptions',
+        value: {
+          args: [],
+        },
+      },
+      firefox: {
+        key: 'moz:firefoxOptions',
+        value: {
+          args: [],
+        },
+      },
+      edge: {
+        key: 'ms:edgeOptions',
+        value: {
+          args: [],
+        },
+      },
+    }
+
+    const capabilities: Capabilities.VendorExtensions & {
+      browserName: WebdriverProviderOptions['browser']
+    } = {
+      browserName: this.browser,
+    }
+
+    if (this.browser !== 'safari') {
+      if (options.headless) {
+        switch (this.browser) {
+          case 'chrome':
+            webdriverBrowserExtensionMap[this.browser].value.args = webdriverBrowserExtensionMap[this.browser].value.args.concat(['headless', 'disable-gpu'])
+            break
+          case 'firefox':
+            webdriverBrowserExtensionMap[this.browser].value.args.push('-headless')
+            break
+          case 'edge':
+            webdriverBrowserExtensionMap[this.browser].value.args.push('--headless')
+        }
+      }
+      capabilities[webdriverBrowserExtensionMap[this.browser].key] = webdriverBrowserExtensionMap[this.browser].value
+    }
+
     // TODO: close everything, if browser is closed from the outside
     this.cachedBrowser = await remote({
       logLevel: 'error',
-      capabilities: {
-        'browserName': this.browser,
-        'wdio:devtoolsOptions': { headless: options.headless },
-      },
+      capabilities,
     })
 
     return this.cachedBrowser

test/browser/specs/runner.test.mjs

@@ -24,8 +24,8 @@ const passedTests = getPassed(browserResultJson.testResults)
 const failedTests = getFailed(browserResultJson.testResults)
 
 await test('tests are actually running', async () => {
-  assert.ok(browserResultJson.testResults.length === 8, 'Not all the tests have been run')
-  assert.ok(passedTests.length === 7, 'Some tests failed')
+  assert.ok(browserResultJson.testResults.length === 9, 'Not all the tests have been run')
+  assert.ok(passedTests.length === 8, 'Some tests failed')
   assert.ok(failedTests.length === 1, 'Some tests have passed but should fail')
 
   assert.doesNotMatch(stderr, /Unhandled Error/, 'doesn\'t have any unhandled errors')

test/browser/test/headless.test.ts

@@ -0,0 +1,5 @@
+import { expect, it } from 'vitest'
+
+it('is running in headless mode', () => {
+  expect(navigator.userAgent.search(/headless/i)).not.toBe(-1)
+})