fix: use boolean for VueWrapper (#763)

vuejs · Jun 27, 2018 · 6fa6ecd · 6fa6ecd
1 parent ff5220d
commit 6fa6ecd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/packages/test-utils/src/vue-wrapper.js b/packages/test-utils/src/vue-wrapper.js
@@ -7,7 +7,7 @@ import { orderWatchers } from './order-watchers'
 
 export default class VueWrapper extends Wrapper implements BaseWrapper {
  constructor (vm: Component, options: WrapperOptions) {
- super(vm._vnode, options)
+ super(vm._vnode, options, true)
 
  // $FlowIgnore : issue with defineProperty
  Object.defineProperty(this, 'vnode', {

diff --git a/packages/test-utils/src/wrapper.js b/packages/test-utils/src/wrapper.js
@@ -34,11 +34,15 @@ export default class Wrapper implements BaseWrapper {
  version: number;
  isFunctionalComponent: boolean;
 
- constructor (node: VNode | Element, options: WrapperOptions) {
+ constructor (
+ node: VNode | Element,
+ options: WrapperOptions,
+ isVueWrapper?: boolean
+ ) {
  const vnode = node instanceof Element ? null : node
  const element = node instanceof Element ? node : node.elm
  // Prevent redefine by VueWrapper
- if (this.constructor.name === 'Wrapper') {
+ if (!isVueWrapper) {
  // $FlowIgnore
  Object.defineProperty(this, 'vnode', {
  get: () => vnode,