Merge pull request #1293 from xanf/fix-refs-on-functional-with-shallow

fix: pass refs to functional component stubs

Author: lmiller1990

packages/create-instance/create-component-stubs.js

@@ -42,7 +42,6 @@ function getCoreProperties(componentOptions: Component): Object {
     props: componentOptions.props,
     on: componentOptions.on,
     key: componentOptions.key,
-    ref: componentOptions.ref,
     domProps: componentOptions.domProps,
     class: componentOptions.class,
     staticClass: componentOptions.staticClass,
@@ -96,6 +95,7 @@ export function createStubFromComponent(
       return h(
         tagName,
         {
+          ref: componentOptions.functional ? context.data.ref : undefined,
           attrs: componentOptions.functional
             ? {
                 ...context.props,

test/specs/mounting-options/stubs.spec.js

@@ -525,6 +525,30 @@ describeWithShallowAndMount('options.stub', mountingMethod => {
     expect(wrapper.html()).to.contain('h1')
   })
 
+  it('maintains refs to components', () => {
+    const FunctionalComponentPassingRef = {
+      functional: true,
+      render: (h, context) => h('div', context.data)
+    }
+
+    const TestComponent = {
+      template: `
+        <div>
+          <test-component ref="normalChild" />
+          <test-functional-component ref="functionalChild" />
+        </div>
+      `,
+      components: {
+        testComponent: Component,
+        testFunctionalComponent: FunctionalComponentPassingRef
+      }
+    }
+
+    const wrapper = mountingMethod(TestComponent)
+    expect(wrapper.vm.$refs.normalChild).to.exist
+    expect(wrapper.vm.$refs.functionalChild).to.exist
+  })
+
   it('uses original component stub', () => {
     const Stub = {
       template: '<div />'