File size: 2,800 Bytes
1f21206 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | import { afterEach, beforeEach, describe, expect, it } from 'bun:test'
import * as fs from 'fs/promises'
import * as os from 'os'
import * as path from 'path'
import {
DEFAULT_AI_REQUEST_TIMEOUT_MS,
MAX_AI_REQUEST_TIMEOUT_MS,
MIN_AI_REQUEST_TIMEOUT_MS,
getManualNetworkProxyUrl,
buildNetworkEnvironment,
loadNetworkSettings,
normalizeNetworkSettings,
} from '../services/networkSettings.js'
import { resetSettingsCache } from '../../utils/settings/settingsCache.js'
let tmpDir: string
let originalConfigDir: string | undefined
async function setup() {
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'network-settings-test-'))
originalConfigDir = process.env.CLAUDE_CONFIG_DIR
process.env.CLAUDE_CONFIG_DIR = tmpDir
resetSettingsCache()
}
async function teardown() {
if (originalConfigDir !== undefined) {
process.env.CLAUDE_CONFIG_DIR = originalConfigDir
} else {
delete process.env.CLAUDE_CONFIG_DIR
}
resetSettingsCache()
await fs.rm(tmpDir, { recursive: true, force: true })
}
describe('network settings', () => {
beforeEach(setup)
afterEach(teardown)
it('normalizes missing settings to the 120s system-proxy default', () => {
expect(normalizeNetworkSettings({})).toEqual({
aiRequestTimeoutMs: DEFAULT_AI_REQUEST_TIMEOUT_MS,
proxy: {
mode: 'system',
url: '',
},
})
})
it('clamps AI request timeouts and trims manual proxy URLs', () => {
expect(normalizeNetworkSettings({
network: {
aiRequestTimeoutMs: 999_999,
proxy: {
mode: 'manual',
url: ' http://127.0.0.1:7890 ',
},
},
})).toEqual({
aiRequestTimeoutMs: MAX_AI_REQUEST_TIMEOUT_MS,
proxy: {
mode: 'manual',
url: 'http://127.0.0.1:7890',
},
})
expect(normalizeNetworkSettings({
network: {
aiRequestTimeoutMs: 100,
},
}).aiRequestTimeoutMs).toBe(MIN_AI_REQUEST_TIMEOUT_MS)
})
it('loads persisted user network settings for provider requests', async () => {
await fs.writeFile(
path.join(tmpDir, 'settings.json'),
JSON.stringify({
network: {
aiRequestTimeoutMs: 180_000,
proxy: {
mode: 'manual',
url: ' http://127.0.0.1:7890 ',
},
},
}),
'utf-8',
)
const settings = await loadNetworkSettings()
expect(settings.aiRequestTimeoutMs).toBe(180_000)
expect(getManualNetworkProxyUrl(settings)).toBe('http://127.0.0.1:7890')
expect(buildNetworkEnvironment(settings)).toEqual({
API_TIMEOUT_MS: '180000',
HTTP_PROXY: 'http://127.0.0.1:7890',
HTTPS_PROXY: 'http://127.0.0.1:7890',
http_proxy: 'http://127.0.0.1:7890',
https_proxy: 'http://127.0.0.1:7890',
})
})
})
|