element-plus/packages/message/__tests__/message-manager.spec.ts

import { getStyle } from '@element-plus/utils/dom'
import * as Vue from 'vue'
import Message, { close } from '../src/message'

type UnknownProps = Record<string, unknown>

jest.useFakeTimers()

const selector = '.el-message'
// TODO: testing the original transition with `nextTick`
const Transition = (_: UnknownProps, { attrs, slots }) =>
  Vue.h('div', attrs, slots)
Transition.displayName = 'Transition'

describe('Message on command', () => {
  const oldTransition = Vue.Transition
  beforeAll(() => {
    // eslint-disable-next-line
    ;(Vue as any).Transition = Transition
  })

  afterAll(() => {
    // eslint-disable-next-line
    ;(Vue as any).Transition = oldTransition
  })

  test('it should get component handle', async () => {
    const handle = Message()
    expect(document.querySelector(selector)).toBeTruthy()
    handle.close()
    expect(document.querySelector(selector)).toBeFalsy()
  })

  test('it should be able to manually close a message', () => {
    Message()
    const element = document.querySelector(selector)
    expect(element).toBeTruthy()
    close(element.id)
    expect(document.querySelector(selector)).toBeNull()
  })

  test('it should close all messages', async () => {
    const onClose = jest.fn()
    const instances = []
    for (let i = 0; i < 4; i++) {
      const instance = Message({
        onClose,
      })
      instances.push(instance)
    }

    const elements = document.querySelectorAll(selector)
    expect(elements.length).toBe(4)
    Message.closeAll()
    await Vue.nextTick()
    for (let i = 0; i < instances.length; i++) {
      const instance = instances[i]
      instance.close()
    }

    expect(onClose).toHaveBeenCalledTimes(4)
    expect(document.querySelectorAll(selector).length).toBe(0)
  })

  test('it should stack messages', () => {
    const messages = [Message(), Message(), Message()]

    const elements = document.querySelectorAll(selector)
    expect(elements.length).toBe(3)

    const getTopValue = (elm: Element): number =>
      parseInt(getStyle(elm as HTMLElement, 'top'), 10)

    const topValues = []
    elements.forEach(e => {
      topValues.push(getTopValue(e))
    })

    for (let i = 1; i < topValues.length; i++) {
      expect(topValues[i - 1]).toBeLessThan(topValues[i])
    }

    messages.forEach(m => m.close())
  })

  test('it should have 4 other types of message', () => {
    expect(Message.success).toBeInstanceOf(Function)
    expect(Message.warning).toBeInstanceOf(Function)
    expect(Message.info).toBeInstanceOf(Function)
    expect(Message.error).toBeInstanceOf(Function)
  })
})