ant-design-vue/components/message/__tests__/index.test.js

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)
  })
})