From 33725ca4cbef8977360d0187ab39a5f5492c3202 Mon Sep 17 00:00:00 2001
From: XieZongChen <46394163+amadeus711@users.noreply.github.com>
Date: Sun, 22 Aug 2021 22:02:34 -0500
Subject: [PATCH] test(modal): add modal test (#941)

---
 src/modal/tests/Modal.spec.ts  |   8 ---
 src/modal/tests/Modal.spec.tsx | 102 +++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+), 8 deletions(-)
 delete mode 100644 src/modal/tests/Modal.spec.ts
 create mode 100644 src/modal/tests/Modal.spec.tsx

diff --git a/src/modal/tests/Modal.spec.ts b/src/modal/tests/Modal.spec.ts
deleted file mode 100644
index 7f67de8ca..000000000
--- a/src/modal/tests/Modal.spec.ts
+++ /dev/null
@@ -1,8 +0,0 @@
-import { mount } from '@vue/test-utils'
-import { NModal } from '../index'
-
-describe('n-modal', () => {
-  it('should work with import on demand', () => {
-    mount(NModal)
-  })
-})
diff --git a/src/modal/tests/Modal.spec.tsx b/src/modal/tests/Modal.spec.tsx
new file mode 100644
index 000000000..4aedac9a8
--- /dev/null
+++ b/src/modal/tests/Modal.spec.tsx
@@ -0,0 +1,102 @@
+import { defineComponent, h, ref } from 'vue'
+import { mount } from '@vue/test-utils'
+import { ModalProps, NModal } from '../index'
+import { NButton } from '../../button'
+
+// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
+function mountModal ({
+  modalProps,
+  show
+}: {
+  modalProps?: ModalProps
+  show?: boolean
+}) {
+  return mount(
+    defineComponent({
+      setup () {
+        return {
+          show: ref(!!show)
+        }
+      },
+      render () {
+        return [
+          <NButton
+            onClick={() => {
+              this.show = true
+            }}
+          >
+            {{ default: () => 'Show' }}
+          </NButton>,
+          <NModal
+            show={this.show}
+            onUpdateShow={(show) => {
+              this.show = show
+            }}
+            {...modalProps}
+          >
+            {{
+              default: () => 'test'
+            }}
+          </NModal>
+        ]
+      }
+    }),
+    {
+      attachTo: document.body
+    }
+  )
+}
+
+describe('n-modal', () => {
+  it('should work with import on demand', () => {
+    mount(NModal)
+  })
+
+  it('should work with `display-directive` prop', async () => {
+    const mousedownEvent = new MouseEvent('mousedown', { bubbles: true })
+    const mouseupEvent = new MouseEvent('mouseup', { bubbles: true })
+    let wrapper = mountModal({})
+    expect(document.querySelector('.n-modal-body-wrapper')).toEqual(null)
+    await wrapper.find('button').trigger('click')
+    expect(document.querySelector('.n-modal-body-wrapper')).not.toEqual(null)
+    document.querySelector('.n-modal-mask')?.dispatchEvent(mousedownEvent)
+    document.querySelector('.n-modal-mask')?.dispatchEvent(mouseupEvent)
+    setTimeout(() => {
+      expect(
+        document.querySelector('.n-modal-body-wrapper')?.children.length
+      ).toBe(0)
+    }, 300)
+    wrapper.unmount()
+    wrapper = mountModal({ modalProps: { displayDirective: 'show' } })
+    expect(document.querySelector('.n-modal-body-wrapper')).toEqual(null)
+    await wrapper.find('button').trigger('click')
+    expect(document.querySelector('.n-modal-body-wrapper')).not.toEqual(null)
+    document.querySelector('.n-modal-mask')?.dispatchEvent(mousedownEvent)
+    document.querySelector('.n-modal-mask')?.dispatchEvent(mouseupEvent)
+    setTimeout(() => {
+      expect(
+        document.querySelector('.n-modal-body-wrapper')?.children.length
+      ).not.toBe(0)
+      expect(
+        document.querySelector('.n-modal-body-wrapper')?.getAttribute('style')
+      ).toContain('display: none')
+    }, 300)
+    wrapper.unmount()
+  })
+
+  it('should work with `preset` prop', async () => {
+    let wrapper = mountModal({ modalProps: { preset: 'card' } })
+    expect(document.querySelector('.n-modal-body-wrapper')).toEqual(null)
+    await wrapper.find('button').trigger('click')
+    expect(document.querySelector('.n-modal-body-wrapper')).not.toEqual(null)
+    expect(document.querySelector('.n-card')).not.toEqual(null)
+    wrapper.unmount()
+
+    wrapper = mountModal({ modalProps: { preset: 'dialog' } })
+    expect(document.querySelector('.n-modal-body-wrapper')).toEqual(null)
+    await wrapper.find('button').trigger('click')
+    expect(document.querySelector('.n-modal-body-wrapper')).not.toEqual(null)
+    expect(document.querySelector('.n-dialog')).not.toEqual(null)
+    wrapper.unmount()
+  })
+})