Move provide work to addProvide function

Author: Austio

Diff for: src/lib/addProvide.js

@@ -0,0 +1,15 @@
+function addProvide (component, options) {
+  const provideOption = Object.assign({}, options.provide)
+  delete options.provide
+
+  const originalBeforeCreate = component.beforeCreate
+  component.beforeCreate = function beforeCreate () {
+    this._provided = provideOption
+
+    if (originalBeforeCreate) {
+      originalBeforeCreate.call(this)
+    }
+  }
+}
+
+export default addProvide

Diff for: src/mount.js

@@ -2,6 +2,7 @@ import Vue from 'vue'
 import VueWrapper from './VueWrapper'
 import addSlots from './lib/addSlots'
 import addGlobals from './lib/addGlobals'
+import addProvide from './lib/addProvide'
 
 Vue.config.productionTip = false
 
@@ -32,12 +33,7 @@ export default function mount (component, options = {}) {
   delete component._Ctor // eslint-disable-line no-param-reassign
 
   if (options.provide) {
-    const provide = Object.assign({}, options.provide)
-    delete options.provide
-
-    component.beforeCreate = function beforeCreate() {
-      this._provided = provide
-    }
+    addProvide(component, options)
   }
 
   const Constructor = Vue.extend(component)

Diff for: test/resources/components/component-with-inject.vue

@@ -9,7 +9,7 @@
         name: 'component-with-inject',
         inject: ['fromMount'],
         beforeCreate() {
-          this.setByOriginalBeforeCreate = 'created'
+          this.setInBeforeCreate = 'created'
         }
     };
 </script>

Diff for: test/unit/specs/mount/options/provide.spec.js

@@ -2,15 +2,19 @@ import mount from '../../../../../src/mount'
 import ComponentWithInject from '../../../../resources/components/component-with-inject.vue'
 
 describe('provide option in mount', () => {
-  const wrapper = mount(ComponentWithInject, {
-    provide: { fromMount: 'providedValue' }
-  })
-
   it('provides value which is injected by mounted component', () => {
+    const wrapper = mount(ComponentWithInject, {
+      provide: { fromMount: 'providedValue' }
+    })
+
     expect(wrapper.text()).to.contain('providedValue')
   })
 
   it('supports beforeCreate in component', () => {
-    expect(wrapper.vm.setByOriginalBeforeCreate).to.toEqual('created')
+    const wrapper = mount(ComponentWithInject, {
+      provide: { fromMount: 'providedValue' }
+    })
+
+    expect(wrapper.vm.setInBeforeCreate).to.equal('created')
   })
 })