bugfix/unmounting-on-multi-root

- Fix the logic which causes unmounting fail when it's multi-root component

Author: jw-foss

src/vueWrapper.ts

@@ -216,10 +216,7 @@ export class VueWrapper<T extends ComponentPublicInstance> {
       )
     }
 
-    if (this.parentElement) {
-      this.parentElement.removeChild(this.element)
-    }
-    this.__app.unmount(this.element)
+    this.__app.unmount(this.parentElement)
   }
 }
 

tests/unmount.spec.ts

@@ -0,0 +1,32 @@
+import { mount } from '../src'
+
+const AXIOM = 'Rem is the best girl'
+
+describe('Unmount', () => {
+  it('works on single root component', () => {
+    const Component = {
+      template: `
+        <div>${AXIOM}</div>
+      `,
+      onErrorCaptured(err: Error) {
+        throw err
+      }
+    }
+    const wrapper = mount(Component)
+    expect(() => wrapper.unmount()).not.toThrowError()
+  })
+
+  it('works on multi-root component', () => {
+    const Component = {
+      template: `
+        <div>${AXIOM}</div>
+        <div>${AXIOM}</div>
+      `,
+      onErrorCaptured(err: Error) {
+        throw err
+      }
+    }
+    const wrapper = mount(Component)
+    wrapper.unmount()
+  })
+})