ComfyUI_frontend/browser_tests/fixtures/helpers/SettingsHelper.ts

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

import { nextFrame } from '@e2e/fixtures/utils/timing'

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

  async setSetting(settingId: string, settingValue: unknown): Promise<void> {
    await this.page.evaluate(
      async ({ id, value }) => {
        await window.app!.extensionManager.setting.set(id, value)
      },
      { id: settingId, value: settingValue }
    )
    await nextFrame(this.page)
  }

  async getSetting<T = unknown>(settingId: string): Promise<T> {
    return (await this.page.evaluate(async (id) => {
      return await window.app!.extensionManager.setting.get(id)
    }, settingId)) as T
  }
}