ComfyUI_frontend/src/renderer/extensions/vueNodes/widgets/components/WidgetSelect.asset-mode.test.ts

import { createTestingPinia } from '@pinia/testing'
import { flushPromises, mount } from '@vue/test-utils'
import PrimeVue from 'primevue/config'
import { beforeEach, describe, expect, it, vi } from 'vitest'

import type { SimplifiedWidget } from '@/types/simplifiedWidget'

import WidgetSelect from '@/renderer/extensions/vueNodes/widgets/components/WidgetSelect.vue'

// Mock modules
vi.mock('@/platform/distribution/types', () => ({
  isCloud: true
}))

vi.mock('@/platform/assets/services/assetService', () => ({
  assetService: {
    isAssetBrowserEligible: vi.fn(() => true)
  }
}))

const mockSettingStoreGet = vi.fn()

vi.mock('@/platform/settings/settingStore', () => ({
  useSettingStore: vi.fn(() => ({
    get: mockSettingStoreGet
  }))
}))

// Import after mocks are defined
import { assetService } from '@/platform/assets/services/assetService'
const mockAssetServiceEligible = vi.mocked(assetService.isAssetBrowserEligible)

describe('WidgetSelect asset mode', () => {
  const createWidget = (): SimplifiedWidget<string | undefined> => ({
    name: 'ckpt_name',
    type: 'combo',
    value: undefined,
    options: {
      values: []
    }
  })

  beforeEach(() => {
    vi.clearAllMocks()
    mockAssetServiceEligible.mockReturnValue(true)
    mockSettingStoreGet.mockReturnValue(true) // Default to true for UseAssetAPI
  })

  // Helper to mount with common setup
  const mountWidget = () => {
    return mount(WidgetSelect, {
      props: {
        widget: createWidget(),
        modelValue: undefined,
        nodeType: 'CheckpointLoaderSimple'
      },
      global: {
        plugins: [PrimeVue, createTestingPinia()]
      }
    })
  }

  it('uses dropdown when isCloud && UseAssetAPI && isEligible', async () => {
    const wrapper = mountWidget()
    await flushPromises()

    expect(
      wrapper.findComponent({ name: 'WidgetSelectDropdown' }).exists()
    ).toBe(true)
  })

  it('uses default widget when UseAssetAPI setting is false', () => {
    mockSettingStoreGet.mockReturnValue(false)
    const wrapper = mountWidget()

    expect(
      wrapper.findComponent({ name: 'WidgetSelectDefault' }).exists()
    ).toBe(true)
  })

  it('uses default widget when node is not eligible', () => {
    mockAssetServiceEligible.mockReturnValue(false)
    const wrapper = mountWidget()

    expect(
      wrapper.findComponent({ name: 'WidgetSelectDefault' }).exists()
    ).toBe(true)
  })
})