ComfyUI_frontend/browser_tests/fixtures/helpers/ToastHelper.ts

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

export class ToastHelper {
  public readonly visibleToasts: Locator
  public readonly toastErrors: Locator

  constructor(private readonly page: Page) {
    this.visibleToasts = page.locator('.p-toast-message:visible')
    this.toastErrors = page.locator('.p-toast-message.p-toast-message-error')
  }

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

    // Assert all toasts are closed
    await expect(this.visibleToasts).toHaveCount(0)
  }
}