mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-21 15:24:09 +00:00
a014e8a5da
Per Playwright best practices, assertions belong in tests, not fixtures.
Fixture methods should use waitFor() for preconditions instead.
- vueNodeFixtures.ts: setTitle(), cancelTitleEdit() use waitFor()
- Topbar.ts: closeTopbarMenu() uses waitFor({ state: 'hidden' })
- ComfyPage.ts: ConfirmDialog.click(), closeDialog(), clickDialogButton(),
closeToasts() all converted to waitFor()
- templates.ts: renamed waitForMinimumCardCount to expectMinimumCardCount
to clarify it contains an assertion (uses expect().toPass() pattern)
Amp-Thread-ID: https://ampcode.com/threads/T-019c11f8-acd2-7429-8bec-525ad47a47c4
Co-authored-by: Amp <amp@ampcode.com>
45 lines
1.3 KiB
TypeScript
45 lines
1.3 KiB
TypeScript
import type { Locator, Page } from '@playwright/test'
|
|
import { expect } from '@playwright/test'
|
|
import path from 'path'
|
|
|
|
import type {
|
|
TemplateInfo,
|
|
WorkflowTemplates
|
|
} from '../../src/platform/workflow/templates/types/template'
|
|
|
|
export class ComfyTemplates {
|
|
readonly content: Locator
|
|
readonly allTemplateCards: Locator
|
|
|
|
constructor(readonly page: Page) {
|
|
this.content = page.getByTestId('template-workflows-content')
|
|
this.allTemplateCards = page.locator('[data-testid^="template-workflow-"]')
|
|
}
|
|
|
|
async expectMinimumCardCount(count: number) {
|
|
await expect(async () => {
|
|
const cardCount = await this.allTemplateCards.count()
|
|
expect(cardCount).toBeGreaterThanOrEqual(count)
|
|
}).toPass({
|
|
timeout: 1_000
|
|
})
|
|
}
|
|
|
|
async loadTemplate(id: string) {
|
|
const templateCard = this.content.getByTestId(`template-workflow-${id}`)
|
|
await templateCard.scrollIntoViewIfNeeded()
|
|
await templateCard.getByRole('img').click()
|
|
}
|
|
|
|
async getAllTemplates(): Promise<TemplateInfo[]> {
|
|
const templates: WorkflowTemplates[] = await this.page.evaluate(() =>
|
|
window['app'].api.getCoreWorkflowTemplates()
|
|
)
|
|
return templates.flatMap((t) => t.templates)
|
|
}
|
|
|
|
getTemplatePath(filename: string): string {
|
|
return path.join('public', 'templates', filename)
|
|
}
|
|
}
|