From f272d68145b4d79a5fc050072244b65e01f8b772 Mon Sep 17 00:00:00 2001
From: CodingOnStar <hanxujiang@dify.ai>
Date: Sun, 18 Jan 2026 14:06:55 +0800
Subject: [PATCH] test: enhance type definitions in dataset component tests for
 better type safety

---
 .../index.spec.tsx                                 | 14 ++++++++++++--
 .../detail/embedding/skeleton/index.spec.tsx       |  4 ++--
 .../datasets/list/new-dataset-card/index.spec.tsx  |  9 +++++++--
 .../settings/chunk-structure/index.spec.tsx        |  9 ++++++++-
 4 files changed, 29 insertions(+), 7 deletions(-)

diff --git a/web/app/components/datasets/common/economical-retrieval-method-config/index.spec.tsx b/web/app/components/datasets/common/economical-retrieval-method-config/index.spec.tsx
index 6aafc4ba95..cd6b050336 100644
--- a/web/app/components/datasets/common/economical-retrieval-method-config/index.spec.tsx
+++ b/web/app/components/datasets/common/economical-retrieval-method-config/index.spec.tsx
@@ -4,7 +4,13 @@ import EconomicalRetrievalMethodConfig from './index'
 
 // Mock dependencies
 vi.mock('../../settings/option-card', () => ({
-  default: ({ children, title, description, disabled, id }: any) => (
+  default: ({ children, title, description, disabled, id }: {
+    children?: React.ReactNode
+    title?: string
+    description?: React.ReactNode
+    disabled?: boolean
+    id?: string
+  }) => (
     <div data-testid="option-card" data-title={title} data-id={id} data-disabled={disabled}>
       <div>{description}</div>
       {children}
@@ -13,7 +19,11 @@ vi.mock('../../settings/option-card', () => ({
 }))
 
 vi.mock('../retrieval-param-config', () => ({
-  default: ({ value, onChange, type }: any) => (
+  default: ({ value, onChange, type }: {
+    value: Record<string, unknown>
+    onChange: (value: Record<string, unknown>) => void
+    type?: string
+  }) => (
     <div data-testid="retrieval-param-config" data-type={type}>
       <button onClick={() => onChange({ ...value, newProp: 'changed' })}>
         Change Value
diff --git a/web/app/components/datasets/documents/detail/embedding/skeleton/index.spec.tsx b/web/app/components/datasets/documents/detail/embedding/skeleton/index.spec.tsx
index 64a1892969..e0dc60b668 100644
--- a/web/app/components/datasets/documents/detail/embedding/skeleton/index.spec.tsx
+++ b/web/app/components/datasets/documents/detail/embedding/skeleton/index.spec.tsx
@@ -3,10 +3,10 @@ import EmbeddingSkeleton from './index'
 
 // Mock Skeleton components
 vi.mock('@/app/components/base/skeleton', () => ({
-  SkeletonContainer: ({ children }: any) => <div data-testid="skeleton-container">{children}</div>,
+  SkeletonContainer: ({ children }: { children?: React.ReactNode }) => <div data-testid="skeleton-container">{children}</div>,
   SkeletonPoint: () => <div data-testid="skeleton-point" />,
   SkeletonRectangle: () => <div data-testid="skeleton-rectangle" />,
-  SkeletonRow: ({ children }: any) => <div data-testid="skeleton-row">{children}</div>,
+  SkeletonRow: ({ children }: { children?: React.ReactNode }) => <div data-testid="skeleton-row">{children}</div>,
 }))
 
 // Mock Divider
diff --git a/web/app/components/datasets/list/new-dataset-card/index.spec.tsx b/web/app/components/datasets/list/new-dataset-card/index.spec.tsx
index 2c6191c6f2..e49af629e5 100644
--- a/web/app/components/datasets/list/new-dataset-card/index.spec.tsx
+++ b/web/app/components/datasets/list/new-dataset-card/index.spec.tsx
@@ -1,9 +1,14 @@
 import { render, screen } from '@testing-library/react'
 import NewDatasetCard from './index'
 
+type MockOptionProps = {
+  text: string
+  href: string
+}
+
 // Mock dependencies
 vi.mock('./option', () => ({
-  default: ({ text, href }: any) => (
+  default: ({ text, href }: MockOptionProps) => (
     <a data-testid="option-link" href={href}>
       {text}
     </a>
@@ -16,7 +21,7 @@ vi.mock('@remixicon/react', () => ({
 }))
 
 vi.mock('@/app/components/base/icons/src/vender/solid/development', () => ({
-  ApiConnectionMod: () => <svg data-testid="icon-api" />,
+  // ApiConnectionMod: () => <svg data-testid="icon-api" />,
 }))
 
 describe('NewDatasetCard', () => {
diff --git a/web/app/components/datasets/settings/chunk-structure/index.spec.tsx b/web/app/components/datasets/settings/chunk-structure/index.spec.tsx
index 590b60352d..878018408d 100644
--- a/web/app/components/datasets/settings/chunk-structure/index.spec.tsx
+++ b/web/app/components/datasets/settings/chunk-structure/index.spec.tsx
@@ -2,9 +2,16 @@ import { render, screen } from '@testing-library/react'
 import { ChunkingMode } from '@/models/datasets'
 import ChunkStructure from './index'
 
+type MockOptionCardProps = {
+  id: string
+  title: string
+  isActive?: boolean
+  disabled?: boolean
+}
+
 // Mock dependencies
 vi.mock('../option-card', () => ({
-  default: ({ id, title, isActive, disabled }: any) => (
+  default: ({ id, title, isActive, disabled }: MockOptionCardProps) => (
     <div
       data-testid="option-card"
       data-id={id}