From 03c62b230ce1dee86cfbb3d5927811e0e3450464 Mon Sep 17 00:00:00 2001
From: Nandor Kraszlan <nandor.kraszlan@gmail.com>
Date: Fri, 18 Feb 2022 20:42:47 +0000
Subject: [PATCH] fix: added missing find overrides

---
 src/domWrapper.ts | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/domWrapper.ts b/src/domWrapper.ts
index cdb062cd1..9744342b6 100644
--- a/src/domWrapper.ts
+++ b/src/domWrapper.ts
@@ -21,7 +21,15 @@ export class DOMWrapper<NodeType extends Node> extends BaseWrapper<NodeType> {
     return this.element.__vueParentComponent
   }
 
-  find(selector: string | RefSelector): DOMWrapper<any> {
+  find<K extends keyof HTMLElementTagNameMap>(
+    selector: K
+  ): DOMWrapper<HTMLElementTagNameMap[K]>
+  find<K extends keyof SVGElementTagNameMap>(
+    selector: K
+  ): DOMWrapper<SVGElementTagNameMap[K]>
+  find<T extends Element = Element>(selector: string): DOMWrapper<T>
+  find<T extends Node = Node>(selector: string | RefSelector): DOMWrapper<T>
+  find(selector: string | RefSelector): DOMWrapper<Node> {
     const result = super.find(selector)
     if (result.exists() && isRefSelector(selector)) {
       return this.element.contains(result.element)