test: improve coverage for some test files (#32916)

Signed-off-by: edvatar <88481784+toroleapinc@users.noreply.github.com>
Signed-off-by: -LAN- <laipz8200@outlook.com>
Signed-off-by: dependabot[bot] <support@github.com>
Signed-off-by: majiayu000 <1835304752@qq.com>
Co-authored-by: Poojan <poojan@infocusp.com>
Co-authored-by: sahil-infocusp <73810410+sahil-infocusp@users.noreply.github.com>
Co-authored-by: 非法操作 <hjlarry@163.com>
Co-authored-by: Pandaaaa906 <ye.pandaaaa906@gmail.com>
Co-authored-by: Asuka Minato <i@asukaminato.eu.org>
Co-authored-by: heyszt <270985384@qq.com>
Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Ijas <ijas.ahmd.ap@gmail.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Co-authored-by: 木之本澪 <kinomotomiovo@gmail.com>
Co-authored-by: KinomotoMio <200703522+KinomotoMio@users.noreply.github.com>
Co-authored-by: 不做了睡大觉 <64798754+stakeswky@users.noreply.github.com>
Co-authored-by: User <user@example.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: edvatar <88481784+toroleapinc@users.noreply.github.com>
Co-authored-by: -LAN- <laipz8200@outlook.com>
Co-authored-by: Leilei <138381132+Inlei@users.noreply.github.com>
Co-authored-by: HaKu <104669497+haku-ink@users.noreply.github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: wangxiaolei <fatelei@gmail.com>
Co-authored-by: Varun Chawla <34209028+veeceey@users.noreply.github.com>
Co-authored-by: Stephen Zhou <38493346+hyoban@users.noreply.github.com>
Co-authored-by: yyh <yuanyouhuilyz@gmail.com>
Co-authored-by: yyh <92089059+lyzno1@users.noreply.github.com>
Co-authored-by: tda <95275462+tda1017@users.noreply.github.com>
Co-authored-by: root <root@DESKTOP-KQLO90N>
Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
Co-authored-by: Niels Kaspers <153818647+nielskaspers@users.noreply.github.com>
Co-authored-by: hj24 <mambahj24@gmail.com>
Co-authored-by: Tyson Cung <45380903+tysoncung@users.noreply.github.com>
Co-authored-by: Stephen Zhou <hi@hyoban.cc>
Co-authored-by: FFXN <31929997+FFXN@users.noreply.github.com>
Co-authored-by: slegarraga <64795732+slegarraga@users.noreply.github.com>
Co-authored-by: 99 <wh2099@pm.me>
Co-authored-by: Br1an <932039080@qq.com>
Co-authored-by: L1nSn0w <l1nsn0w@qq.com>
Co-authored-by: Yunlu Wen <yunlu.wen@dify.ai>
Co-authored-by: akkoaya <151345394+akkoaya@users.noreply.github.com>
Co-authored-by: 盐粒 Yanli <yanli@dify.ai>
Co-authored-by: lif <1835304752@qq.com>
Co-authored-by: weiguang li <codingpunk@gmail.com>
Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
Co-authored-by: HanWenbo <124024253+hwb96@users.noreply.github.com>
Co-authored-by: Coding On Star <447357187@qq.com>
Co-authored-by: CodingOnStar <hanxujiang@dify.com>
Co-authored-by: Stable Genius <stablegenius043@gmail.com>
Co-authored-by: Stable Genius <259448942+stablegenius49@users.noreply.github.com>
Co-authored-by: ふるい <46769295+Echo0ff@users.noreply.github.com>
Co-authored-by: Xiyuan Chen <52963600+GareArc@users.noreply.github.com>

web/app/components/base/markdown/__tests__/markdown-utils.spec.ts

@@ -1,6 +1,3 @@
-// app/components/base/markdown/preprocess.spec.ts
-import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
-
 /**
  * Helper to (re)load the module with a mocked config value.
  * We need to reset modules because the tested module imports

web/app/components/base/markdown/__tests__/react-markdown-wrapper.spec.tsx

@@ -8,9 +8,9 @@ vi.mock('@/app/components/base/markdown-blocks', () => ({
   Link: ({ children, href }: { children?: ReactNode, href?: string }) => <a href={href}>{children}</a>,
   MarkdownButton: ({ children }: PropsWithChildren) => <button>{children}</button>,
   MarkdownForm: ({ children }: PropsWithChildren) => <form>{children}</form>,
-  Paragraph: ({ children }: PropsWithChildren) => <p>{children}</p>,
+  Paragraph: ({ children }: PropsWithChildren) => <p data-testid="paragraph">{children}</p>,
   PluginImg: ({ alt }: { alt?: string }) => <span data-testid="plugin-img">{alt}</span>,
-  PluginParagraph: ({ children }: PropsWithChildren) => <p>{children}</p>,
+  PluginParagraph: ({ children }: PropsWithChildren) => <p data-testid="plugin-paragraph">{children}</p>,
   ScriptBlock: () => null,
   ThinkBlock: ({ children }: PropsWithChildren) => <details>{children}</details>,
   VideoBlock: ({ children }: PropsWithChildren) => <div data-testid="video-block">{children}</div>,
@@ -105,5 +105,85 @@ describe('ReactMarkdownWrapper', () => {
       expect(screen.getByText('italic text')).toBeInTheDocument()
       expect(document.querySelector('em')).not.toBeNull()
     })
+
+    it('should render standard Image component when pluginInfo is not provided', () => {
+      // Act
+      render(<ReactMarkdownWrapper latexContent="![standard-img](https://example.com/img.png)" />)
+
+      // Assert
+      expect(screen.getByTestId('img')).toBeInTheDocument()
+    })
+
+    it('should render a CodeBlock component for code markdown', async () => {
+      // Arrange
+      const content = '```javascript\nconsole.log("hello")\n```'
+
+      // Act
+      render(<ReactMarkdownWrapper latexContent={content} />)
+
+      // Assert
+      // We mocked code block to return <code>{children}</code>
+      const codeElement = await screen.findByText('console.log("hello")')
+      expect(codeElement).toBeInTheDocument()
+    })
+  })
+
+  describe('Plugin Info behavior', () => {
+    it('should render PluginImg and PluginParagraph when pluginInfo is provided', () => {
+      // Arrange
+      const content = 'This is a plugin paragraph\n\n![plugin-img](https://example.com/plugin.png)'
+      const pluginInfo = { pluginUniqueIdentifier: 'test-plugin', pluginId: 'plugin-1' }
+
+      // Act
+      render(<ReactMarkdownWrapper latexContent={content} pluginInfo={pluginInfo} />)
+
+      // Assert
+      expect(screen.getByTestId('plugin-img')).toBeInTheDocument()
+      expect(screen.queryByTestId('img')).toBeNull()
+
+      expect(screen.getAllByTestId('plugin-paragraph').length).toBeGreaterThan(0)
+      expect(screen.queryByTestId('paragraph')).toBeNull()
+    })
+  })
+
+  describe('Custom elements configuration', () => {
+    it('should use customComponents if provided', () => {
+      // Arrange
+      const customComponents = {
+        a: ({ children }: PropsWithChildren) => <a data-testid="custom-link">{children}</a>,
+      }
+
+      // Act
+      render(<ReactMarkdownWrapper latexContent="[link](https://example.com)" customComponents={customComponents} />)
+
+      // Assert
+      expect(screen.getByTestId('custom-link')).toBeInTheDocument()
+    })
+
+    it('should disallow customDisallowedElements', () => {
+      // Act - disallow strong (which is usually **bold**)
+      render(<ReactMarkdownWrapper latexContent="**bold**" customDisallowedElements={['strong']} />)
+
+      // Assert - strong element shouldn't be rendered (it will be stripped out)
+      expect(document.querySelector('strong')).toBeNull()
+    })
+  })
+
+  describe('Rehype AST modification', () => {
+    it('should remove ref attributes from elements', () => {
+      // Act
+      render(<ReactMarkdownWrapper latexContent={'<div ref="someRef">content</div>'} />)
+
+      // Assert - If ref isn't stripped, it gets passed to React DOM causing warnings, but here we just ensure content renders
+      expect(screen.getByText('content')).toBeInTheDocument()
+    })
+
+    it('should convert invalid tag names to text nodes', () => {
+      // Act - <custom-element> is invalid because it contains a hyphen
+      render(<ReactMarkdownWrapper latexContent="<custom-element>content</custom-element>" />)
+
+      // Assert - The AST node is changed to text with value `<custom-element`
+      expect(screen.getByText(/<custom-element/)).toBeInTheDocument()
+    })
   })
 })