Skip to content

Commit 0dc0a9a

Browse files
committed
fix: tests
1 parent bfe671d commit 0dc0a9a

File tree

1 file changed

+16
-15
lines changed

1 file changed

+16
-15
lines changed

web-app/src/services/__tests__/providers.test.ts

Lines changed: 16 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import {
55
updateSettings,
66
} from '../providers'
77
import { models as providerModels } from 'token.js'
8-
import { predefinedProviders } from '@/mock/data'
8+
import { predefinedProviders } from '@/consts/providers'
99
import { EngineManager } from '@janhq/core'
1010
import { fetchModels } from '../models'
1111
import { ExtensionManager } from '@/lib/extension'
@@ -21,7 +21,7 @@ vi.mock('token.js', () => ({
2121
},
2222
}))
2323

24-
vi.mock('@/mock/data', () => ({
24+
vi.mock('@/consts/providers', () => ({
2525
predefinedProviders: [
2626
{
2727
active: true,
@@ -69,6 +69,7 @@ vi.mock('../models', () => ({
6969
{ id: 'llama-2-7b', name: 'Llama 2 7B', description: 'Llama model' },
7070
])
7171
),
72+
isToolSupported: vi.fn(() => Promise.resolve(false)),
7273
}))
7374

7475
vi.mock('@/lib/extension', () => ({
@@ -116,7 +117,7 @@ describe('providers service', () => {
116117
it('should return builtin and runtime providers', async () => {
117118
const providers = await getProviders()
118119

119-
expect(providers).toHaveLength(9) // 8 runtime + 1 builtin
120+
expect(providers).toHaveLength(2) // 1 runtime + 1 builtin (mocked)
120121
expect(providers.some((p) => p.provider === 'llamacpp')).toBe(true)
121122
expect(providers.some((p) => p.provider === 'openai')).toBe(true)
122123
})
@@ -156,7 +157,7 @@ describe('providers service', () => {
156157
provider: 'openai',
157158
base_url: 'https://api.openai.com/v1',
158159
api_key: 'test-key',
159-
} as ModelProvider
160+
}
160161

161162
const models = await fetchModelsFromProvider(provider)
162163

@@ -185,7 +186,7 @@ describe('providers service', () => {
185186
provider: 'custom',
186187
base_url: 'https://api.custom.com',
187188
api_key: '',
188-
} as ModelProvider
189+
}
189190

190191
const models = await fetchModelsFromProvider(provider)
191192

@@ -204,7 +205,7 @@ describe('providers service', () => {
204205
const provider = {
205206
provider: 'custom',
206207
base_url: 'https://api.custom.com',
207-
} as ModelProvider
208+
}
208209

209210
const models = await fetchModelsFromProvider(provider)
210211

@@ -214,7 +215,7 @@ describe('providers service', () => {
214215
it('should throw error when provider has no base_url', async () => {
215216
const provider = {
216217
provider: 'custom',
217-
} as ModelProvider
218+
}
218219

219220
await expect(fetchModelsFromProvider(provider)).rejects.toThrow(
220221
'Provider must have base_url configured'
@@ -232,10 +233,10 @@ describe('providers service', () => {
232233
const provider = {
233234
provider: 'custom',
234235
base_url: 'https://api.custom.com',
235-
} as ModelProvider
236+
}
236237

237238
await expect(fetchModelsFromProvider(provider)).rejects.toThrow(
238-
'Cannot connect to custom at https://api.custom.com'
239+
'Cannot connect to custom at https://api.custom.com. Please check that the service is running and accessible.'
239240
)
240241
})
241242

@@ -245,10 +246,10 @@ describe('providers service', () => {
245246
const provider = {
246247
provider: 'custom',
247248
base_url: 'https://api.custom.com',
248-
} as ModelProvider
249+
}
249250

250251
await expect(fetchModelsFromProvider(provider)).rejects.toThrow(
251-
'Cannot connect to custom at https://api.custom.com'
252+
'Cannot connect to custom at https://api.custom.com. Please check that the service is running and accessible.'
252253
)
253254
})
254255

@@ -264,7 +265,7 @@ describe('providers service', () => {
264265
const provider = {
265266
provider: 'custom',
266267
base_url: 'https://api.custom.com',
267-
} as ModelProvider
268+
}
268269

269270
const models = await fetchModelsFromProvider(provider)
270271

@@ -298,7 +299,7 @@ describe('providers service', () => {
298299
controller_type: 'input',
299300
controller_props: { value: 'test-key' },
300301
},
301-
] as ProviderSetting[]
302+
]
302303

303304
await updateSettings('openai', settings)
304305

@@ -324,7 +325,7 @@ describe('providers service', () => {
324325
mockExtensionManager
325326
)
326327

327-
const settings = [] as ProviderSetting[]
328+
const settings = []
328329

329330
const result = await updateSettings('nonexistent', settings)
330331

@@ -350,7 +351,7 @@ describe('providers service', () => {
350351
controller_type: 'input',
351352
controller_props: { value: undefined },
352353
},
353-
] as ProviderSetting[]
354+
]
354355

355356
await updateSettings('openai', settings)
356357

0 commit comments

Comments
 (0)