Possible feature request

Author: shortdiv

packages/test-utils/src/wrapper.js

@@ -407,6 +407,17 @@ export default class Wrapper implements BaseWrapper {
     return _props || {} // Return an empty object if no props exist
   }
 
+  /**
+   * Sets vm data object while preserving existing k/v
+   */
+  setDataObj (obj: string, key: string, value: any) {
+    if (obj === undefined) {
+      this.vm.$set(this.vm, key, value)
+    } else {
+      this.vm.$set(obj, key, value)
+    }
+  }
+
   /**
    * Sets vm data
    */

test/specs/wrapper/setDataObj.spec.js

@@ -0,0 +1,39 @@
+import {
+  describeWithShallowAndMount,
+  vueVersion
+} from '~resources/utils'
+
+describeWithShallowAndMount('setDataObj', (mountingMethod) => {
+  let info
+
+  beforeEach(() => {
+    info = sinon.stub(console, 'info')
+  })
+
+  afterEach(() => {
+    info.restore()
+  })
+  it('should allow for setting a data value', () => {
+    const TestComponent = {
+      data: () => ({
+        text: 'hello'
+      })
+    }
+    const wrapper = mountingMethod(TestComponent)
+    wrapper.setDataObj(wrapper.vm, 'text', 'goodbye')
+    expect(wrapper.vm.text).to.equal('goodbye')
+  })
+  it('should allow for setting a data object', () => {
+    const TestComponent = {
+      data: () => ({
+        message: {
+          text: 'hello'
+        }
+      })
+    }
+    const wrapper = mountingMethod(TestComponent)
+    wrapper.setDataObj(wrapper.vm.message, 'read', true)
+    expect(wrapper.vm.message.text).to.equal('hello')
+    expect(wrapper.vm.message.read).to.equal(true)
+  })
+})