chore: add overloads for attributes method

Author: cexbrayat

src/dom-wrapper.ts

@@ -18,11 +18,13 @@ export class DOMWrapper<ElementType extends Element> {
     return Array.from(classes)
   }
 
-  attributes(key?: string) {
+  attributes(): { [key: string]: string }
+  attributes(key: string): string
+  attributes(key?: string): { [key: string]: string } | string {
     const attributes = this.element.attributes
-    const attributeMap = {}
+    const attributeMap: { [key: string]: string } = {}
     for (let i = 0; i < attributes.length; i++) {
-      const att = attributes.item(i)
+      const att = attributes.item(i)!
       attributeMap[att.localName] = att.value
     }
 

src/vue-wrapper.ts

@@ -54,7 +54,9 @@ export class VueWrapper<T extends ComponentPublicInstance> {
     return new DOMWrapper(this.element).classes(className)
   }
 
-  attributes(key?: string) {
+  attributes(): { [key: string]: string }
+  attributes(key: string): string
+  attributes(key?: string): { [key: string]: string } | string {
     return new DOMWrapper(this.element).attributes(key)
   }
 

test-dts/wrapper.d-test.ts

@@ -86,3 +86,9 @@ expectType<SVGLineElement>(line.element)
 // string selector
 byClass = domWrapper.get('.todo')
 expectType<Element>(byClass.element)
+
+// attributes
+expectType<{ [key: string]: string }>(wrapper.attributes())
+expectType<string>(wrapper.attributes('key'))
+expectType<{ [key: string]: string }>(domWrapper.attributes())
+expectType<string>(domWrapper.attributes('key'))