ComfyUI_frontend/src/services/providers/registrySearchProvider.test.ts

import { beforeEach, describe, expect, it, vi } from 'vitest'

import { useComfyRegistrySearchProvider } from '@/services/providers/registrySearchProvider'
import { useComfyRegistryStore } from '@/stores/comfyRegistryStore'

// Mock the store
vi.mock('@/stores/comfyRegistryStore', () => ({
  useComfyRegistryStore: vi.fn()
}))

describe('useComfyRegistrySearchProvider', () => {
  const mockSearchCall = vi.fn()
  const mockSearchClear = vi.fn()
  const mockListAllPacksCall = vi.fn()
  const mockListAllPacksClear = vi.fn()

  const createMockStore = (
    params: Partial<ReturnType<typeof useComfyRegistryStore>> = {}
  ) => {
    return {
      search: {
        call: mockSearchCall,
        clear: mockSearchClear,
        cancel: vi.fn()
      },
      listAllPacks: {
        call: mockListAllPacksCall,
        clear: mockListAllPacksClear,
        cancel: vi.fn()
      },
      ...params
    } as Partial<ReturnType<typeof useComfyRegistryStore>> as ReturnType<
      typeof useComfyRegistryStore
    >
  }

  beforeEach(() => {
    vi.clearAllMocks()

    // Setup store mock
    vi.mocked(useComfyRegistryStore).mockReturnValue(createMockStore())
  })

  describe('searchPacks', () => {
    it('should search for packs by name', async () => {
      const mockResults = {
        nodes: [
          { id: '1', name: 'Test Pack 1' },
          { id: '2', name: 'Test Pack 2' }
        ]
      }
      mockSearchCall.mockResolvedValue(mockResults)

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('test', {
        pageSize: 10,
        pageNumber: 0,
        restrictSearchableAttributes: ['name', 'description']
      })

      expect(mockSearchCall).toHaveBeenCalledWith({
        search: 'test',
        comfy_node_search: undefined,
        limit: 10,
        page: 1
      })
      expect(result.nodePacks).toEqual(mockResults.nodes)
      expect(result.querySuggestions).toEqual([])
    })

    it('should search for packs by node names', async () => {
      const mockResults = {
        nodes: [{ id: '1', name: 'Pack with LoadImage node' }]
      }
      mockSearchCall.mockResolvedValue(mockResults)

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('LoadImage', {
        pageSize: 20,
        pageNumber: 1,
        restrictSearchableAttributes: ['comfy_nodes']
      })

      expect(mockSearchCall).toHaveBeenCalledWith({
        search: undefined,
        comfy_node_search: 'LoadImage',
        limit: 20,
        page: 2
      })
      expect(result.nodePacks).toEqual(mockResults.nodes)
    })

    it('should handle empty results', async () => {
      mockSearchCall.mockResolvedValue({ nodes: [] })

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('nonexistent', {
        pageSize: 10,
        pageNumber: 0
      })

      expect(result.nodePacks).toEqual([])
      expect(result.querySuggestions).toEqual([])
    })

    it('should handle null results', async () => {
      mockSearchCall.mockResolvedValue(null)

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('test', {
        pageSize: 10,
        pageNumber: 0
      })

      expect(result.nodePacks).toEqual([])
      expect(result.querySuggestions).toEqual([])
    })

    it('should handle results without nodes property', async () => {
      mockSearchCall.mockResolvedValue({})

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('test', {
        pageSize: 10,
        pageNumber: 0
      })

      expect(result.nodePacks).toEqual([])
      expect(result.querySuggestions).toEqual([])
    })

    it('should use listAllPacks for empty query', async () => {
      const mockResults = {
        nodes: [
          { id: '1', name: 'Pack 1' },
          { id: '2', name: 'Pack 2' }
        ]
      }
      mockListAllPacksCall.mockResolvedValue(mockResults)

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('', {
        pageSize: 20,
        pageNumber: 0
      })

      expect(mockListAllPacksCall).toHaveBeenCalledWith({
        limit: 20,
        page: 1
      })
      expect(mockSearchCall).not.toHaveBeenCalled()
      expect(result.nodePacks).toEqual(mockResults.nodes)
      expect(result.querySuggestions).toEqual([])
    })

    it('should use listAllPacks for whitespace-only query', async () => {
      const mockResults = {
        nodes: [{ id: '1', name: 'Pack 1' }]
      }
      mockListAllPacksCall.mockResolvedValue(mockResults)

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('   ', {
        pageSize: 10,
        pageNumber: 0
      })

      expect(mockListAllPacksCall).toHaveBeenCalledWith({
        limit: 10,
        page: 1
      })
      expect(mockSearchCall).not.toHaveBeenCalled()
      expect(result.nodePacks).toEqual(mockResults.nodes)
    })

    it('should handle empty results from listAllPacks', async () => {
      mockListAllPacksCall.mockResolvedValue({ nodes: [] })

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('', {
        pageSize: 10,
        pageNumber: 0
      })

      expect(result.nodePacks).toEqual([])
      expect(result.querySuggestions).toEqual([])
    })

    it('should handle null results from listAllPacks', async () => {
      mockListAllPacksCall.mockResolvedValue(null)

      const provider = useComfyRegistrySearchProvider()
      const result = await provider.searchPacks('', {
        pageSize: 10,
        pageNumber: 0
      })

      expect(result.nodePacks).toEqual([])
      expect(result.querySuggestions).toEqual([])
    })
  })

  describe('clearSearchCache', () => {
    it('should clear both search and listAllPacks caches', () => {
      const provider = useComfyRegistrySearchProvider()
      provider.clearSearchCache()

      expect(mockSearchClear).toHaveBeenCalled()
      expect(mockListAllPacksClear).toHaveBeenCalled()
    })
  })

  describe('getSortValue', () => {
    const testPack = {
      id: '1',
      name: 'Test Pack',
      downloads: 100,
      publisher: { id: 'pub1', name: 'Publisher One' },
      latest_version: {
        version: '1.0.0',
        createdAt: '2024-01-15T10:00:00Z'
      }
    }

    it('should return download count for downloads field', () => {
      const provider = useComfyRegistrySearchProvider()
      expect(provider.getSortValue(testPack, 'downloads')).toBe(100)
    })

    it('should return pack name for name field', () => {
      const provider = useComfyRegistrySearchProvider()
      expect(provider.getSortValue(testPack, 'name')).toBe('Test Pack')
    })

    it('should return publisher name for publisher field', () => {
      const provider = useComfyRegistrySearchProvider()
      expect(provider.getSortValue(testPack, 'publisher')).toBe('Publisher One')
    })

    it('should return timestamp for updated field', () => {
      const provider = useComfyRegistrySearchProvider()
      const timestamp = new Date('2024-01-15T10:00:00Z').getTime()
      expect(provider.getSortValue(testPack, 'updated')).toBe(timestamp)
    })

    it('should handle missing values gracefully', () => {
      const incompletePack = { id: '1', name: 'Incomplete' }
      const provider = useComfyRegistrySearchProvider()

      expect(provider.getSortValue(incompletePack, 'downloads')).toBe(0)
      expect(provider.getSortValue(incompletePack, 'publisher')).toBe('')
      expect(provider.getSortValue(incompletePack, 'updated')).toBe(0)
    })

    it('should return 0 for unknown sort fields', () => {
      const provider = useComfyRegistrySearchProvider()
      expect(provider.getSortValue(testPack, 'unknown')).toBe(0)
    })
  })

  describe('getSortableFields', () => {
    it('should return supported sort fields', () => {
      const provider = useComfyRegistrySearchProvider()
      const fields = provider.getSortableFields()

      expect(fields).toEqual([
        { id: 'downloads', label: 'Downloads', direction: 'desc' },
        { id: 'name', label: 'Name', direction: 'asc' },
        { id: 'publisher', label: 'Publisher', direction: 'asc' },
        { id: 'updated', label: 'Updated', direction: 'desc' }
      ])
    })
  })
})