From b8a8f19892a4622db08b2a94187e070465732a0b Mon Sep 17 00:00:00 2001
From: icarus <eurfelux@gmail.com>
Date: Wed, 22 Oct 2025 07:26:08 +0800
Subject: [PATCH] test: update snapshots and add mock for ui components

Update test snapshots to reflect changes in component styling and structure
Add mock for @cherrystudio/ui components in CodeBlock tests
---
 .../InputEmbeddingDimension.test.tsx.snap     |  4 ++--
 .../__snapshots__/Spinner.test.tsx.snap       |  1 -
 .../Markdown/__tests__/CodeBlock.test.tsx     |  6 +++++
 .../__snapshots__/CodeBlock.test.tsx.snap     | 22 +++++++++++--------
 4 files changed, 21 insertions(+), 12 deletions(-)

diff --git a/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap b/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap
index 4d9a94f58d..f8ddaa407d 100644
--- a/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap
+++ b/src/renderer/src/components/__tests__/__snapshots__/InputEmbeddingDimension.test.tsx.snap
@@ -8,7 +8,7 @@ exports[`InputEmbeddingDimension > basic rendering > should match snapshot with
   <input
     data-testid="input-number"
     placeholder="请输入维度大小"
-    style="flex: 1 1 0%;"
+    style="flex: 1;"
     type="number"
     value="1536"
   />
@@ -43,7 +43,7 @@ exports[`InputEmbeddingDimension > basic rendering > should match snapshot with
   <input
     data-testid="input-number"
     placeholder="请输入维度大小"
-    style="flex: 1 1 0%;"
+    style="flex: 1;"
     type="number"
     value=""
   />
diff --git a/src/renderer/src/components/__tests__/__snapshots__/Spinner.test.tsx.snap b/src/renderer/src/components/__tests__/__snapshots__/Spinner.test.tsx.snap
index 00ecf953ec..c05a945d24 100644
--- a/src/renderer/src/components/__tests__/__snapshots__/Spinner.test.tsx.snap
+++ b/src/renderer/src/components/__tests__/__snapshots__/Spinner.test.tsx.snap
@@ -24,7 +24,6 @@ exports[`Spinner > should match snapshot 1`] = `
     stroke-linecap="round"
     stroke-linejoin="round"
     stroke-width="2"
-    style="color: unset;"
     viewBox="0 0 24 24"
     width="16"
     xmlns="http://www.w3.org/2000/svg"
diff --git a/src/renderer/src/pages/home/Markdown/__tests__/CodeBlock.test.tsx b/src/renderer/src/pages/home/Markdown/__tests__/CodeBlock.test.tsx
index 9ada24b829..4250850c51 100644
--- a/src/renderer/src/pages/home/Markdown/__tests__/CodeBlock.test.tsx
+++ b/src/renderer/src/pages/home/Markdown/__tests__/CodeBlock.test.tsx
@@ -58,6 +58,12 @@ vi.mock('@renderer/hooks/useSettings', () => ({
   useSettings: () => mocks.useSettings()
 }))
 
+vi.mock('@cherrystudio/ui', () => ({
+  cn: vi.fn((...classes) => classes.filter(Boolean).join(' ')),
+  Spinner: vi.fn(() => <div data-testid="spinner" />),
+  BlockingOverlay: vi.fn(({ children }) => <div>{children}</div>)
+}))
+
 vi.mock('@renderer/components/CodeBlockView', () => ({
   CodeBlockView: mocks.CodeBlockView,
   HtmlArtifactsCard: mocks.HtmlArtifactsCard
diff --git a/src/renderer/src/pages/home/Markdown/__tests__/__snapshots__/CodeBlock.test.tsx.snap b/src/renderer/src/pages/home/Markdown/__tests__/__snapshots__/CodeBlock.test.tsx.snap
index 3e8dabb857..c5ca0feaec 100644
--- a/src/renderer/src/pages/home/Markdown/__tests__/__snapshots__/CodeBlock.test.tsx.snap
+++ b/src/renderer/src/pages/home/Markdown/__tests__/__snapshots__/CodeBlock.test.tsx.snap
@@ -2,15 +2,19 @@
 
 exports[`CodeBlock > rendering > should render a snapshot 1`] = `
 <div>
-  <div>
-    <code>
-      console.log("hello world")
-    </code>
-    <button
-      type="button"
-    >
-      Save
-    </button>
+  <div
+    class="relative"
+  >
+    <div>
+      <code>
+        console.log("hello world")
+      </code>
+      <button
+        type="button"
+      >
+        Save
+      </button>
+    </div>
   </div>
 </div>
 `;