ccs/tests/unit/proxy/proxy-daemon-state.test.ts

import { afterEach, beforeEach, describe, expect, it } from 'bun:test';
import * as fs from 'fs';
import * as os from 'os';
import * as path from 'path';
import {
  getLegacyOpenAICompatProxySessionPath,
  getOpenAICompatProxyPidPath,
  getOpenAICompatProxySessionPath,
} from '../../../src/proxy/proxy-daemon-paths';
import { listOpenAICompatProxyProfileNames } from '../../../src/proxy/proxy-daemon-state';

let originalCcsHome: string | undefined;
let tempDir: string;

beforeEach(() => {
  originalCcsHome = process.env.CCS_HOME;
  tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'ccs-proxy-state-'));
  process.env.CCS_HOME = tempDir;
});

afterEach(() => {
  if (originalCcsHome !== undefined) {
    process.env.CCS_HOME = originalCcsHome;
  } else {
    delete process.env.CCS_HOME;
  }
  fs.rmSync(tempDir, { recursive: true, force: true });
});

describe('listOpenAICompatProxyProfileNames', () => {
  it('ignores the legacy singleton session file', () => {
    fs.mkdirSync(path.dirname(getLegacyOpenAICompatProxySessionPath()), { recursive: true });
    fs.writeFileSync(getLegacyOpenAICompatProxySessionPath(), '{}\n', 'utf8');
    fs.writeFileSync(getOpenAICompatProxySessionPath('ccg'), '{}\n', 'utf8');

    expect(listOpenAICompatProxyProfileNames()).toEqual(['ccg']);
  });

  it('skips malformed percent-encoded profile keys', () => {
    fs.mkdirSync(path.dirname(getLegacyOpenAICompatProxySessionPath()), { recursive: true });
    fs.writeFileSync(path.join(path.dirname(getLegacyOpenAICompatProxySessionPath()), '%E0%A4%.session.json'), '{}\n', 'utf8');
    fs.writeFileSync(getOpenAICompatProxySessionPath('ccg'), '{}\n', 'utf8');

    expect(listOpenAICompatProxyProfileNames()).toEqual(['ccg']);
  });

  it('includes pid-only profile state in the discovered profile list', () => {
    fs.mkdirSync(path.dirname(getLegacyOpenAICompatProxySessionPath()), { recursive: true });
    fs.writeFileSync(getOpenAICompatProxyPidPath('ccg'), '123\n', 'utf8');

    expect(listOpenAICompatProxyProfileNames()).toEqual(['ccg']);
  });
});