import { describe, expect, it } from 'vitest';
import { createKeyRotator, loadConfiguredKeys, parseKeys, parseKeysJson } from '../src/key-store.js';

describe('key store', () => {
  it('parses JSON keys and prefers the environment variable', async () => {
    expect(parseKeysJson('[" a ","b"]')).toEqual(['a', 'b']);
    expect(() => parseKeysJson('{"key":"a"}')).toThrow();
    await expect(loadConfiguredKeys({ env: { OPENCODE_API_KEYS: '["env-key"]' }, path: '/does-not-exist' })).resolves.toEqual(['env-key']);
  });

  it('parses keys, blank lines, and comments', () => {
    expect(parseKeys(' first \n\nsecond # note\n# ignored\n')).toEqual(['first', 'second']);
  });

  it('rotates keys round-robin', () => {
    const rotator = createKeyRotator(['a', 'b']);
    expect([rotator.next(), rotator.next(), rotator.next()]).toEqual(['a', 'b', 'a']);
  });
});