From a9f56716fcda80228e9e6f62124077beae6cebbb Mon Sep 17 00:00:00 2001 From: CodingOnStar Date: Tue, 10 Feb 2026 21:01:53 +0800 Subject: [PATCH] test: update unit tests to use Vitest framework - Refactored test files for data source options, drawer, and pipeline settings to utilize Vitest for improved testing capabilities. - Ensured consistent testing practices across components by importing necessary Vitest functions. --- .../create-from-pipeline/data-source-options/hooks.spec.tsx | 2 +- .../datasets/documents/detail/completed/common/drawer.spec.tsx | 3 +-- .../settings/pipeline-settings/process-documents/hooks.spec.ts | 2 +- 3 files changed, 3 insertions(+), 4 deletions(-) diff --git a/web/app/components/datasets/documents/create-from-pipeline/data-source-options/hooks.spec.tsx b/web/app/components/datasets/documents/create-from-pipeline/data-source-options/hooks.spec.tsx index 36c4ab8754..f3e97a9e77 100644 --- a/web/app/components/datasets/documents/create-from-pipeline/data-source-options/hooks.spec.tsx +++ b/web/app/components/datasets/documents/create-from-pipeline/data-source-options/hooks.spec.tsx @@ -1,6 +1,6 @@ import type { DataSourceNodeType } from '@/app/components/workflow/nodes/data-source/types' import { renderHook } from '@testing-library/react' - +import { beforeEach, describe, expect, it, vi } from 'vitest' import { useDatasourceIcon } from './hooks' const mockTransformDataSourceToTool = vi.fn() diff --git a/web/app/components/datasets/documents/detail/completed/common/drawer.spec.tsx b/web/app/components/datasets/documents/detail/completed/common/drawer.spec.tsx index bdc4ce0c90..f6b08d9bae 100644 --- a/web/app/components/datasets/documents/detail/completed/common/drawer.spec.tsx +++ b/web/app/components/datasets/documents/detail/completed/common/drawer.spec.tsx @@ -1,8 +1,7 @@ import { render, screen } from '@testing-library/react' - +import { beforeEach, describe, expect, it, vi } from 'vitest' import Drawer from './drawer' -// Capture the useKeyPress callback so tests can invoke it let capturedKeyPressCallback: ((e: KeyboardEvent) => void) | undefined vi.mock('ahooks', () => ({ diff --git a/web/app/components/datasets/documents/detail/settings/pipeline-settings/process-documents/hooks.spec.ts b/web/app/components/datasets/documents/detail/settings/pipeline-settings/process-documents/hooks.spec.ts index 7f72bca8e1..8d93dfb835 100644 --- a/web/app/components/datasets/documents/detail/settings/pipeline-settings/process-documents/hooks.spec.ts +++ b/web/app/components/datasets/documents/detail/settings/pipeline-settings/process-documents/hooks.spec.ts @@ -1,5 +1,5 @@ import { renderHook } from '@testing-library/react' - +import { beforeEach, describe, expect, it, vi } from 'vitest' import { useInputVariables } from './hooks' let mockPipelineId: string | undefined