From f8f897bcb7f6a3906019b3aba83c6597ee20acff Mon Sep 17 00:00:00 2001
From: freakzlike <freakzlike@gmail.com>
Date: Thu, 20 Oct 2022 19:27:10 +0200
Subject: [PATCH] feat(setValue): allow array for multiselect .setValue()
 method

---
 docs/api/index.md                       | 18 ++++++++++++-
 src/domWrapper.ts                       | 20 +++++++++------
 tests/components/ComponentWithInput.vue |  7 +++++
 tests/setValue.spec.ts                  | 34 +++++++++++++++++++++++++
 4 files changed, 70 insertions(+), 9 deletions(-)

diff --git a/docs/api/index.md b/docs/api/index.md
index a5826c6e2..2b8043a42 100644
--- a/docs/api/index.md
+++ b/docs/api/index.md
@@ -1633,6 +1633,12 @@ setValue(value: unknown, prop?: string): Promise<void>
 
   <input type="checkbox" v-model="checked" />
   <div v-if="checked">The input has been checked!</div>
+
+  <select v-model="multiselectValue" multiple>
+    <option value="value1"></option>
+    <option value="value2"></option>
+    <option value="value3"></option>
+  </select>
 </template>
 
 <script>
@@ -1640,7 +1646,8 @@ export default {
   data() {
     return {
       text: '',
-      checked: false
+      checked: false,
+      multiselectValue: []
     }
   }
 }
@@ -1669,6 +1676,15 @@ test('setValue on input text', () => {
   await wrapper.find('input[type="text"]').setValue('hello!')
   expect(wrapper.find('p').text()).toBe('Text: hello!')
 })
+
+test('setValue on multi select', () => {
+  const wrapper = mount(Component)
+
+  // For select without multiple
+  await wrapper.find('select').setValue('value1')
+  // For select with multiple
+  await wrapper.find('select').setValue(['value1', 'value3'])
+})
 ```
 
 ::: warning
diff --git a/src/domWrapper.ts b/src/domWrapper.ts
index f06b7eaeb..ccd3a2f9b 100644
--- a/src/domWrapper.ts
+++ b/src/domWrapper.ts
@@ -105,16 +105,20 @@ export class DOMWrapper<NodeType extends Node> extends BaseWrapper<NodeType> {
       return this.setChecked(value)
     } else if (tagName === 'INPUT' && type === 'radio') {
       return this.setChecked(value)
-    } else if (
-      tagName === 'INPUT' ||
-      tagName === 'TEXTAREA' ||
-      tagName === 'SELECT'
-    ) {
+    } else if (tagName === 'SELECT') {
+      if (Array.isArray(value)) {
+        const selectElement = element as unknown as HTMLSelectElement
+        for (let i = 0; i < selectElement.options.length; i++) {
+          const option = selectElement.options[i]
+          option.selected = value.includes(option.value)
+        }
+      } else {
+        element.value = value
+      }
+      return this.trigger('change')
+    } else if (tagName === 'INPUT' || tagName === 'TEXTAREA') {
       element.value = value
 
-      if (tagName === 'SELECT') {
-        return this.trigger('change')
-      }
       this.trigger('input')
       // trigger `change` for `v-model.lazy`
       return this.trigger('change')
diff --git a/tests/components/ComponentWithInput.vue b/tests/components/ComponentWithInput.vue
index 832263dd6..64f958079 100644
--- a/tests/components/ComponentWithInput.vue
+++ b/tests/components/ComponentWithInput.vue
@@ -30,12 +30,18 @@
         <option value="selectC"></option>
       </optgroup>
     </select>
+    <select v-model="multiselectVal" class="multiselect" multiple>
+      <option value="selectA"></option>
+      <option value="selectB"></option>
+      <option value="selectC"></option>
+    </select>
     <label id="label-el"></label>
 
     <span class="checkboxResult" v-if="checkboxVal">checkbox checked</span>
     <span class="counter">{{ counter }}</span>
     {{ textVal }}
     {{ selectVal }}
+    {{ multiselectVal }}
     {{ radioVal }}
     {{ lazy }}
   </div>
@@ -55,6 +61,7 @@ export default defineComponent({
       textareaVal: undefined,
       radioVal: undefined,
       selectVal: undefined,
+      multiselectVal: [],
       counter: 0
     }
   },
diff --git a/tests/setValue.spec.ts b/tests/setValue.spec.ts
index 3b89c588d..bf85020eb 100644
--- a/tests/setValue.spec.ts
+++ b/tests/setValue.spec.ts
@@ -40,6 +40,9 @@ describe('setValue', () => {
 
       expect(select.element.value).toEqual('selectB')
       expect(wrapper.text()).toContain('selectB')
+
+      expect(wrapper.emitted('change')).toHaveLength(1)
+      expect(wrapper.emitted('input')).toBeUndefined()
     })
 
     it('as an option of a select as selected', async () => {
@@ -82,6 +85,37 @@ describe('setValue', () => {
 
       expect(handle).toHaveBeenCalledTimes(1)
     })
+
+    it('sets element of multiselect value', async () => {
+      const wrapper = mount(ComponentWithInput)
+      const select = wrapper.find<HTMLSelectElement>('select.multiselect')
+      await select.setValue(['selectA', 'selectC'])
+
+      const selectedOptions = Array.from(select.element.selectedOptions).map(
+        (o) => o.value
+      )
+      expect(selectedOptions).toEqual(['selectA', 'selectC'])
+      expect(wrapper.vm.multiselectVal).toEqual(['selectA', 'selectC'])
+
+      expect(wrapper.emitted('change')).toHaveLength(1)
+      expect(wrapper.emitted('input')).toBeUndefined()
+    })
+
+    it('overrides elements of multiselect', async () => {
+      const wrapper = mount(ComponentWithInput)
+      const select = wrapper.find<HTMLSelectElement>('select.multiselect')
+      await select.setValue(['selectA', 'selectC'])
+      await select.setValue(['selectB'])
+
+      const selectedOptions = Array.from(select.element.selectedOptions).map(
+        (o) => o.value
+      )
+      expect(selectedOptions).toEqual(['selectB'])
+      expect(wrapper.vm.multiselectVal).toEqual(['selectB'])
+
+      expect(wrapper.emitted('change')).toHaveLength(2)
+      expect(wrapper.emitted('input')).toBeUndefined()
+    })
   })
 
   describe('on radio and checkbox', () => {