From 15c6abc9919a2e5c2661514b1ea936d153cc7dec Mon Sep 17 00:00:00 2001
From: Xubin Ren <52506698+Re-bin@users.noreply.github.com>
Date: Sun, 31 May 2026 15:15:40 +0800
Subject: [PATCH] test(webui): assert code block language fallback

---
 webui/src/tests/code-block.test.tsx             | 4 ++++
 webui/src/tests/markdown-text-renderer.test.tsx | 1 +
 2 files changed, 5 insertions(+)

diff --git a/webui/src/tests/code-block.test.tsx b/webui/src/tests/code-block.test.tsx
index 4347354d7..94f2c4832 100644
--- a/webui/src/tests/code-block.test.tsx
+++ b/webui/src/tests/code-block.test.tsx
@@ -12,13 +12,16 @@ const mockedStyles = vi.hoisted(() => ({
 vi.mock("react-syntax-highlighter/dist/esm/prism-async-light", () => ({
   default: ({
     children,
+    language,
     style,
   }: {
     children: string;
+    language?: string;
     style: Record<string, unknown>;
   }) => (
     <pre
       data-testid="highlighted-code"
+      data-language={language}
       data-theme={style === mockedStyles.dark ? "dark" : "light"}
     >
       <code>{children}</code>
@@ -61,6 +64,7 @@ describe("CodeBlock", () => {
     });
 
     expect(screen.getByTestId("highlighted-code")).toBeInTheDocument();
+    expect(screen.getByTestId("highlighted-code")).toHaveAttribute("data-language", "text");
     expect(screen.getByText("const value = 1;")).toBeInTheDocument();
   });
 
diff --git a/webui/src/tests/markdown-text-renderer.test.tsx b/webui/src/tests/markdown-text-renderer.test.tsx
index 2eb9488f5..b0282b756 100644
--- a/webui/src/tests/markdown-text-renderer.test.tsx
+++ b/webui/src/tests/markdown-text-renderer.test.tsx
@@ -24,6 +24,7 @@ describe("MarkdownTextRenderer", () => {
     );
 
     expect(screen.getByText("code without language")).toBeInTheDocument();
+    expect(screen.getByText("text")).toBeInTheDocument();
     expect(container.querySelectorAll("pre")).toHaveLength(1);
   });