Merge pull request #3 from Austio/provideInject

Add provide option to mount

Author: eddyerburgh

Diff for: src/lib/addProvide.js

@@ -0,0 +1,15 @@
+function addProvide (component, options) {
+  const provide = typeof options.provide === 'function'
+    ? options.provide
+    : Object.assign({}, options.provide)
+
+  delete options.provide
+
+  options.beforeCreate = function vueTestUtilBeforeCreate () {
+    this._provided = typeof provide === 'function'
+      ? provide.call(this)
+      : provide
+  }
+}
+
+export default addProvide

Diff for: src/mount.js

@@ -4,6 +4,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
 
@@ -42,7 +43,12 @@ export default function mount (component: Component, options: MountOptions = {})
   // Remove cached constructor
   delete component._Ctor // eslint-disable-line no-param-reassign
 
+  if (options.provide) {
+    addProvide(component, options)
+  }
+
   const Constructor = Vue.extend(component)
+
   const vm = new Constructor(options)
 
   if (options.slots) {

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

@@ -0,0 +1,15 @@
+<template>
+    <div>
+      {{ fromMount }}
+    </div>
+</template>
+
+<script>
+    export default {
+        name: 'component-with-inject',
+        inject: ['fromMount'],
+        beforeCreate() {
+          this.setInBeforeCreate = 'created'
+        }
+    };
+</script>

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

@@ -0,0 +1,32 @@
+import mount from '../../../../../src/mount'
+import ComponentWithInject from '../../../../resources/components/component-with-inject.vue'
+
+describe('provide option in mount', () => {
+  it('provides objects which is injected by mounted component', () => {
+    const wrapper = mount(ComponentWithInject, {
+      provide: { fromMount: 'objectValue' }
+    })
+
+    expect(wrapper.text()).to.contain('objectValue')
+  })
+
+  it('provides function which is injected by mounted component', () => {
+    const wrapper = mount(ComponentWithInject, {
+      provide () {
+        return {
+          fromMount: 'functionValue'
+        }
+      }
+    })
+
+    expect(wrapper.text()).to.contain('functionValue')
+  })
+
+  it('supports beforeCreate in component', () => {
+    const wrapper = mount(ComponentWithInject, {
+      provide: { fromMount: '_' }
+    })
+
+    expect(wrapper.vm.setInBeforeCreate).to.equal('created')
+  })
+})