Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2025-12-26 03:31:24 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

test(services): 更新ApiService测试中的模型配置模拟

Browse Source 
修改测试文件中的模型配置模拟，使用vi.importActual获取原始模块并扩展模拟实现，移除不再使用的SYSTEM_MODELS导入
This commit is contained in:
icarus 2025-08-06 20:23:10 +08:00
parent 176430a31f
commit f670802b65
1 changed files with 24 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/renderer/src/services/__tests__/ApiService.test.ts
View File

@ -15,7 +15,7 @@ import { ApiClientFactory } from '@renderer/aiCore/clients/ApiClientFactory'
import { GeminiAPIClient } from '@renderer/aiCore/clients/gemini/GeminiAPIClient'
import { OpenAIResponseAPIClient } from '@renderer/aiCore/clients/openai/OpenAIResponseAPIClient'
import { GenericChunk } from '@renderer/aiCore/middleware/schemas'
import { isVisionModel, SYSTEM_MODELS } from '@renderer/config/models'
import { isVisionModel } from '@renderer/config/models'
import { Assistant, MCPCallToolResponse, MCPToolResponse, Model, Provider, WebSearchSource } from '@renderer/types'
import {
Chunk,
@ -48,25 +48,29 @@ vi.mock('@renderer/aiCore/clients/ApiClientFactory', () => ({
}))
// Mock the models config
vi.mock('@renderer/config/models', () => ({
isDedicatedImageGenerationModel: vi.fn(() => false),
isTextToImageModel: vi.fn(() => false),
isEmbeddingModel: vi.fn(() => false),
isRerankModel: vi.fn(() => false),
isVisionModel: vi.fn(() => false),
isReasoningModel: vi.fn(() => false),
isWebSearchModel: vi.fn(() => false),
isOpenAIModel: vi.fn(() => false),
isFunctionCallingModel: vi.fn(() => true),
models: {
gemini: {
id: 'gemini-2.5-pro',
name: 'Gemini 2.5 Pro'
}
},
isAnthropicModel: vi.fn(() => false),
SYSTEM_MODELS
}))
vi.mock('@renderer/config/models', async () => {
const origin = await vi.importActual('@renderer/config/models')
return {
...origin,
isDedicatedImageGenerationModel: vi.fn(() => false),
isTextToImageModel: vi.fn(() => false),
isEmbeddingModel: vi.fn(() => false),
isRerankModel: vi.fn(() => false),
isVisionModel: vi.fn(() => false),
isReasoningModel: vi.fn(() => false),
isWebSearchModel: vi.fn(() => false),
isOpenAIModel: vi.fn(() => false),
isFunctionCallingModel: vi.fn(() => true),
models: {
gemini: {
id: 'gemini-2.5-pro',
name: 'Gemini 2.5 Pro'
}
},
isAnthropicModel: vi.fn(() => false)
}
})
// Mock uuid
vi.mock('uuid', () => ({