Wdio Percy support v8

packages/wdio-browserstack-service/package.json

@@ -37,18 +37,23 @@
     "browserstack-local": "^1.5.1",
     "chalk": "^5.3.0",
     "csv-writer": "^1.6.0",
+    "follow-redirects": "^1.15.3",
     "formdata-node": "5.0.1",
     "git-repo-info": "^2.1.1",
     "gitconfiglocal": "^2.1.0",
-    "got": "^ 12.6.1",
+    "got": "^12.6.1",
     "uuid": "^9.0.0",
     "webdriverio": "8.22.0",
-    "winston-transport": "^4.5.0"
+    "winston-transport": "^4.5.0",
+    "yauzl": "^2.10.0",
+    "@percy/appium-app": "^2.0.1",
+    "@percy/selenium-webdriver": "^2.0.2"
   },
   "peerDependencies": {
     "@wdio/cli": "^5.0.0 || ^6.0.0 || ^7.0.0 || ^8.0.0"
   },
   "devDependencies": {
+    "@types/follow-redirects": "^1.14.4",
     "@types/node": "^20.1.0",
     "@wdio/globals": "8.22.0"
   },

packages/wdio-browserstack-service/src/@types/bstack-service-types.d.ts

@@ -1,11 +1,13 @@
 declare namespace WebdriverIO {
     interface Browser {
         getAccessibilityResultsSummary: () => Promise<{ [key: string]: any; }>,
-        getAccessibilityResults: () => Promise<Array<{ [key: string]: any; }>>
+        getAccessibilityResults: () => Promise<Array<{ [key: string]: any; }>>,
+        percyCaptureMap: any
     }
 
     interface MultiRemoteBrowser {
         getAccessibilityResultsSummary: () => Promise<{ [key: string]: any; }>,
-        getAccessibilityResults: () => Promise<Array<{ [key: string]: any; }>>
+        getAccessibilityResults: () => Promise<Array<{ [key: string]: any; }>>,
+        percyCaptureMap: any
     }
 }

packages/wdio-browserstack-service/src/Percy/Percy-Handler.ts

@@ -0,0 +1,88 @@
+import type { Capabilities } from '@wdio/types'
+import type { BeforeCommandArgs, AfterCommandArgs } from '@wdio/reporter'
+
+import {
+    o11yClassErrorHandler
+} from '../util.js'
+import PercyCaptureMap from './PercyCaptureMap.js'
+
+import * as PercySDK from './PercySDK.js'
+
+class _PercyHandler {
+    private _testMetadata: { [key: string]: any } = {}
+    private sessionName?: string
+    private _isAppAutomate?: boolean
+    public _percyScreenshotCounter: any = 0
+
+    constructor (
+        private _percyAutoCaptureMode: string | undefined,
+        private _browser: WebdriverIO.Browser | WebdriverIO.MultiRemoteBrowser,
+        private _capabilities: Capabilities.RemoteCapability,
+        isAppAutomate?: boolean,
+        private _framework?: string
+    ) {
+        this._isAppAutomate = isAppAutomate
+    }
+
+    _setSessionName(name: string) {
+        this.sessionName = name
+    }
+
+    async teardown () {
+        await new Promise<void>((resolve) => {
+            setInterval(() => {
+                if (this._percyScreenshotCounter === 0) {
+                    resolve()
+                }
+            }, 1000)
+        })
+    }
+
+    async percyAutoCapture(eventName: string | null) {
+        if (eventName) {
+            this._percyScreenshotCounter += 1
+            this._isAppAutomate ? await PercySDK.screenshotApp(this._browser, (this._browser.percyCaptureMap as PercyCaptureMap).getName((this.sessionName as string), eventName)) : await PercySDK.screenshot(this._browser, (this._browser.percyCaptureMap as PercyCaptureMap).getName((this.sessionName as string), eventName));
+            (this._browser.percyCaptureMap as PercyCaptureMap).increment((this.sessionName as string), eventName)
+            this._percyScreenshotCounter -= 1
+        }
+    }
+
+    async before () {
+        this._browser.percyCaptureMap = new PercyCaptureMap()
+    }
+
+    async browserCommand (args: BeforeCommandArgs & AfterCommandArgs) {
+        if (args.endpoint && this._percyAutoCaptureMode) {
+            let eventName = null
+            if ((args.endpoint as string).includes('click') && ['click', 'auto'].includes(this._percyAutoCaptureMode as string)) {
+                eventName = 'click'
+            } else if ((args.endpoint as string).includes('screenshot') && ['screenshot', 'auto'].includes(this._percyAutoCaptureMode as string)) {
+                eventName = 'screenshot'
+            } else if ((args.endpoint as string).includes('actions') && ['auto'].includes(this._percyAutoCaptureMode as string)) {
+                if (args.body && args.body.actions && Array.isArray(args.body.actions) && args.body.actions.length && args.body.actions[0].type === 'key') {
+                    eventName = 'keys'
+                }
+            }
+            await this.percyAutoCapture(eventName)
+        }
+    }
+
+    async afterTest () {
+        if (this._percyAutoCaptureMode && this._percyAutoCaptureMode === 'testcase') {
+            await this.percyAutoCapture('testcase')
+        }
+    }
+
+    async afterScenario () {
+        if (this._percyAutoCaptureMode && this._percyAutoCaptureMode === 'testcase') {
+            await this.percyAutoCapture('testcase')
+        }
+    }
+}
+
+// https://github.com/microsoft/TypeScript/issues/6543
+const PercyHandler: typeof _PercyHandler = o11yClassErrorHandler(_PercyHandler)
+type PercyHandler = _PercyHandler
+
+export default PercyHandler
+

