fix: add support for vue.extended components + add test (#112)

* fix: add support for vue.extended components + add test

* Update test

Author: feedthejim

package.json

@@ -83,7 +83,7 @@
     "sinon": "^2.3.2",
     "sinon-chai": "^2.10.0",
     "typescript": "^2.4.1",
-    "vue": "^2.4.4",
+    "vue": "^2.5.2",
     "vue-loader": "^13.0.5",
     "vue-router": "^2.7.0",
     "vue-template-compiler": "^2.4.4",

src/lib/create-instance.js

@@ -42,7 +42,7 @@ export default function createConstructor (component: Component, options: Option
     compileTemplate(component)
   }
 
-  const Constructor = vue.extend(component)
+  const Constructor = vue.extend(component.extend ? component.options : component)
 
   if (options.mocks) {
     addMocks(options.mocks, Constructor)

test/unit/specs/mount.spec.js

@@ -1,3 +1,4 @@
+import Vue from 'vue'
 import { compileToFunctions } from 'vue-template-compiler'
 import mount from '~src/mount'
 import ComponentWithProps from '~resources/components/component-with-props.vue'
@@ -41,6 +42,17 @@ describe('mount', () => {
     }
   })
 
+  it('returns new VueWrapper with mounted Vue instance initialized with Vue.extend with props, if passed as propsData', () => {
+    const prop1 = { test: 'TEST' }
+    const wrapper = mount(Vue.extend(ComponentWithProps), { propsData: { prop1 }})
+    expect(wrapper.vm).to.be.an('object')
+    if (wrapper.vm.$props) {
+      expect(wrapper.vm.$props.prop1).to.equal(prop1)
+    } else {
+      expect(wrapper.vm.$options.propsData.prop1).to.equal(prop1)
+    }
+  })
+
   it('does not use cached component', () => {
     ComponentWithMixin.methods.someMethod = sinon.stub()
     mount(ComponentWithMixin)