refactor: Update setData method (address #563)

packages/test-utils/src/wrapper.js

@@ -420,8 +420,16 @@ export default class Wrapper implements BaseWrapper {
     }
 
     Object.keys(data).forEach((key) => {
-      // $FlowIgnore : Problem with possibly null this.vm
-      this.vm.$set(this.vm, [key], data[key])
+      if (typeof data[key] === 'object' && data[key] !== null) {
+        Object.keys(data[key]).forEach((key2) => {
+          const newObj = { ...data[key], ...data[key][key2] }
+          // $FlowIgnore : Problem with possibly null this.vm
+          this.vm.$set(this.vm, [key], newObj)
+        })
+      } else {
+        // $FlowIgnore : Problem with possibly null this.vm
+        this.vm.$set(this.vm, [key], data[key])
+      }
     })
   }
 

test/specs/wrapper/setData.spec.js

@@ -132,4 +132,25 @@ describeWithShallowAndMount('setData', (mountingMethod) => {
     wrapper.setData({ message: null })
     expect(wrapper.text()).to.equal('There is no message yet')
   })
+  it('should update a data object', () => {
+    const Cmp = {
+      data: () => ({
+        anObject: {
+          propA: {
+            prop1: 'a'
+          },
+          propB: 'b'
+        }
+      })
+    }
+    const wrapper = mountingMethod(Cmp)
+    wrapper.setData({
+      anObject: {
+        propA: {
+          prop1: 'c'
+        }
+      }
+    })
+    expect(wrapper.vm.anObject.propA.prop1).to.equal('c')
+  })
 })