William Valentin
80 lines
2.4 KiB
TypeScript
80 lines
2.4 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
|
|
import { mkdtempSync, statSync } from 'fs';
|
|
import { tmpdir } from 'os';
|
|
import { join } from 'path';
|
|
|
|
describe('auth/anthropic', () => {
|
|
const originalHome = process.env.HOME;
|
|
const originalEnvKey = process.env.ANTHROPIC_API_KEY;
|
|
const originalEnvToken = process.env.ANTHROPIC_AUTH_TOKEN;
|
|
|
|
let homeDir: string;
|
|
|
|
beforeEach(() => {
|
|
homeDir = mkdtempSync(join(tmpdir(), 'flynn-auth-anthropic-'));
|
|
process.env.HOME = homeDir;
|
|
delete process.env.ANTHROPIC_API_KEY;
|
|
delete process.env.ANTHROPIC_AUTH_TOKEN;
|
|
vi.resetModules();
|
|
});
|
|
|
|
afterEach(() => {
|
|
process.env.HOME = originalHome;
|
|
if (originalEnvKey) {
|
|
process.env.ANTHROPIC_API_KEY = originalEnvKey;
|
|
} else {
|
|
delete process.env.ANTHROPIC_API_KEY;
|
|
}
|
|
|
|
if (originalEnvToken) {
|
|
process.env.ANTHROPIC_AUTH_TOKEN = originalEnvToken;
|
|
} else {
|
|
delete process.env.ANTHROPIC_AUTH_TOKEN;
|
|
}
|
|
});
|
|
|
|
it('stores, loads, and clears Anthropic API key', async () => {
|
|
const mod = await import('./anthropic.js');
|
|
|
|
expect(mod.loadStoredAnthropicAuth()).toBeNull();
|
|
|
|
mod.storeAnthropicAuth('sk-test');
|
|
expect(mod.loadStoredAnthropicAuth()?.api_key).toBe('sk-test');
|
|
|
|
const authFile = join(homeDir, '.config/flynn/auth.json');
|
|
const mode = statSync(authFile).mode & 0o777;
|
|
expect(mode).toBe(0o600);
|
|
|
|
mod.clearAnthropicAuth();
|
|
expect(mod.loadStoredAnthropicAuth()).toBeNull();
|
|
});
|
|
|
|
it('stores, loads, and clears Anthropic auth token', async () => {
|
|
const mod = await import('./anthropic.js');
|
|
|
|
expect(mod.loadStoredAnthropicAuthToken()).toBeNull();
|
|
|
|
mod.storeAnthropicAuthToken('tok-test');
|
|
expect(mod.loadStoredAnthropicAuthToken()).toBe('tok-test');
|
|
|
|
const authFile = join(homeDir, '.config/flynn/auth.json');
|
|
const mode = statSync(authFile).mode & 0o777;
|
|
expect(mode).toBe(0o600);
|
|
|
|
mod.clearAnthropicAuthToken();
|
|
expect(mod.loadStoredAnthropicAuthToken()).toBeNull();
|
|
});
|
|
|
|
it('getAnthropicApiKey prefers environment variable', async () => {
|
|
process.env.ANTHROPIC_API_KEY = 'sk-env';
|
|
const mod = await import('./anthropic.js');
|
|
expect(mod.getAnthropicApiKey()).toBe('sk-env');
|
|
});
|
|
|
|
it('getAnthropicAuthToken prefers environment variable', async () => {
|
|
process.env.ANTHROPIC_AUTH_TOKEN = 'tok-env';
|
|
const mod = await import('./anthropic.js');
|
|
expect(mod.getAnthropicAuthToken()).toBe('tok-env');
|
|
});
|
|
});
|