fix: run watchers in setData

Closes #149

Author: eddyerburgh

src/wrappers/wrapper.js

@@ -293,6 +293,14 @@ export default class Wrapper implements BaseWrapper {
       // $FlowIgnore : Problem with possibly null this.vm
       this.vm.$set(this.vm, [key], data[key])
     })
+
+    Object.keys(data).forEach((key) => {
+      // $FlowIgnore : Problem with possibly null this.vm
+      this.vm._watchers.forEach((watcher) => {
+        if (watcher.expression === key) { watcher.run() }
+      })
+    })
+
     this.update()
   }
 

test/resources/components/component-with-watch.vue

@@ -7,6 +7,12 @@
 
 <script>
   export default {
+    data: function () {
+      return {
+        data1: 'default data1',
+        data2: 'default data2'
+      }
+    },
     props: {
       prop1: { default: 'default prop1' },
       prop2: { default: 'default prop2' }
@@ -17,6 +23,12 @@
       },
       prop2 () {
         console.info(this.prop1)
+      },
+      data1 (val) {
+        this.data2 = val
+      },
+      data2 () {
+        console.info(this.data1)
       }
     }
   }

test/unit/specs/mount/Wrapper/setData.spec.js

@@ -1,8 +1,19 @@
 import { compileToFunctions } from 'vue-template-compiler'
 import mount from '~src/mount'
 import ComponentWithVIf from '~resources/components/component-with-v-if.vue'
+import ComponentWithWatch from '~resources/components/component-with-watch.vue'
 
 describe('setData', () => {
+  let info
+
+  beforeEach(() => {
+    info = sinon.stub(console, 'info')
+  })
+
+  afterEach(() => {
+    info.restore()
+  })
+
   it('sets component data and updates nested vm nodes when called on Vue instance', () => {
     const wrapper = mount(ComponentWithVIf)
     expect(wrapper.findAll('.child.ready').length).to.equal(0)
@@ -26,6 +37,20 @@ describe('setData', () => {
     expect(wrapper.hasClass('some-class')).to.be.true
   })
 
+  it('runs watch function when prop is updated', () => {
+    const wrapper = mount(ComponentWithWatch)
+    const data1 = 'testest'
+    wrapper.setData({ data1 })
+    expect(wrapper.vm.data2).to.equal(data1)
+  })
+
+  it('runs watch function after all props are updated', () => {
+    const wrapper = mount(ComponentWithWatch)
+    const data1 = 'testest'
+    wrapper.setData({ data2: 'newProp', data1 })
+    expect(info.args[0][0]).to.equal(data1)
+  })
+
   it('throws an error if node is not a Vue instance', () => {
     const message = 'wrapper.setData() can only be called on a Vue instance'
     const compiled = compileToFunctions('<div><p></p></div>')