From d62e336dadd0875f65ef0b11029d1fb4a1073b7c Mon Sep 17 00:00:00 2001 From: icarus Date: Sat, 30 Aug 2025 02:38:02 +0800 Subject: [PATCH] =?UTF-8?q?test(ActionUtils):=20=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E5=AF=B9ConversationService=E5=92=8Cmodels=E6=A8=A1=E5=9D=97?= =?UTF-8?q?=E7=9A=84mock=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../components/__tests__/ActionUtils.test.ts | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts b/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts index d6b34021b8..d48e6cb665 100644 --- a/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts +++ b/src/renderer/src/windows/selection/action/components/__tests__/ActionUtils.test.ts @@ -10,6 +10,14 @@ vi.mock('@renderer/services/ApiService', () => ({ fetchChatCompletion: vi.fn() })) +vi.mock('@renderer/services/ConversationService', () => ({ + ConversationService: class { + static async prepareMessagesForModel() { + // do nothing + } + } +})) + vi.mock('@renderer/services/MessagesService', () => ({ getUserMessage: vi.fn(), getAssistantMessage: vi.fn() @@ -50,6 +58,20 @@ vi.mock('@renderer/utils/messageUtils/create', () => ({ createErrorBlock: vi.fn() })) +vi.mock('@renderer/config/models', () => ({ + SYSTEM_MODELS: { + defaultModel: [ + { id: 'gpt-4', name: 'GPT-4' }, + { id: 'gpt-4', name: 'GPT-4' }, + { id: 'gpt-4', name: 'GPT-4' } + ], + silicon: [], + openai: [], + anthropic: [], + gemini: [] + } +})) + // Import mocked modules import { fetchChatCompletion } from '@renderer/services/ApiService' import { getAssistantMessage, getUserMessage } from '@renderer/services/MessagesService'