feat: Store selector in Wrapper.find() / .findAll()

Author: winniehell

Diff for: docs/api/wrapper/README.md

@@ -24,6 +24,10 @@ A `Wrapper` is an object that contains a mounted component or vnode and methods
 
 `Boolean` (read-only): True if `sync` in mounting options was not `false`
 
+### `selector`
+
+`Selector`: the selector that was used by [`find()`](./find.md) or [`findAll()`](./findAll.md) to create this wrapper
+
 ## Methods
 
 !!!include(docs/api/wrapper/attributes.md)!!!

Diff for: packages/test-utils/src/wrapper.js

@@ -197,7 +197,9 @@ export default class Wrapper implements BaseWrapper {
       return new ErrorWrapper(selector)
     }
 
-    return createWrapper(node, this.options)
+    const wrapper = createWrapper(node, this.options)
+    wrapper.selector = selector
+    return wrapper
   }
 
   /**
@@ -210,9 +212,14 @@ export default class Wrapper implements BaseWrapper {
     const wrappers = nodes.map(node => {
       // Using CSS Selector, returns a VueWrapper instance if the root element
       // binds a Vue instance.
-      return createWrapper(node, this.options)
+      const wrapper = createWrapper(node, this.options)
+      wrapper.selector = selector
+      return wrapper
     })
-    return new WrapperArray(wrappers)
+
+    const wrapperArray = new WrapperArray(wrappers)
+    wrapperArray.selector = selector
+    return wrapperArray
   }
 
   /**

Diff for: test/specs/wrapper/find.spec.js

@@ -542,4 +542,34 @@ describeWithShallowAndMount('find', mountingMethod => {
       expect(wrapper.find('p').vm.$options.name).to.equal('bar')
     }
   )
+
+  it('stores CSS selector', () => {
+    const compiled = compileToFunctions('<div><p></p><p></p></div>')
+    const wrapper = mountingMethod(compiled)
+    const selector = 'p'
+    const result = wrapper.find(selector)
+    expect(result.selector.value).to.equal(selector)
+  })
+
+  it('stores ref selector', () => {
+    const compiled = compileToFunctions('<div><p ref="foo"></p></div>')
+    const wrapper = mountingMethod(compiled)
+    const selector = { ref: 'foo' }
+    const result = wrapper.find(selector)
+    expect(result.selector.value).to.equal(selector)
+  })
+
+  it('stores component selector', () => {
+    const wrapper = mountingMethod(ComponentWithChild)
+    const selector = Component
+    const result = wrapper.find(selector)
+    expect(result.selector.value).to.equal(selector)
+  })
+
+  it('stores name selector', () => {
+    const wrapper = mountingMethod(ComponentWithChild)
+    const selector = { name: 'test-component' }
+    const result = wrapper.find(selector)
+    expect(result.selector.value).to.equal(selector)
+  })
 })

Diff for: test/specs/wrapper/findAll.spec.js

@@ -343,4 +343,38 @@ describeWithShallowAndMount('findAll', mountingMethod => {
       expect(wrappers.at(2).vm.$options.name).to.equal('bar')
     }
   )
+
+  it('stores CSS selector', () => {
+    const compiled = compileToFunctions('<div><p></p><p></p></div>')
+    const wrapper = mountingMethod(compiled)
+    const selector = 'p'
+    const result = wrapper.findAll('p')
+    expect(result.selector.value).to.equal(selector)
+    expect(result.at(0).selector.value).to.equal(selector)
+  })
+
+  it('stores ref selector', () => {
+    const compiled = compileToFunctions('<div><div ref="foo" /></div>')
+    const wrapper = mountingMethod(compiled)
+    const selector = { ref: 'foo' }
+    const result = wrapper.findAll(selector)
+    expect(result.selector.value).to.equal(selector)
+    expect(result.at(0).selector.value).to.equal(selector)
+  })
+
+  it('stores component selector', () => {
+    const wrapper = mountingMethod(ComponentWithChild)
+    const selector = Component
+    const result = wrapper.findAll(selector)
+    expect(result.selector.value).to.equal(selector)
+    expect(result.at(0).selector.value).to.equal(selector)
+  })
+
+  it('stores name selector', () => {
+    const wrapper = mountingMethod(ComponentWithChild)
+    const selector = { name: 'test-component' }
+    const result = wrapper.findAll(selector)
+    expect(result.selector.value).to.equal(selector)
+    expect(result.at(0).selector.value).to.equal(selector)
+  })
 })