fix: added missing find overrides (#1333)

vuejs · Feb 21, 2022 · 17d5984 · 17d5984
1 parent aa84fed
commit 17d5984
Showing 1 changed file with 9 additions and 1 deletion.
diff --git 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)