element-plus/packages/hooks/__tests__/use-prevent-global.ts

import { ref } from 'vue'
import { on, off } from '@element-plus/utils/dom'
import triggerEvent from '@element-plus/test-utils/trigger-event'
import usePreventGlobal from '../use-prevent-global'

describe('usePreventGlobal', () => {
  const evtName = 'keydown'
  const evt = jest.fn()
  beforeAll(() => {
    on(document.body, evtName, evt)
  })

  beforeEach(() => {
    evt.mockClear()
  })

  afterAll(() => {
    off(document.body, evtName, evt)
  })

  it('should prevent global event from happening', () => {
    const visible = ref(true)
    const evt2Trigger = jest.fn().mockReturnValue(true)
    usePreventGlobal(visible, evtName, evt2Trigger)

    triggerEvent(document.body, evtName)

    expect(evt).not.toBeCalled()
    expect(evt2Trigger).toHaveBeenCalled()
    visible.value = false
    // clean up
  })

  it('should not prevent global event from happening', () => {
    const visible = ref(true)
    const evt2Trigger = jest.fn().mockReturnValue(false)
    usePreventGlobal(visible, evtName, evt2Trigger)

    triggerEvent(document.body, evtName)

    expect(evt).toHaveBeenCalled()
    expect(evt2Trigger).toHaveBeenCalled()

    visible.value = false
  })
})