test(workflow): add unit tests for workflow components (#33741)

Co-authored-by: CodingOnStar <hanxujiang@dify.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
2026-06-20 07:07:35 +08:00 · 2026-03-19 18:35:16 +08:00
parent df0ded210f
commit 4df602684b
115 changed files with 8239 additions and 1470 deletions
@@ -0,0 +1,101 @@
+import type { ComponentProps, ReactNode } from 'react'
+import type { OnSelectBlock } from '@/app/components/workflow/types'
+import { render, screen } from '@testing-library/react'
+import userEvent from '@testing-library/user-event'
+import { BlockEnum } from '@/app/components/workflow/types'
+import DataSourceEmptyNode from '../index'
+
+const mockUseReplaceDataSourceNode = vi.hoisted(() => vi.fn())
+
+vi.mock('../hooks', () => ({
+  useReplaceDataSourceNode: mockUseReplaceDataSourceNode,
+}))
+
+vi.mock('@/app/components/workflow/block-selector', () => ({
+  default: ({
+    onSelect,
+    trigger,
+  }: {
+    onSelect: OnSelectBlock
+    trigger: ((open?: boolean) => ReactNode) | ReactNode
+  }) => (
+    <div>
+      {typeof trigger === 'function' ? trigger(false) : trigger}
+      <button
+        type="button"
+        onClick={() => onSelect(BlockEnum.DataSource, {
+          plugin_id: 'plugin-id',
+          provider_type: 'datasource',
+          provider_name: 'file',
+          datasource_name: 'local-file',
+          datasource_label: 'Local File',
+          title: 'Local File',
+        })}
+      >
+        select data source
+      </button>
+    </div>
+  ),
+}))
+
+type DataSourceEmptyNodeProps = ComponentProps<typeof DataSourceEmptyNode>
+
+const createNodeProps = (): DataSourceEmptyNodeProps => ({
+  id: 'data-source-empty-node',
+  data: {
+    width: 240,
+    height: 88,
+  },
+  type: 'default',
+  selected: false,
+  zIndex: 0,
+  isConnectable: true,
+  xPos: 0,
+  yPos: 0,
+  dragging: false,
+  dragHandle: undefined,
+} as unknown as DataSourceEmptyNodeProps)
+
+describe('DataSourceEmptyNode', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    mockUseReplaceDataSourceNode.mockReturnValue({
+      handleReplaceNode: vi.fn(),
+    })
+  })
+
+  // The empty datasource node should render the add trigger and forward selector choices.
+  describe('Rendering and Selection', () => {
+    it('should render the datasource add trigger', () => {
+      render(
+        <DataSourceEmptyNode {...createNodeProps()} />,
+      )
+
+      expect(screen.getByText('workflow.nodes.dataSource.add')).toBeInTheDocument()
+      expect(screen.getByText('workflow.blocks.datasource')).toBeInTheDocument()
+    })
+
+    it('should forward block selections to the replace hook', async () => {
+      const user = userEvent.setup()
+      const handleReplaceNode = vi.fn()
+      mockUseReplaceDataSourceNode.mockReturnValue({
+        handleReplaceNode,
+      })
+
+      render(
+        <DataSourceEmptyNode {...createNodeProps()} />,
+      )
+
+      await user.click(screen.getByRole('button', { name: 'select data source' }))
+
+      expect(handleReplaceNode).toHaveBeenCalledWith(BlockEnum.DataSource, {
+        plugin_id: 'plugin-id',
+        provider_type: 'datasource',
+        provider_name: 'file',
+        datasource_name: 'local-file',
+        datasource_label: 'Local File',
+        title: 'Local File',
+      })
+    })
+  })
+})