mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2025-12-28 05:11:24 +08:00
test(ActionUtils): 添加对ConversationService和models模块的mock测试
This commit is contained in:
parent
7f643d058c
commit
d62e336dad
@ -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'
|
||||
|
||||
Loading…
Reference in New Issue
Block a user