test: refactor mock implementation in markdown component tests (#33350)

Co-authored-by: CodingOnStar <hanxujiang@dify.com>
This commit is contained in:
Coding On Star 2026-03-12 15:50:07 +08:00 committed by GitHub
parent dc50e4c4f2
commit ceaa399351
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 7 deletions

View File

@ -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
},
}))