packages/wdio-browserstack-service/src/Percy/Percy.ts

@@ -0,0 +1,178 @@
+import fs from 'node:fs'
+import path from 'node:path'
+import os from 'node:os'
+
+import { createRequire } from 'node:module'
+const require = createRequire(import.meta.url)
+const { spawn } = require('node:child_process')
+
+import { nodeRequest, getBrowserStackUser, getBrowserStackKey } from '../util.js'
+import { PercyLogger } from './PercyLogger.js'
+
+import PercyBinary from './PercyBinary.js'
+
+import type { BrowserstackConfig, UserConfig } from '../types.js'
+import type { Options } from '@wdio/types'
+
+const logDir = 'logs'
+
+class Percy {
+    #logfile: string = path.join(logDir, 'percy.log')
+    #address: string = process.env.PERCY_SERVER_ADDRESS || 'http://localhost:5338'
+
+    #binaryPath: string | any = null
+    #options: BrowserstackConfig & Options.Testrunner
+    #config: Options.Testrunner
+    #proc: any = null
+    #isApp: boolean = false
+    #projectName: string | undefined = undefined
+
+    isProcessRunning = false
+
+    constructor(options: BrowserstackConfig & Options.Testrunner, config: Options.Testrunner, bsConfig: UserConfig) {
+        this.#options = options
+        this.#config = config
+        if (options.app) {
+            this.#isApp = true
+        }
+        this.#projectName = bsConfig.projectName
+    }
+
+    async #getBinaryPath(): Promise<string> {
+        if (!this.#binaryPath) {
+            const pb = new PercyBinary()
+            this.#binaryPath = await pb.getBinaryPath(this.#config)
+        }
+        return this.#binaryPath
+    }
+
+    async #sleep(ms: number) {
+        return new Promise(resolve => setTimeout(resolve, ms))
+    }
+
+    async healthcheck() {
+        try {
+            const resp = await nodeRequest('GET', 'percy/healthcheck', null, this.#address)
+            if (resp) {
+                return true
+            }
+        } catch (err) {
+            return false
+        }
+    }
+
+    async start() {
+        const binaryPath: string = await this.#getBinaryPath()
+        const logStream = fs.createWriteStream(this.#logfile, { flags: 'a' })
+        const token = await this.fetchPercyToken()
+        const configPath = await this.createPercyConfig()
+
+        if (!token) {
+            return false
+        }
+
+        const commandArgs = [`${this.#isApp ? 'app:exec' : 'exec'}:start`]
+
+        if (configPath) {
+            commandArgs.push('-c', configPath as string)
+        }
+
+        this.#proc = spawn(
+            binaryPath,
+            commandArgs,
+            { env: Object.assign(process.env, { PERCY_TOKEN: token }) }
+        )
+
+        this.#proc.stdout.pipe(logStream)
+        this.#proc.stderr.pipe(logStream)
+        this.isProcessRunning = true
+        const that = this
+
+        /* eslint-disable @typescript-eslint/no-unused-vars */
+        this.#proc.on('close', function (code: any) {
+            that.isProcessRunning = false
+        })
+
+        do {
+            const healthcheck = await this.healthcheck()
+            if (healthcheck) {
+                PercyLogger.debug('Percy healthcheck successful')
+                return true
+            }
+
+            await this.#sleep(1000)
+        } while (this.isProcessRunning)
+
+        return false
+    }
+
+    async stop() {
+        const binaryPath = await this.#getBinaryPath()
+        return new Promise( (resolve) => {
+            const proc = spawn(binaryPath, ['exec:stop'])
+            proc.on('close', (code: any) => {
+                this.isProcessRunning = false
+                resolve(code)
+            })
+        })
+    }
+
+    isRunning() {
+        return this.isProcessRunning
+    }
+
+    async fetchPercyToken() {
+        const projectName = this.#projectName
+
+        try {
+            const type = this.#isApp ? 'app' : 'automate'
+            const response = await nodeRequest(
+                'GET',
+                `api/app_percy/get_project_token?name=${projectName}&type=${type}`,
+                {
+                    username: getBrowserStackUser(this.#config),
+                    password: getBrowserStackKey(this.#config)
+                },
+                'https://api.browserstack.com'
+            )
+            PercyLogger.debug('Percy fetch token success : ' + response.token)
+            return response.token
+        } catch (err) {
+            PercyLogger.error(`Percy unable to fetch project token: ${err}`)
+            return null
+        }
+    }
+
+    async createPercyConfig() {
+        if (!this.#options.percyOptions) {
+            return null
+        }
+
+        const configPath = path.join(os.tmpdir(), 'percy.json')
+        const percyOptions = this.#options.percyOptions
+
+        if (!percyOptions.version) {
+            percyOptions.version = '2'
+        }
+
+        return new Promise((resolve) => {
+            fs.writeFile(
+                configPath,
+                JSON.stringify(
+                    percyOptions
+                ),
+                (err: any) => {
+                    if (err) {
+                        PercyLogger.error(`Error creating percy config: ${err}`)
+                        resolve(null)
+                    }
+
+                    PercyLogger.debug('Percy config created at ' + configPath)
+                    resolve(configPath)
+                }
+            )
+        })
+    }
+}
+
+export default Percy