import { nextTick } from 'vue' import { afterEach, describe, expect, it, vi } from 'vitest' import { rAF } from '@element-plus/test-utils/tick' import Notification, { closeAll } from '../src/notify' import { ElNotification } from '..' import type { NotificationHandle } from '../src/notification' const selector = '.el-notification' describe('Notification on command', () => { afterEach(() => { closeAll() }) it('it should get component handle', async () => { const handle = Notification() await rAF() expect(document.querySelector(selector)).toBeDefined() handle.close() await rAF() await nextTick() expect(document.querySelector(selector)).toBeNull() expect( document.querySelector('[class^="container_notification"]') ).toBeNull() }) it('it should be able to render vnode', async () => { const testClassName = 'test-classname' const { close } = Notification({ duration: 0, message:
test-content
, }) await rAF() expect(document.querySelector(`.${testClassName}`)).toBeDefined() close() }) it('it should be able to close notification by manually close', async () => { const { close } = Notification({ duration: 0, }) await rAF() const element = document.querySelector(selector) expect(element).toBeDefined() close() await rAF() await nextTick() expect(document.querySelector(selector)).toBeNull() }) it('it should close all notifications', async () => { const notifications: NotificationHandle[] = [] const onClose = vi.fn() for (let i = 0; i < 4; i++) { notifications.push( Notification({ onClose, duration: 0, }) ) } // vi.runAllTicks() await rAF() expect(document.querySelectorAll(selector).length).toBe(4) closeAll() // vi.runAllTicks() await rAF() expect(onClose).toHaveBeenCalledTimes(notifications.length) expect(document.querySelectorAll(selector).length).toBe(0) }) it('it should be able to render all types notification', () => { for (const type of ['success', 'warning', 'error', 'info'] as const) { Notification[type]({}) expect(document.querySelector(`.el-icon-${type}`)).toBeDefined() } }) it('it should appendTo specified HTMLElement', async () => { const htmlElement = document.createElement('div') const handle = Notification({ appendTo: htmlElement, }) await rAF() expect(htmlElement.querySelector(selector)).toBeDefined() handle.close() await rAF() await nextTick() expect(htmlElement.querySelector(selector)).toBeNull() }) it('it should appendTo specified selector', async () => { const htmlElement = document.createElement('div') htmlElement.classList.add('notification-manager') document.body.appendChild(htmlElement) const handle = Notification({ appendTo: '.notification-manager', }) await rAF() expect(htmlElement.querySelector(selector)).toBeDefined() handle.close() await rAF() await nextTick() expect(htmlElement.querySelector(selector)).toBeNull() }) describe('context inheritance', () => { it('should globally inherit context correctly', () => { expect(ElNotification._context).toBe(null) const testContext = { config: { globalProperties: {}, }, _context: {}, } ElNotification.install?.(testContext as any) expect(ElNotification._context).not.toBe(null) expect(ElNotification._context).toBe(testContext._context) // clean up ElNotification._context = null }) }) })