mirror of
https://github.com/Wei-Shaw/claude-relay-service.git
synced 2026-04-25 02:28:39 +00:00
fix: use selected model in claude-console connectivity test payload
This commit is contained in:
70
tests/claudeConsoleRelayService.test.js
Normal file
70
tests/claudeConsoleRelayService.test.js
Normal file
@@ -0,0 +1,70 @@
|
||||
jest.mock('../src/utils/logger', () => ({
|
||||
debug: jest.fn(),
|
||||
info: jest.fn(),
|
||||
warn: jest.fn(),
|
||||
error: jest.fn()
|
||||
}))
|
||||
|
||||
jest.mock('../src/services/account/claudeConsoleAccountService', () => ({
|
||||
getAccount: jest.fn(),
|
||||
_createProxyAgent: jest.fn()
|
||||
}))
|
||||
|
||||
jest.mock('/Users/mike/projects/claude-relay-service/config/config', () => ({}), {
|
||||
virtual: true
|
||||
})
|
||||
jest.mock('../src/models/redis', () => ({}))
|
||||
|
||||
jest.mock('../src/utils/testPayloadHelper', () => ({
|
||||
createClaudeTestPayload: jest.fn(),
|
||||
sendStreamTestRequest: jest.fn()
|
||||
}))
|
||||
|
||||
const claudeConsoleRelayService = require('../src/services/relay/claudeConsoleRelayService')
|
||||
const claudeConsoleAccountService = require('../src/services/account/claudeConsoleAccountService')
|
||||
const {
|
||||
createClaudeTestPayload,
|
||||
sendStreamTestRequest
|
||||
} = require('../src/utils/testPayloadHelper')
|
||||
|
||||
describe('claudeConsoleRelayService.testAccountConnection', () => {
|
||||
beforeEach(() => {
|
||||
jest.clearAllMocks()
|
||||
})
|
||||
|
||||
it('passes selected model stream payload to sendStreamTestRequest', async () => {
|
||||
claudeConsoleAccountService.getAccount.mockResolvedValue({
|
||||
name: 'Console A1',
|
||||
apiUrl: 'https://console.example.com',
|
||||
apiKey: 'test-key',
|
||||
proxy: null,
|
||||
userAgent: null
|
||||
})
|
||||
claudeConsoleAccountService._createProxyAgent.mockReturnValue(undefined)
|
||||
|
||||
const payload = {
|
||||
model: 'claude-sonnet-4-6',
|
||||
stream: true
|
||||
}
|
||||
createClaudeTestPayload.mockReturnValue(payload)
|
||||
sendStreamTestRequest.mockResolvedValue(undefined)
|
||||
|
||||
const res = {}
|
||||
await claudeConsoleRelayService.testAccountConnection('a1', res, 'claude-sonnet-4-6')
|
||||
|
||||
expect(createClaudeTestPayload).toHaveBeenCalledWith('claude-sonnet-4-6', { stream: true })
|
||||
expect(sendStreamTestRequest).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
payload
|
||||
})
|
||||
)
|
||||
expect(sendStreamTestRequest).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
payload: expect.objectContaining({
|
||||
model: 'claude-sonnet-4-6',
|
||||
stream: true
|
||||
})
|
||||
})
|
||||
)
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user