fix(browser): update expect.element type to match the implementation (#8597)

Author: sheremet-va

packages/browser/matchers.d.ts

@@ -22,7 +22,7 @@ declare module 'vitest' {
      * You can set default timeout via `expect.poll.timeout` option in the config.
      * @see {@link https://vitest.dev/api/expect#poll}
      */
-    element: <T extends Element | Locator | null>(element: T, options?: ExpectPollOptions) => PromisifyDomAssertion<Awaited<Element | null>>
+    element: <T extends HTMLElement | SVGElement | null | Locator>(element: T, options?: ExpectPollOptions) => PromisifyDomAssertion<Awaited<HTMLElement | SVGElement | null>>
   }
 }
 

packages/browser/src/client/tester/expect-element.ts

@@ -1,13 +1,13 @@
 import type { Locator } from '@vitest/browser/context'
-import type { ExpectPollOptions } from 'vitest'
+import type { ExpectPollOptions, PromisifyDomAssertion } from 'vitest'
 import { chai, expect } from 'vitest'
 import { getType } from 'vitest/internal/browser'
 import { matchers } from './expect'
 import { processTimeoutOptions } from './utils'
 
 const kLocator = Symbol.for('$$vitest:locator')
 
-function element<T extends HTMLElement | SVGElement | null | Locator>(elementOrLocator: T, options?: ExpectPollOptions): unknown {
+function element<T extends HTMLElement | SVGElement | null | Locator>(elementOrLocator: T, options?: ExpectPollOptions): PromisifyDomAssertion<HTMLElement | SVGElement | null> {
   if (elementOrLocator != null && !(elementOrLocator instanceof HTMLElement) && !(elementOrLocator instanceof SVGElement) && !(kLocator in elementOrLocator)) {
     throw new Error(`Invalid element or locator: ${elementOrLocator}. Expected an instance of HTMLElement, SVGElement or Locator, received ${getType(elementOrLocator)}`)
   }
@@ -50,5 +50,4 @@ function element<T extends HTMLElement | SVGElement | null | Locator>(elementOrL
 }
 
 expect.extend(matchers)
-// Vitest typecheck doesn't pick up this assignment for some reason
-Object.assign(expect, { element })
+expect.element = element