From ceaa399351009401721582c4f7e1f7f2bd682cbc Mon Sep 17 00:00:00 2001
From: Coding On Star <447357187@qq.com>
Date: Thu, 12 Mar 2026 15:50:07 +0800
Subject: [PATCH] test: refactor mock implementation in markdown component
 tests (#33350)

Co-authored-by: CodingOnStar <hanxujiang@dify.com>
---
 .../base/markdown/__tests__/index.spec.tsx           | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/web/app/components/base/markdown/__tests__/index.spec.tsx b/web/app/components/base/markdown/__tests__/index.spec.tsx
index 4d98fcba2e..5d0261b074 100644
--- a/web/app/components/base/markdown/__tests__/index.spec.tsx
+++ b/web/app/components/base/markdown/__tests__/index.spec.tsx
@@ -7,17 +7,15 @@ const { mockReactMarkdownWrapper } = vi.hoisted(() => ({
   mockReactMarkdownWrapper: vi.fn(),
 }))
 
-vi.mock('../react-markdown-wrapper', () => ({
-  ReactMarkdownWrapper: () => null,
-}))
-
 vi.mock('next/dynamic', () => ({
-  default: (loader: () => Promise<unknown>) => {
-    void loader()
-    return (props: { latexContent: string }) => {
+  default: () => {
+    const MockStreamdownWrapper = (props: { latexContent: string }) => {
       mockReactMarkdownWrapper(props)
       return <div data-testid="react-markdown-wrapper">{props.latexContent}</div>
     }
+
+    MockStreamdownWrapper.displayName = 'MockStreamdownWrapper'
+    return MockStreamdownWrapper
   },
 }))