From 17d5984ce7149eba89f1169f0b5d8da908673bc8 Mon Sep 17 00:00:00 2001
From: nandi95 <41805560+nandi95@users.noreply.github.com>
Date: Mon, 21 Feb 2022 00:34:25 +0000
Subject: [PATCH] fix: added missing find overrides (#1333)

---
 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)