dify/web/app/components/workflow/__tests__/index.spec.tsx

import type { Edge, Node } from '../types'
import { render, screen } from '@testing-library/react'
import { useStoreApi } from 'reactflow'
import { useDatasetsDetailStore } from '../datasets-detail-store/store'
import WorkflowWithDefaultContext from '../index'
import { BlockEnum } from '../types'
import { useWorkflowHistoryStore } from '../workflow-history-store'

const nodes: Node[] = [
  {
    id: 'node-start',
    type: 'custom',
    position: { x: 0, y: 0 },
    data: {
      title: 'Start',
      desc: '',
      type: BlockEnum.Start,
    },
  },
]

const edges: Edge[] = [
  {
    id: 'edge-1',
    source: 'node-start',
    target: 'node-end',
    sourceHandle: null,
    targetHandle: null,
    type: 'custom',
    data: {
      sourceType: BlockEnum.Start,
      targetType: BlockEnum.End,
    },
  },
]

const ContextConsumer = () => {
  const { store, shortcutsEnabled } = useWorkflowHistoryStore()
  const datasetCount = useDatasetsDetailStore(state => Object.keys(state.datasetsDetail).length)
  const reactFlowStore = useStoreApi()

  return (
    <div>
      {`history:${store.getState().nodes.length}`}
      {` shortcuts:${String(shortcutsEnabled)}`}
      {` datasets:${datasetCount}`}
      {` reactflow:${String(!!reactFlowStore)}`}
    </div>
  )
}

describe('WorkflowWithDefaultContext', () => {
  it('wires the ReactFlow, workflow history, and datasets detail providers around its children', () => {
    render(
      <WorkflowWithDefaultContext
        nodes={nodes}
        edges={edges}
      >
        <ContextConsumer />
      </WorkflowWithDefaultContext>,
    )

    expect(
      screen.getByText('history:1 shortcuts:true datasets:0 reactflow:true'),
    ).toBeInTheDocument()
  })
})