diff --git a/tests/server/provider-update-controller.test.ts b/tests/server/provider-update-controller.test.ts new file mode 100644 index 0000000..957da80 --- /dev/null +++ b/tests/server/provider-update-controller.test.ts @@ -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, 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') + }) +})