fix: find ref childComponent (#268)

#267
vuejs · Dec 15, 2017 · 256086b · 256086b
1 parent 1897300
commit 256086b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/wrappers/wrapper.js b/src/wrappers/wrapper.js
@@ -244,6 +244,9 @@ export default class Wrapper implements BaseWrapper {
       if (!this.isVueComponent) {
         throwError('$ref selectors can only be used on Vue component wrappers')
       }
+      if (this.vm && this.vm.$refs && selector.ref in this.vm.$refs && this.vm.$refs[selector.ref] instanceof Vue) {
+        return new VueWrapper(this.vm.$refs[selector.ref], this.options)
+      }
       const nodes = findVNodesByRef(this.vnode, selector.ref)
       if (nodes.length === 0) {
         return new ErrorWrapper(`ref="${selector.ref}"`)
@@ -278,6 +281,9 @@ export default class Wrapper implements BaseWrapper {
       if (!this.isVueComponent) {
         throwError('$ref selectors can only be used on Vue component wrappers')
       }
+      if (this.vm && this.vm.$refs && selector.ref in this.vm.$refs && this.vm.$refs[selector.ref] instanceof Vue) {
+        return new WrapperArray([new VueWrapper(this.vm.$refs[selector.ref], this.options)])
+      }
       const nodes = findVNodesByRef(this.vnode, selector.ref)
       return new WrapperArray(nodes.map(node => new Wrapper(node, this.update, this.options)))
     }

diff --git a/test/unit/specs/mount/Wrapper/find.spec.js b/test/unit/specs/mount/Wrapper/find.spec.js
@@ -6,6 +6,7 @@ import ComponentWithoutName from '~resources/components/component-without-name.v
 import ComponentWithSlots from '~resources/components/component-with-slots.vue'
 import ComponentWithVFor from '~resources/components/component-with-v-for.vue'
 import Component from '~resources/components/component.vue'
+import VueWrapper from '~src/wrappers/vue-wrapper'
 import Wrapper from '~src/wrappers/wrapper'
 import ErrorWrapper from '~src/wrappers/error-wrapper'
 
@@ -172,7 +173,7 @@ describe('find', () => {
 
   it('returns Wrapper of Vue Components matching the ref in options object', () => {
     const wrapper = mount(ComponentWithChild)
-    expect(wrapper.find({ ref: 'child' })).to.be.instanceOf(Wrapper)
+    expect(wrapper.find({ ref: 'child' })).to.be.instanceOf(VueWrapper)
   })
 
   it('throws an error when ref selector is called on a wrapper that is not a Vue component', () => {