versun/claude-code
1
0
Fork 0
mirror of https://github.com/claude-code-best/claude-code.git synced 2026-06-21 15:55:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 增强 providers 测试的环境变量隔离，防止 mock 污染

Browse Source
This commit is contained in:
claude-code-best
2026-06-11 16:15:27 +08:00
parent 6942f5393f
commit 2a1f5697cc
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/utils/model/__tests__/providers.test.ts
View File

@@ -1,4 +1,5 @@
import { describe, expect, test, beforeEach, afterEach } from 'bun:test' import { describe, expect, test, beforeEach, afterEach } from 'bun:test'
import { isEnvTruthy } from '../../envUtils.js'
const { getAPIProvider, isFirstPartyAnthropicBaseUrl } = await import( const { getAPIProvider, isFirstPartyAnthropicBaseUrl } = await import(
'../providers' '../providers'
@@ -12,11 +13,12 @@ describe('getAPIProvider', () => {
'CLAUDE_CODE_USE_FOUNDRY', 'CLAUDE_CODE_USE_FOUNDRY',
'CLAUDE_CODE_USE_OPENAI', 'CLAUDE_CODE_USE_OPENAI',
'CLAUDE_CODE_USE_GROK', 'CLAUDE_CODE_USE_GROK',
'OPENAI_BASE_URL',
'GEMINI_BASE_URL',
] as const ] as const
const savedEnv: Record<string, string | undefined> = {} const savedEnv: Record<string, string | undefined> = {}
beforeEach(() => { beforeEach(() => {
// Save and clear environment variables
for (const key of envKeys) { for (const key of envKeys) {
savedEnv[key] = process.env[key] savedEnv[key] = process.env[key]
delete process.env[key] delete process.env[key]
@@ -24,7 +26,6 @@ describe('getAPIProvider', () => {
}) })
afterEach(() => { afterEach(() => {
// Restore environment variables
for (const key of envKeys) { for (const key of envKeys) {
if (savedEnv[key] !== undefined) { if (savedEnv[key] !== undefined) {
process.env[key] = savedEnv[key] process.env[key] = savedEnv[key]
@@ -67,6 +68,16 @@ describe('getAPIProvider', () => {
expect(getAPIProvider({})).toBe('foundry') expect(getAPIProvider({})).toBe('foundry')
}) })
test('returns "openai" when CLAUDE_CODE_USE_OPENAI is set', () => {
process.env.CLAUDE_CODE_USE_OPENAI = '1'
expect(getAPIProvider({})).toBe('openai')
})
test('returns "grok" when CLAUDE_CODE_USE_GROK is set', () => {
process.env.CLAUDE_CODE_USE_GROK = '1'
expect(getAPIProvider({})).toBe('grok')
})
test('bedrock takes precedence over gemini', () => { test('bedrock takes precedence over gemini', () => {
process.env.CLAUDE_CODE_USE_BEDROCK = '1' process.env.CLAUDE_CODE_USE_BEDROCK = '1'
process.env.CLAUDE_CODE_USE_GEMINI = '1' process.env.CLAUDE_CODE_USE_GEMINI = '1'
@@ -88,16 +99,19 @@ describe('getAPIProvider', () => {
test('"true" is truthy', () => { test('"true" is truthy', () => {
process.env.CLAUDE_CODE_USE_BEDROCK = 'true' process.env.CLAUDE_CODE_USE_BEDROCK = 'true'
expect(isEnvTruthy(process.env.CLAUDE_CODE_USE_BEDROCK)).toBe(true)
expect(getAPIProvider({})).toBe('bedrock') expect(getAPIProvider({})).toBe('bedrock')
}) })
test('"0" is not truthy', () => { test('"0" is not truthy', () => {
process.env.CLAUDE_CODE_USE_BEDROCK = '0' process.env.CLAUDE_CODE_USE_BEDROCK = '0'
expect(isEnvTruthy(process.env.CLAUDE_CODE_USE_BEDROCK)).toBe(false)
expect(getAPIProvider({})).toBe('firstParty') expect(getAPIProvider({})).toBe('firstParty')
}) })
test('empty string is not truthy', () => { test('empty string is not truthy', () => {
process.env.CLAUDE_CODE_USE_BEDROCK = '' process.env.CLAUDE_CODE_USE_BEDROCK = ''
expect(isEnvTruthy(process.env.CLAUDE_CODE_USE_BEDROCK)).toBe(false)
expect(getAPIProvider({})).toBe('firstParty') expect(getAPIProvider({})).toBe('firstParty')
}) })
}) })