136 lines
3.8 KiB
TypeScript
136 lines
3.8 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from 'vitest'
|
|
import { testProviderConnection } from '@/lib/user-api/provider-test'
|
|
|
|
const fetchMock = vi.hoisted(() =>
|
|
vi.fn(async (input: unknown) => {
|
|
const url = String(input)
|
|
if (url.includes('dashscope.aliyuncs.com/compatible-mode/v1/models')) {
|
|
return new Response(JSON.stringify({ data: [{ id: 'qwen-plus' }] }), { status: 200 })
|
|
}
|
|
if (url.includes('api.siliconflow.cn/v1/models')) {
|
|
return new Response(JSON.stringify({ data: [{ id: 'Qwen/Qwen3-32B' }] }), { status: 200 })
|
|
}
|
|
if (url.includes('api.siliconflow.cn/v1/user/info')) {
|
|
return new Response(JSON.stringify({ data: { balance: '12.3000' } }), { status: 200 })
|
|
}
|
|
return new Response('not-found', { status: 404 })
|
|
}),
|
|
)
|
|
|
|
describe('provider test connection', () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks()
|
|
vi.stubGlobal('fetch', fetchMock)
|
|
})
|
|
|
|
it('passes bailian probe with models step and credits skip', async () => {
|
|
const result = await testProviderConnection({
|
|
apiType: 'bailian',
|
|
apiKey: 'bl-key',
|
|
})
|
|
|
|
expect(result.success).toBe(true)
|
|
expect(result.steps).toEqual([
|
|
{
|
|
name: 'models',
|
|
status: 'pass',
|
|
message: 'Found 1 models',
|
|
},
|
|
{
|
|
name: 'credits',
|
|
status: 'skip',
|
|
message: 'Not supported by Bailian probe API',
|
|
},
|
|
])
|
|
})
|
|
|
|
it('passes siliconflow probe with models and credits steps', async () => {
|
|
const result = await testProviderConnection({
|
|
apiType: 'siliconflow',
|
|
apiKey: 'sf-key',
|
|
})
|
|
|
|
expect(result.success).toBe(true)
|
|
expect(result.steps[0]).toEqual({
|
|
name: 'models',
|
|
status: 'pass',
|
|
message: 'Found 1 models',
|
|
})
|
|
expect(result.steps[1]).toEqual({
|
|
name: 'credits',
|
|
status: 'pass',
|
|
message: 'Balance: 12.3000',
|
|
})
|
|
})
|
|
|
|
it('classifies auth failures for bailian models probe', async () => {
|
|
fetchMock.mockImplementationOnce(async () => new Response('unauthorized', { status: 401 }))
|
|
|
|
const result = await testProviderConnection({
|
|
apiType: 'bailian',
|
|
apiKey: 'bad-key',
|
|
})
|
|
|
|
expect(result.success).toBe(false)
|
|
expect(result.steps[0]).toEqual({
|
|
name: 'models',
|
|
status: 'fail',
|
|
message: 'Authentication failed (401)',
|
|
detail: 'unauthorized',
|
|
})
|
|
expect(result.steps[1]).toEqual({
|
|
name: 'credits',
|
|
status: 'skip',
|
|
message: 'Not supported by Bailian probe API',
|
|
})
|
|
})
|
|
|
|
it('classifies rate limit failures for siliconflow models probe', async () => {
|
|
fetchMock.mockImplementationOnce(async () => new Response('rate limit', { status: 429 }))
|
|
|
|
const result = await testProviderConnection({
|
|
apiType: 'siliconflow',
|
|
apiKey: 'sf-key',
|
|
})
|
|
|
|
expect(result.success).toBe(false)
|
|
expect(result.steps[0]).toEqual({
|
|
name: 'models',
|
|
status: 'fail',
|
|
message: 'Rate limited (429)',
|
|
detail: 'rate limit',
|
|
})
|
|
expect(result.steps[1]).toEqual({
|
|
name: 'credits',
|
|
status: 'skip',
|
|
message: 'Skipped because model probe failed',
|
|
})
|
|
})
|
|
|
|
it('classifies network failures for siliconflow user info probe', async () => {
|
|
fetchMock.mockImplementationOnce(async () =>
|
|
new Response(JSON.stringify({ data: [{ id: 'Qwen/Qwen3-32B' }] }), { status: 200 }),
|
|
)
|
|
fetchMock.mockImplementationOnce(async () => {
|
|
throw new Error('socket hang up')
|
|
})
|
|
|
|
const result = await testProviderConnection({
|
|
apiType: 'siliconflow',
|
|
apiKey: 'sf-key',
|
|
})
|
|
|
|
expect(result.success).toBe(false)
|
|
expect(result.steps[0]).toEqual({
|
|
name: 'models',
|
|
status: 'pass',
|
|
message: 'Found 1 models',
|
|
})
|
|
expect(result.steps[1]).toEqual({
|
|
name: 'credits',
|
|
status: 'fail',
|
|
message: 'Network error: socket hang up',
|
|
})
|
|
})
|
|
})
|