ComfyUI_frontend/browser_tests/tests/vueNodes/widgets/float/floatWidget.spec.ts

import {
  comfyExpect as expect,
  comfyPageFixture as test
} from '@e2e/fixtures/ComfyPage'

test.describe('Vue Float Widget', { tag: '@vue-nodes' }, () => {
  test('evaluates expression input using the widget precision', async ({
    comfyPage
  }) => {
    await comfyPage.workflow.loadWorkflow('vueNodes/linked-int-widget')

    const cfgWidget = comfyPage.vueNodes
      .getWidgetByName('KSampler', 'cfg')
      .first()
    const { input } = comfyPage.vueNodes.getInputNumberControls(cfgWidget)

    // 2.1 + 1.23 = 3.33 — well away from any half-point, so the expected
    // value is the same under half-up, half-even, and half-away-from-zero.
    await input.fill('2.1 + 1.23')
    await input.blur()

    await expect(input).toHaveValue('3.3')
  })

  test('increment and decrement buttons update the value', async ({
    comfyPage
  }) => {
    await comfyPage.workflow.loadWorkflow('vueNodes/linked-int-widget')

    // cfg has room to grow (initial 8, max 100); denoise is at its max (1)
    // so its increment button is disabled and can't be used here.
    const cfgWidget = comfyPage.vueNodes
      .getWidgetByName('KSampler', 'cfg')
      .first()
    const controls = comfyPage.vueNodes.getInputNumberControls(cfgWidget)

    const initial = Number(await controls.input.inputValue())

    await controls.incrementButton.click()
    await expect
      .poll(async () => Number(await controls.input.inputValue()))
      .toBeGreaterThan(initial)
    const afterIncrement = Number(await controls.input.inputValue())

    await controls.decrementButton.click()
    await expect
      .poll(async () => Number(await controls.input.inputValue()))
      .toBeLessThan(afterIncrement)
  })

  test('persists value across a serialize and reload round-trip', async ({
    comfyPage
  }) => {
    await comfyPage.workflow.loadWorkflow('vueNodes/linked-int-widget')

    const cfgWidget = comfyPage.vueNodes
      .getWidgetByName('KSampler', 'cfg')
      .first()
    const { input } = comfyPage.vueNodes.getInputNumberControls(cfgWidget)

    // KSampler's cfg is precision=1, so pick a value that lands on the grid
    await input.fill('7.5')
    await input.blur()
    await expect(input).toHaveValue('7.5')

    const serialized = await comfyPage.workflow.getExportedWorkflow()
    await comfyPage.workflow.loadGraphData(serialized)
    await comfyPage.vueNodes.waitForNodes()

    const [ksamplerNode] = await comfyPage.nodeOps.getNodeRefsByType('KSampler')
    if (!ksamplerNode) {
      throw new Error('KSampler node not found after reload')
    }

    const cfgWidgetAfterReload = await ksamplerNode.getWidgetByName('cfg')
    await expect.poll(() => cfgWidgetAfterReload.getValue()).toBe(7.5)

    // Also assert the Vue input mirrors the persisted value, so a broken
    // rehydration path that keeps litegraph state correct but renders stale
    // UI (or vice versa) still fails this test.
    await expect(input).toHaveValue('7.5')
  })
})