Class WebdriverIOLocator

packages/browser-webdriverio/src/locators.ts:29–144

import { __INTERNAL } from 'vitest/internal/browser'

class WebdriverIOLocator extends Locator {
  constructor(protected _pwSelector: string, protected _container?: Element) {
    super()
  }

  // This exists to avoid calling `this.elements` in `this.selector`'s getter in interactive actions
  private withElement(element: Element, error: Error | undefined) {
    const pwSelector = selectorEngine.generateSelectorSimple(element)
    const cssSelector = convertElementToCssSelector(element)
    return new ElementWebdriverIOLocator(cssSelector, error, pwSelector, element)
  }

  override get selector(): string {
    const selectors = this.elements().map(element => convertElementToCssSelector(element))
    if (!selectors.length) {
      throw utils.getElementError(this._pwSelector, this._container || document.body)
    }
    let hasShadowRoot = false
    const newSelectors = selectors.map((selector) => {
      if (selector.startsWith('>>>')) {
        hasShadowRoot = true
        return selector.slice(3)
      }
      return selector
    })
    return (hasShadowRoot ? '>>>' : '') + newSelectors.join(', ')
  }

  public override click(options?: UserEventClickOptions): Promise<void> {
    return ensureAwaited(async (error) => {
      const element = await this.findElement(options)
      return this.withElement(element, error).click(processClickOptions(options))
    })
  }

  public override dblClick(options?: UserEventClickOptions): Promise<void> {
    return ensureAwaited(async (error) => {
      const element = await this.findElement(options)
      return this.withElement(element, error).dblClick(processClickOptions(options))
    })
  }

  public override tripleClick(options?: UserEventClickOptions): Promise<void> {
    return ensureAwaited(async (error) => {
      const element = await this.findElement(options)
      return this.withElement(element, error).tripleClick(processClickOptions(options))
    })
  }

  public selectOptions(
    value: HTMLElement | HTMLElement[] | Locator | Locator[] | string | string[],
    options?: UserEventSelectOptions,
  ): Promise<void> {
    return ensureAwaited(async (error) => {
      const element = await this.findElement(options)
      const values = getWebdriverioSelectOptions(element, value)
      return triggerCommandWithTrace<void>({
        name: '__vitest_selectOptions',