From f355e084cc69b8e23c1f5843444dc2518e1d8437 Mon Sep 17 00:00:00 2001
From: tjz <415800467@qq.com>
Date: Sun, 10 Jun 2018 15:42:38 +0800
Subject: [PATCH] test: add message test

---
 components/message/__tests__/index.test.js | 129 +++++++++++++++++++++
 1 file changed, 129 insertions(+)
 create mode 100644 components/message/__tests__/index.test.js

diff --git a/components/message/__tests__/index.test.js b/components/message/__tests__/index.test.js
new file mode 100644
index 000000000..f7ef4bb90
--- /dev/null
+++ b/components/message/__tests__/index.test.js
@@ -0,0 +1,129 @@
+import { mount } from '@vue/test-utils'
+import { asyncExpect } from '@/tests/utils'
+import message from '..'
+
+describe('message', () => {
+  beforeEach(() => {
+    document.body.outerHTML = ''
+  })
+
+  afterEach(() => {
+    message.destroy()
+  })
+
+  it('should be able to config top', async () => {
+    message.config({
+      top: '100px',
+    })
+    message.info('whatever')
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message')[0].style.top).toBe('100px')
+    })
+  })
+
+  it('should be able to config getContainer', () => {
+    message.config({
+      getContainer: () => {
+        const div = document.createElement('div')
+        div.className = 'custom-container'
+        document.body.appendChild(div)
+        return div
+      },
+    })
+    message.info('whatever')
+    expect(document.querySelectorAll('.custom-container').length).toBe(1)
+  })
+
+  it('should be able to hide manually', async () => {
+    const hide1 = message.info('whatever', 0)
+    const hide2 = message.info('whatever', 0)
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(2)
+      hide1()
+    }, 0)
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(1)
+      hide2()
+    }, 0)
+    expect(document.querySelectorAll('.ant-message-notice').length).toBe(0)
+  })
+
+  it('should be able to destroy globally', async () => {
+    await asyncExpect(() => {
+      message.info('whatever', 0)
+    })
+    await asyncExpect(() => {
+      message.info('whatever', 0)
+    })
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message').length).toBe(1)
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(2)
+    })
+    await asyncExpect(() => {
+      message.destroy()
+    })
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message').length).toBe(0)
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(0)
+    })
+  })
+
+  it('should not need to use duration argument when using the onClose arguments', () => {
+    message.info('whatever', () => {})
+  })
+
+  it('should have the default duration when using the onClose arguments', (done) => {
+    const defaultDuration = 3
+    const now = Date.now()
+    message.info('whatever', () => {
+      // calculate the approximately duration value
+      const aboutDuration = parseInt((Date.now() - now) / 1000, 10)
+      expect(aboutDuration).toBe(defaultDuration)
+      done()
+    })
+  })
+
+  // https:// github.com/ant-design/ant-design/issues/8201
+  it('should hide message correctly', async () => {
+    let hide
+    const Test = {
+      mounted () {
+        hide = message.loading('Action in progress..', 0)
+      },
+      render () {
+        return <div>test</div>
+      },
+    }
+    mount(Test, { sync: false })
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(1)
+      hide()
+    }, 0)
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(0)
+    }, 0)
+  })
+
+  // https://github.com/ant-design/ant-design/issues/8201
+  it('should destroy messages correctly', async () => {
+    // eslint-disable-next-line
+    const Test = {
+      mounted () {
+        message.loading('Action in progress1..', 0)
+        message.loading('Action in progress2..', 0)
+        setTimeout(() => message.destroy(), 1000)
+      },
+      render () {
+        return <div>test</div>
+      },
+    }
+    mount(Test, { sync: false })
+
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(2)
+    }, 0)
+    await asyncExpect(() => {
+      expect(document.querySelectorAll('.ant-message-notice').length).toBe(0)
+    }, 1500)
+  })
+})