ComfyUI_frontend/browser_tests/fixtures/helpers/ToastHelper.ts

import type { Locator, Page } from '@playwright/test'

export class ToastHelper {
  constructor(private readonly page: Page) {}

  get visibleToasts(): Locator {
    return this.page.locator('.p-toast-message:visible')
  }

  async getToastErrorCount(): Promise<number> {
    return await this.page
      .locator('.p-toast-message.p-toast-message-error')
      .count()
  }

  async getVisibleToastCount(): Promise<number> {
    return await this.visibleToasts.count()
  }

  async closeToasts(requireCount = 0): Promise<void> {
    if (requireCount) {
      await this.visibleToasts
        .nth(requireCount - 1)
        .waitFor({ state: 'visible' })
    }

    // Clear all toasts
    const toastCloseButtons = await this.page
      .locator('.p-toast-close-button')
      .all()
    for (const button of toastCloseButtons) {
      await button.click()
    }

    // Wait for toasts to disappear
    await this.visibleToasts
      .first()
      .waitFor({ state: 'hidden', timeout: 1000 })
      .catch(() => {})
  }
}