Northern_Lights/cherry-studio
1
0
Fork 0
mirror of https://github.com/CherryHQ/cherry-studio.git synced 2026-01-06 13:19:33 +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

14
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 { GeminiAPIClient } from '@renderer/aiCore/clients/gemini/GeminiAPIClient'
import { OpenAIResponseAPIClient } from '@renderer/aiCore/clients/openai/OpenAIResponseAPIClient' import { OpenAIResponseAPIClient } from '@renderer/aiCore/clients/openai/OpenAIResponseAPIClient'
import { GenericChunk } from '@renderer/aiCore/middleware/schemas' 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 { Assistant, MCPCallToolResponse, MCPToolResponse, Model, Provider, WebSearchSource } from '@renderer/types'
import { import {
Chunk, Chunk,
@ -48,7 +48,11 @@ vi.mock('@renderer/aiCore/clients/ApiClientFactory', () => ({
})) }))
// Mock the models config // Mock the models config
vi.mock('@renderer/config/models', () => ({ vi.mock('@renderer/config/models', async () => {
const origin = await vi.importActual('@renderer/config/models')
return {
...origin,
isDedicatedImageGenerationModel: vi.fn(() => false), isDedicatedImageGenerationModel: vi.fn(() => false),
isTextToImageModel: vi.fn(() => false), isTextToImageModel: vi.fn(() => false),
isEmbeddingModel: vi.fn(() => false), isEmbeddingModel: vi.fn(() => false),
@ -64,9 +68,9 @@ vi.mock('@renderer/config/models', () => ({
name: 'Gemini 2.5 Pro' name: 'Gemini 2.5 Pro'
} }
}, },
isAnthropicModel: vi.fn(() => false), isAnthropicModel: vi.fn(() => false)
SYSTEM_MODELS }
})) })
// Mock uuid // Mock uuid
vi.mock('uuid', () => ({ vi.mock('uuid', () => ({