Add case of element not found which throws for single element

- Note that for multiple element so ElementArray, there is no exception but an empty array

Author: dprevost-LMI

test/__mocks__/@wdio/globals.ts

@@ -74,12 +74,48 @@ export const elementFactory = (_selector: string, index?: number, parent: Webdri
     element.getElement = vi.fn().mockResolvedValue(element)
 
     // Note: an element found has element.elementId while a not found has element.error
-    // element.error = new Error(`Couldn't find element with selector "${_selector}"`)
-    // element.elementId = `elementId-${index ?? 0}-${_selector}`
+    element.elementId = `${_selector}${index ? '-' + index : ''}`
 
     return element
 }
 
+export const notFoundElementFactory = (_selector: string, index?: number, parent: WebdriverIO.Browser | WebdriverIO.Element = browser): WebdriverIO.Element => {
+    const partialElement = {
+        selector: _selector,
+        index,
+        $,
+        $$,
+        isExisting: vi.fn().mockResolvedValue(false),
+        parent
+    } satisfies Partial<WebdriverIO.Element>
+
+    const element = partialElement as unknown as WebdriverIO.Element
+
+    // Note: an element found has element.elementId while a not found has element.error
+    const elementId = `${_selector}${index ? '-' + index : ''}`
+    const error = (functionName: string) => new Error(`Can't call ${functionName} on element with selector ${elementId} because element wasn't found`)
+
+    // Mimic element not found by throwing error on any method call beisde isExisting
+    const notFoundElement = new Proxy(element, {
+        get(target, prop) {
+            if (prop in element) {
+                const value = element[prop as keyof WebdriverIO.Element]
+                return value
+            }
+            if (['then', 'catch', 'toStringTag'].includes(prop as string) || typeof prop === 'symbol') {
+                const value = Reflect.get(target, prop)
+                return typeof value === 'function' ? value.bind(target) : value
+            }
+            element.error = error(prop as string)
+            return () => { throw element.error }
+        }
+    })
+
+    element.getElement = vi.fn().mockResolvedValue(notFoundElement)
+
+    return notFoundElement
+}
+
 const $ = vi.fn((_selector: string) => {
     const element = elementFactory(_selector)
 

test/matchers/element/toBeDisplayed.test.ts

@@ -3,6 +3,7 @@ import { $, $$ } from '@wdio/globals'
 
 import { toBeDisplayed } from '../../../src/matchers/element/toBeDisplayed.js'
 import { executeCommandBe, waitUntil } from '../../../src/utils.js'
+import { notFoundElementFactory } from '../../__mocks__/@wdio/globals.js'
 
 vi.mock('@wdio/globals')
 
@@ -549,4 +550,16 @@ Expect ${selectorName} to be displayed
 Expected: "displayed"
 Received: "not displayed"`)
     })
+
+    describe('not found element', async () => {
+        let element: WebdriverIO.Element
+
+        beforeEach(async () => {
+            element = notFoundElementFactory('sel')
+        })
+
+        test('throws error when an element does not exists', async () => {
+            await expect(thisContext.toBeDisplayed(element)).rejects.toThrow("Can't call isDisplayed on element with selector sel because element wasn't found")
+        })
+    })
 })

test/util/elementsUtil.test.ts

@@ -1,8 +1,8 @@
 import { vi, test, describe, expect, beforeEach } from 'vitest'
 import { $, $$, } from '@wdio/globals'
 
-import { awaitElementOrArray, awaitElementArray, wrapExpectedWithArray, map, isStrictlyElementArray, isElement, isElementArrayLike, isElementOrArrayLike, isElementOrNotEmptyElementArray } from '../../src/util/elementsUtil.js'
-import { chainableElementArrayFactory, elementArrayFactory, elementFactory } from '../__mocks__/@wdio/globals.js'
+import { awaitElementOrArray, awaitElementArray, wrapExpectedWithArray, map, isStrictlyElementArray, isElement, isElementArrayLike, isElementOrArrayLike } from '../../src/util/elementsUtil.js'
+import { chainableElementArrayFactory, elementArrayFactory, elementFactory, notFoundElementFactory } from '../__mocks__/@wdio/globals.js'
 
 vi.mock('@wdio/globals')
 
@@ -383,6 +383,7 @@ describe('elementsUtil', () => {
             await $('element').getElement(),
             await $('element'),
             elementFactory('element'),
+            notFoundElementFactory('notFoundElement')
         ])('should return true for Element: %s', async (element) => {
             const isElementResult = isElement(element)
 
@@ -478,27 +479,4 @@ describe('elementsUtil', () => {
             expect(result).toBe(false)
         })
     })
-
-    describe(isElementOrNotEmptyElementArray, async () => {
-        test.for([
-            await $('element'),
-            await $$('elements'),
-            [elementFactory('element1'), elementFactory('element2')],
-        ])('should return true for Element or non-empty ElementArray/Element[]: %s', async (element) => {
-            const result = isElementOrNotEmptyElementArray(element)
-
-            expect(result).toBe(true)
-        })
-        test.for([
-            [],
-            $$('elements'),
-            $('element'),
-            '1',
-            {},
-        ])('should return false for non-Element or empty array: %s', async (element) => {
-            const result = isElementOrNotEmptyElementArray(element)
-
-            expect(result).toBe(false)
-        })
-    })
 })