ComfyUI_frontend/tests-ui/tests/composables/useAssetWidgetData.desktop.test.ts

import { describe, expect, it, vi } from 'vitest'
import { ref } from 'vue'

import { useAssetWidgetData } from '@/renderer/extensions/vueNodes/widgets/composables/useAssetWidgetData'

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

const mockUpdateModelsForNodeType = vi.fn()
const mockGetCategoryForNodeType = vi.fn()

vi.mock('@/stores/assetsStore', () => ({
  useAssetsStore: () => ({
    modelAssetsByNodeType: new Map(),
    modelLoadingByNodeType: new Map(),
    modelErrorByNodeType: new Map(),
    updateModelsForNodeType: mockUpdateModelsForNodeType
  })
}))

vi.mock('@/stores/modelToNodeStore', () => ({
  useModelToNodeStore: () => ({
    getCategoryForNodeType: mockGetCategoryForNodeType
  })
}))

describe('useAssetWidgetData (desktop/isCloud=false)', () => {
  it('returns empty/default values without calling stores', () => {
    const nodeType = ref('CheckpointLoaderSimple')
    const { category, assets, dropdownItems, isLoading, error } =
      useAssetWidgetData(nodeType)

    expect(category.value).toBeUndefined()
    expect(assets.value).toEqual([])
    expect(dropdownItems.value).toEqual([])
    expect(isLoading.value).toBe(false)
    expect(error.value).toBeNull()
    expect(mockUpdateModelsForNodeType).not.toHaveBeenCalled()
    expect(mockGetCategoryForNodeType).not.toHaveBeenCalled()
  })
})