From 817de3a0ae55c13e63cdc88c9edec2c706d0e769 Mon Sep 17 00:00:00 2001 From: Will Miao Date: Thu, 11 Dec 2025 18:27:55 +0800 Subject: [PATCH] test: improve vi.mock calls to preserve original module exports Updated vi.mock calls in test files to use async importOriginal pattern, ensuring original module exports are preserved while mocking specific functions. This prevents unintended side effects and maintains better test isolation. --- .../components/modelMetadata.renamePath.test.js | 14 +++++++++----- .../components/modelModal.licenseIcons.test.js | 14 +++++++++----- 2 files changed, 18 insertions(+), 10 deletions(-) diff --git a/tests/frontend/components/modelMetadata.renamePath.test.js b/tests/frontend/components/modelMetadata.renamePath.test.js index 01202d35..e5e2f764 100644 --- a/tests/frontend/components/modelMetadata.renamePath.test.js +++ b/tests/frontend/components/modelMetadata.renamePath.test.js @@ -55,11 +55,15 @@ vi.mock(MODEL_TAGS_MODULE, () => ({ setupTagEditMode: vi.fn(), })); -vi.mock(UTILS_MODULE, () => ({ - renderCompactTags: vi.fn(() => ''), - setupTagTooltip: vi.fn(), - formatFileSize: vi.fn(() => '1 MB'), -})); +vi.mock(UTILS_MODULE, async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + renderCompactTags: vi.fn(() => ''), + setupTagTooltip: vi.fn(), + formatFileSize: vi.fn(() => '1 MB'), + }; +}); vi.mock(TRIGGER_WORDS_MODULE, () => ({ renderTriggerWords: vi.fn(() => ''), diff --git a/tests/frontend/components/modelModal.licenseIcons.test.js b/tests/frontend/components/modelModal.licenseIcons.test.js index 355fdd63..dfd1d837 100644 --- a/tests/frontend/components/modelModal.licenseIcons.test.js +++ b/tests/frontend/components/modelModal.licenseIcons.test.js @@ -53,11 +53,15 @@ vi.mock(MODEL_TAGS_MODULE, () => ({ setupTagEditMode: vi.fn(), })); -vi.mock(UTILS_MODULE, () => ({ - renderCompactTags: vi.fn(() => ''), - setupTagTooltip: vi.fn(), - formatFileSize: vi.fn(() => '1 MB'), -})); +vi.mock(UTILS_MODULE, async (importOriginal) => { + const actual = await importOriginal(); + return { + ...actual, + renderCompactTags: vi.fn(() => ''), + setupTagTooltip: vi.fn(), + formatFileSize: vi.fn(() => '1 MB'), + }; +}); vi.mock(TRIGGER_WORDS_MODULE, () => ({ renderTriggerWords: vi.fn(() => ''),