mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-04-20 06:20:11 +00:00
eaf1f9ac77
## Summary Add 2 reusable test helpers for Playwright E2E tests, integrated into the ComfyPage fixture. These provide standardized patterns for mocking feature flags and queue state across all E2E tests. ## Changes - **`FeatureFlagHelper.ts`** — manage localStorage `ff:` prefixed feature flags (`seedFlags` for init-time, `setFlags` for runtime) and mock `/api/features` route - **`QueueHelper.ts`** — mock `/api/queue` and `/api/history` routes with configurable running/pending counts and success/error job entries - **`ComfyPage.ts`** — integrate both helpers as `comfyPage.featureFlags` and `comfyPage.queue` ## Review Focus - Helper API design: are `seedFlags`/`setFlags`/`mockServerFeatures` the right abstractions for feature flag testing? - Queue mock fidelity: does the mock history shape match real ComfyUI API responses closely enough? - These are test-only infrastructure — no production code changes. ## Stack This is the base PR for the Playwright E2E coverage stack. Waves 1-4 all branch from this and can merge independently once this lands: - **→ This PR**: Test infrastructure helpers - #9555: Toasts, error overlay, selection toolbox, linear mode, selection rectangle - #9556: Node search, bottom panel, focus mode, job history, side panel - #9557: Errors tab, node headers, queue notifications, settings sidebar - #9558: Minimap, widget copy, floating menus, node library essentials --------- Co-authored-by: GitHub Action <action@github.com>
80 lines
2.2 KiB
TypeScript
80 lines
2.2 KiB
TypeScript
import type { Page, Route } from '@playwright/test'
|
|
|
|
export class QueueHelper {
|
|
private queueRouteHandler: ((route: Route) => void) | null = null
|
|
private historyRouteHandler: ((route: Route) => void) | null = null
|
|
|
|
constructor(private readonly page: Page) {}
|
|
|
|
/**
|
|
* Mock the /api/queue endpoint to return specific queue state.
|
|
*/
|
|
async mockQueueState(
|
|
running: number = 0,
|
|
pending: number = 0
|
|
): Promise<void> {
|
|
this.queueRouteHandler = (route: Route) =>
|
|
route.fulfill({
|
|
status: 200,
|
|
contentType: 'application/json',
|
|
body: JSON.stringify({
|
|
queue_running: Array.from({ length: running }, (_, i) => [
|
|
i,
|
|
`running-${i}`,
|
|
{},
|
|
{},
|
|
[]
|
|
]),
|
|
queue_pending: Array.from({ length: pending }, (_, i) => [
|
|
i,
|
|
`pending-${i}`,
|
|
{},
|
|
{},
|
|
[]
|
|
])
|
|
})
|
|
})
|
|
await this.page.route('**/api/queue', this.queueRouteHandler)
|
|
}
|
|
|
|
/**
|
|
* Mock the /api/history endpoint with completed/failed job entries.
|
|
*/
|
|
async mockHistory(
|
|
jobs: Array<{ promptId: string; status: 'success' | 'error' }>
|
|
): Promise<void> {
|
|
const history: Record<string, unknown> = {}
|
|
for (const job of jobs) {
|
|
history[job.promptId] = {
|
|
prompt: [0, job.promptId, {}, {}, []],
|
|
outputs: {},
|
|
status: {
|
|
status_str: job.status === 'success' ? 'success' : 'error',
|
|
completed: true
|
|
}
|
|
}
|
|
}
|
|
this.historyRouteHandler = (route: Route) =>
|
|
route.fulfill({
|
|
status: 200,
|
|
contentType: 'application/json',
|
|
body: JSON.stringify(history)
|
|
})
|
|
await this.page.route('**/api/history**', this.historyRouteHandler)
|
|
}
|
|
|
|
/**
|
|
* Clear all route mocks set by this helper.
|
|
*/
|
|
async clearMocks(): Promise<void> {
|
|
if (this.queueRouteHandler) {
|
|
await this.page.unroute('**/api/queue', this.queueRouteHandler)
|
|
this.queueRouteHandler = null
|
|
}
|
|
if (this.historyRouteHandler) {
|
|
await this.page.unroute('**/api/history**', this.historyRouteHandler)
|
|
this.historyRouteHandler = null
|
|
}
|
|
}
|
|
}
|