dify/web/app/components/rag-pipeline/hooks/__tests__/use-pipeline-template.spec.ts

import { renderHook } from '@testing-library/react'
import { describe, expect, it, vi } from 'vitest'
import { usePipelineTemplate } from '../use-pipeline-template'

vi.mock('@/app/components/workflow/constants', () => ({
  START_INITIAL_POSITION: { x: 100, y: 200 },
}))

vi.mock('@/app/components/workflow/nodes/knowledge-base/default', () => ({
  default: {
    metaData: { type: 'knowledge-base' },
    defaultValue: { title: 'Knowledge Base' },
  },
}))

vi.mock('@/app/components/workflow/utils', () => ({
  generateNewNode: ({
    id,
    data,
    position,
  }: {
    id: string
    data: Record<string, unknown>
    position: { x: number; y: number }
  }) => ({
    newNode: { id, data, position, type: 'custom' },
  }),
}))

describe('usePipelineTemplate', () => {
  it('should return nodes array with one knowledge base node', () => {
    const { result } = renderHook(() => usePipelineTemplate())

    expect(result.current.nodes).toHaveLength(1)
    expect(result.current.nodes[0]!.id).toBe('knowledgeBase')
  })

  it('should return empty edges array', () => {
    const { result } = renderHook(() => usePipelineTemplate())

    expect(result.current.edges).toEqual([])
  })

  it('should set node type from knowledge-base default', () => {
    const { result } = renderHook(() => usePipelineTemplate())

    expect(result.current.nodes[0]!.data.type).toBe('knowledge-base')
  })

  it('should set node as selected', () => {
    const { result } = renderHook(() => usePipelineTemplate())

    expect(result.current.nodes[0]!.data.selected).toBe(true)
  })

  it('should position node offset from START_INITIAL_POSITION', () => {
    const { result } = renderHook(() => usePipelineTemplate())

    expect(result.current.nodes[0]!.position.x).toBe(600)
    expect(result.current.nodes[0]!.position.y).toBe(200)
  })

  it('should translate node title', () => {
    const { result } = renderHook(() => usePipelineTemplate())

    expect(result.current.nodes[0]!.data.title).toBe('workflow.blocks.knowledge-base')
  })
})