fix(find): Allow finding self from DOMWrapper (#1077)

vuejs · Nov 17, 2021 · 60e47a2 · 60e47a2
1 parent 1809ef0
commit 60e47a2
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/domWrapper.ts b/src/domWrapper.ts
@@ -34,6 +34,10 @@ export class DOMWrapper<ElementType extends Element>
   ): DOMWrapper<SVGElementTagNameMap[K]>
   find<T extends Element>(selector: string): DOMWrapper<T>
   find(selector: string): DOMWrapper<Element> {
+    // allow finding the root element
+    if (this.element.matches(selector)) {
+      return this
+    }
     const result = this.element.querySelector(selector)
     if (result) {
       return new DOMWrapper(result)

diff --git a/tests/find.spec.ts b/tests/find.spec.ts
@@ -39,7 +39,7 @@ describe('find', () => {
           <component-b v-for="item in [1, 2]" :key="item">
             <input type="text" :value="item">
           </component-b>
-        </div> 
+        </div>
       `,
       components: { ComponentB }
     })
@@ -63,6 +63,18 @@ describe('find', () => {
     expect(wrapper.find('.foo').exists()).toBe(true)
   })
 
+  it('returns the root element from dom wrapper if it matches', () => {
+    const Component = defineComponent({
+      render() {
+        return h('div', { class: 'foo' }, 'text')
+      }
+    })
+
+    const wrapper = mount(Component)
+    const domWrapper = wrapper.find('.foo')
+    expect(domWrapper.find('.foo').exists()).toBe(true)
+  })
+
   it('can be chained', () => {
     const Component = defineComponent({
       render() {