ComfyUI_frontend/tests-ui/tests/composables/widgets/useComboWidget.test.ts

import { createPinia, setActivePinia } from 'pinia'

import { useComboWidget } from '@/composables/widgets/useComboWidget'
import type { InputSpec } from '@/types/apiTypes'

jest.mock('@/scripts/widgets', () => ({
  addValueControlWidgets: jest.fn()
}))

describe('useComboWidget', () => {
  beforeEach(() => {
    setActivePinia(createPinia())
    jest.clearAllMocks()
  })

  it('should handle undefined spec', () => {
    const constructor = useComboWidget()
    const mockNode = {
      addWidget: jest.fn().mockReturnValue({ options: {} } as any)
    }

    const inputSpec: InputSpec = ['COMBO', undefined]

    const widget = constructor(
      mockNode as any,
      'inputName',
      inputSpec,
      undefined as any
    )

    expect(mockNode.addWidget).toHaveBeenCalledWith(
      'combo',
      'inputName',
      undefined, // default value
      expect.any(Function), // callback
      expect.objectContaining({
        values: 'COMBO'
      })
    )
    expect(widget).toEqual({
      widget: { options: {} }
    })
  })
})