feat: improve returned type for element getter (#2406)

Author: nandi95

src/vueWrapper.ts

@@ -213,7 +213,7 @@ export class VueWrapper<
     }
   }
 
-  get element(): Element {
+  get element(): T['$el'] {
     // if the component has multiple root elements, we use the parent's element
     return this.hasMultipleRoots ? this.parentElement : this.vm.$el
   }

test-dts/wrapper.d-test.ts

@@ -14,6 +14,16 @@ const domWrapper = wrapper.find('#other')
 let inputMaybe = wrapper.find('input')
 expectType<HTMLInputElement | undefined>(inputMaybe.element)
 
+let buttonComp = mount(
+  defineComponent({ template: '<button>Click me</button>' })
+)
+expectType<HTMLButtonElement>(buttonComp.element)
+
+let multiRootComp = mount(defineComponent({ template: '<span /><span />' }))
+// this will be a wrapper div element like so: <div data-v-app="">
+expectType<HTMLDivElement>(multiRootComp.element)
+expectType<HTMLSpanElement>(multiRootComp.element.firstElementChild)
+
 // SVG element selector
 let lineMaybe = wrapper.find('line')
 expectType<SVGLineElement | undefined>(lineMaybe.element)