Test provider API key profile scoping

2026-05-24 08:31:55 +08:00
parent 65a984c31a
commit 4f8bda9836
1 changed files with 103 additions and 0 deletions
@@ -0,0 +1,103 @@
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
+import { mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from 'fs'
+import { join } from 'path'
+import { tmpdir } from 'os'
+import YAML from 'js-yaml'
+
+vi.mock('../../packages/server/src/services/hermes/hermes-cli', () => ({
+  restartGateway: vi.fn().mockResolvedValue(undefined),
+}))
+
+let hermesHome = ''
+
+async function loadProvidersController() {
+  vi.resetModules()
+  process.env.HERMES_HOME = hermesHome
+  return import('../../packages/server/src/controllers/hermes/providers')
+}
+
+function makeCtx(poolKey: string, body: Record<string, any>, profile = 'research') {
+  return {
+    params: { poolKey: encodeURIComponent(poolKey) },
+    request: { body },
+    state: { profile: { name: profile } },
+    status: 200,
+    body: undefined as unknown,
+  }
+}
+
+function readYaml(filePath: string) {
+  return YAML.load(readFileSync(filePath, 'utf-8')) as any
+}
+
+describe('providers controller update', () => {
+  beforeEach(() => {
+    hermesHome = mkdtempSync(join(tmpdir(), 'hwui-provider-update-'))
+    mkdirSync(join(hermesHome, 'profiles', 'research'), { recursive: true })
+    writeFileSync(join(hermesHome, 'config.yaml'), 'model:\n  provider: deepseek\n  default: keep-default-model\n')
+    writeFileSync(join(hermesHome, '.env'), [
+      'DEEPSEEK_API_KEY=keep-default-key',
+      '',
+    ].join('\n'))
+    writeFileSync(join(hermesHome, 'profiles', 'research', 'config.yaml'), [
+      'model:',
+      '  provider: custom:research-proxy',
+      '  default: research-model',
+      'custom_providers:',
+      '  - name: research-proxy',
+      '    base_url: https://research.invalid/v1',
+      '    api_key: old-research-custom-key',
+      '    model: research-model',
+      '',
+    ].join('\n'))
+    writeFileSync(join(hermesHome, 'profiles', 'research', '.env'), [
+      'DEEPSEEK_API_KEY=old-research-key',
+      '',
+    ].join('\n'))
+  })
+
+  afterEach(() => {
+    delete process.env.HERMES_HOME
+    vi.doUnmock('../../packages/server/src/controllers/hermes/providers')
+    vi.clearAllMocks()
+    if (hermesHome) rmSync(hermesHome, { recursive: true, force: true })
+    hermesHome = ''
+  })
+
+  it('updates built-in provider API keys in the request-scoped profile env only', async () => {
+    const { update } = await loadProvidersController()
+    const ctx = makeCtx('deepseek', { api_key: 'new-research-key' })
+
+    await update(ctx)
+
+    expect(ctx.body).toEqual({ success: true })
+    expect(readFileSync(join(hermesHome, '.env'), 'utf-8')).toContain('DEEPSEEK_API_KEY=keep-default-key')
+    expect(readFileSync(join(hermesHome, 'profiles', 'research', '.env'), 'utf-8')).toContain('DEEPSEEK_API_KEY=new-research-key')
+  })
+
+  it('updates custom provider API keys in the request-scoped profile config only', async () => {
+    const defaultConfigPath = join(hermesHome, 'config.yaml')
+    writeFileSync(defaultConfigPath, [
+      'model:',
+      '  provider: custom:research-proxy',
+      '  default: default-model',
+      'custom_providers:',
+      '  - name: research-proxy',
+      '    base_url: https://default.invalid/v1',
+      '    api_key: keep-default-custom-key',
+      '    model: default-model',
+      '',
+    ].join('\n'))
+
+    const { update } = await loadProvidersController()
+    const ctx = makeCtx('custom:research-proxy', { api_key: 'new-research-custom-key' })
+
+    await update(ctx)
+
+    expect(ctx.body).toEqual({ success: true })
+    const defaultConfig = readYaml(defaultConfigPath)
+    const researchConfig = readYaml(join(hermesHome, 'profiles', 'research', 'config.yaml'))
+    expect(defaultConfig.custom_providers[0].api_key).toBe('keep-default-custom-key')
+    expect(researchConfig.custom_providers[0].api_key).toBe('new-research-custom-key')
+  })
+})