fix(browser): locator.element() returns HTMLElement or SVGElement (#8440)

Author: sheremet-va

packages/browser/context.d.ts

@@ -544,15 +544,15 @@ export interface Locator extends LocatorSelectors {
    *
    * @see {@link https://vitest.dev/guide/browser/locators#element}
    */
-  element(): Element
+  element(): HTMLElement | SVGElement
   /**
    * Returns an array of elements matching the selector.
    *
    * If no elements match the selector, an empty array is returned.
    *
    * @see {@link https://vitest.dev/guide/browser/locators#elements}
    */
-  elements(): Element[]
+  elements(): (HTMLElement | SVGElement)[]
   /**
    * Returns an element matching the selector.
    *
@@ -561,7 +561,7 @@ export interface Locator extends LocatorSelectors {
    *
    * @see {@link https://vitest.dev/guide/browser/locators#query}
    */
-  query(): Element | null
+  query(): HTMLElement | SVGElement | null
   /**
    * Wraps an array of `.elements()` matching the selector in a new `Locator`.
    *

packages/browser/src/client/tester/locators/index.ts

@@ -204,22 +204,22 @@ export abstract class Locator {
     return this.locator(`internal:or=${JSON.stringify(locator._pwSelector || locator.selector)}`)
   }
 
-  public query(): Element | null {
+  public query(): HTMLElement | SVGElement | null {
     const parsedSelector = this._parsedSelector || (this._parsedSelector = selectorEngine.parseSelector(this._pwSelector || this.selector))
-    return selectorEngine.querySelector(parsedSelector, document.documentElement, true)
+    return selectorEngine.querySelector(parsedSelector, document.documentElement, true) as HTMLElement | SVGElement
   }
 
-  public element(): Element {
+  public element(): HTMLElement | SVGElement {
     const element = this.query()
     if (!element) {
       throw getElementError(this._pwSelector || this.selector, this._container || document.body)
     }
     return element
   }
 
-  public elements(): Element[] {
+  public elements(): (HTMLElement | SVGElement)[] {
     const parsedSelector = this._parsedSelector || (this._parsedSelector = selectorEngine.parseSelector(this._pwSelector || this.selector))
-    return selectorEngine.querySelectorAll(parsedSelector, document.documentElement)
+    return selectorEngine.querySelectorAll(parsedSelector, document.documentElement) as (HTMLElement | SVGElement)[]
   }
 
   public all(): Locator[] {