mirror of
https://github.com/CherryHQ/cherry-studio.git
synced 2026-01-08 22:39:36 +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()
|
fetchChatCompletion: vi.fn()
|
||||||
}))
|
}))
|
||||||
|
|
||||||
|
vi.mock('@renderer/services/ConversationService', () => ({
|
||||||
|
ConversationService: class {
|
||||||
|
static async prepareMessagesForModel() {
|
||||||
|
// do nothing
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}))
|
||||||
|
|
||||||
vi.mock('@renderer/services/MessagesService', () => ({
|
vi.mock('@renderer/services/MessagesService', () => ({
|
||||||
getUserMessage: vi.fn(),
|
getUserMessage: vi.fn(),
|
||||||
getAssistantMessage: vi.fn()
|
getAssistantMessage: vi.fn()
|
||||||
@ -50,6 +58,20 @@ vi.mock('@renderer/utils/messageUtils/create', () => ({
|
|||||||
createErrorBlock: vi.fn()
|
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 mocked modules
|
||||||
import { fetchChatCompletion } from '@renderer/services/ApiService'
|
import { fetchChatCompletion } from '@renderer/services/ApiService'
|
||||||
import { getAssistantMessage, getUserMessage } from '@renderer/services/MessagesService'
|
import { getAssistantMessage, getUserMessage } from '@renderer/services/MessagesService'
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user