gstack/test/setup-codex-model.test.ts

import { describe, expect, test } from 'bun:test';
import * as fs from 'fs';
import * as path from 'path';

const ROOT = path.resolve(import.meta.dir, '..');
const setup = fs.readFileSync(path.join(ROOT, 'setup'), 'utf8');

describe('setup Codex model activation', () => {
  test('exposes --model and limits it to Codex installs', () => {
    expect(setup).toContain('--model <id>');
    expect(setup).toContain('MODEL_OVERRIDE_SET=1');
    expect(setup).toContain('--model is supported only when Codex is selected');
    // The override reaches the resolver as QUOTED argv — an unquoted
    // regression would word-split/glob user input.
    expect(setup).toContain('--explicit "$MODEL_OVERRIDE"');
  });

  test('resolver runs on every setup, before any INSTALL_CODEX gate', () => {
    // A plain `./setup` (claude host) still regenerates .agents/, and live
    // ~/.codex/skills symlinks point into it — resolution must not be gated
    // on the codex host being selected, or a Sol user's profile gets
    // clobbered back to the hardcoded fallback.
    const blockStart = setup.indexOf('# Resolve the model overlay');
    const blockEnd = setup.indexOf('# 1. Build browse binary', blockStart);
    expect(blockStart).toBeGreaterThan(-1);
    const block = setup.slice(blockStart, blockEnd);
    const resolverAt = block.indexOf('_CODEX_MODEL_OUTPUT=');
    const firstGateAt = block.indexOf('INSTALL_CODEX');
    expect(resolverAt).toBeGreaterThan(-1);
    expect(firstGateAt === -1 || resolverAt < firstGateAt).toBe(true);
  });

  test('resolves the profile once, fails closed, and passes it as quoted argv', () => {
    expect(setup).toContain('scripts/resolve-codex-generation-model.ts');
    expect(setup).toContain('CODEX_GENERATION_MODEL="gpt-6-astra"');
    expect(setup).toContain('Codex skill profile: $CODEX_GENERATION_MODEL');
    expect(setup).toContain('Source: $CODEX_GENERATION_MODEL_SOURCE');
    expect(setup).toContain('gen:skill-docs --host codex --model "$CODEX_GENERATION_MODEL"');
    // Positive pin of the parse mechanism (an eval-shaped regression would
    // remove this line rather than merely rephrase an eval call).
    expect(setup).toContain(`IFS=$'\\t' read -r CODEX_GENERATION_MODEL CODEX_GENERATION_MODEL_SOURCE`);
    // Fail-closed: empty resolver output aborts setup, including the exit.
    const guardAt = setup.indexOf('gstack setup failed: Codex model resolver returned no model');
    expect(guardAt).toBeGreaterThan(-1);
    expect(setup.slice(guardAt, guardAt + 200)).toContain('exit 1');
  });

  test('regenerates Codex after both fresh and stale build paths', () => {
    const generationStart = setup.indexOf('# 1b. Generate .agents/ Codex skill docs');
    const generationEnd = setup.indexOf('# 1c. Generate .factory/', generationStart);
    const block = setup.slice(generationStart, generationEnd);
    expect(block).toContain('if [ "$NEEDS_AGENTS_GEN" -eq 1 ]; then');
    expect(block).not.toContain('NEEDS_BUILD" -eq 0');
  });

  test('fallback generation and handoff preserve the selected profile', () => {
    const linkStart = setup.indexOf('link_codex_skill_dirs()');
    const linkEnd = setup.indexOf('create_agents_sidecar()', linkStart);
    const block = setup.slice(linkStart, linkEnd);
    expect(block).toContain('gen:skill-docs --host codex --model "$CODEX_GENERATION_MODEL"');
    expect(block).toContain('gen:skill-docs --host codex --model $CODEX_GENERATION_MODEL');
    expect(setup).toContain('model changes: rerun ./setup --host codex');
    expect(setup).toContain('model profile: $CODEX_GENERATION_MODEL');
  });

  test('Kiro uses its own host render without rewriting the Codex profile', () => {
    const kiroStart = setup.indexOf('# 6. Install for Kiro CLI');
    const kiroEnd = setup.indexOf('# 6b.', kiroStart);
    expect(kiroStart).toBeGreaterThan(-1);
    const block = setup.slice(kiroStart, kiroEnd);
    expect(block).toContain('gen:skill-docs --host kiro');
    expect(block).toContain('KIRO_DIR="$SOURCE_GSTACK_DIR/.kiro/skills"');
    expect(block).not.toContain('gen:skill-docs --host codex');
    expect(block).not.toContain('CODEX_GENERATION_MODEL');
  });

  test('Kiro installs its native root, upgrade skill, and bin/lib runtime assets', () => {
    const kiroStart = setup.indexOf('# 6. Install for Kiro CLI');
    const kiroEnd = setup.indexOf('# 6b.', kiroStart);
    const block = setup.slice(kiroStart, kiroEnd);
    expect(block).toContain('_copy_skill_md "$KIRO_DIR/gstack-upgrade/SKILL.md"');
    expect(block).toContain('_copy_skill_md "$KIRO_DIR/gstack/SKILL.md"');
    expect(block).toContain('_link_or_copy "$SOURCE_GSTACK_DIR/bin" "$KIRO_GSTACK/bin"');
    expect(block).toContain('_link_or_copy "$SOURCE_GSTACK_DIR/lib" "$KIRO_GSTACK/lib"');
    expect(block).not.toContain('$AGENTS_DIR');
  });

  test('Codex skills path honors CODEX_HOME', () => {
    expect(setup).toContain('CODEX_SKILLS="${CODEX_HOME:-$HOME/.codex}/skills"');
  });

  test('--model prints the one-shot persistence note', () => {
    expect(setup).toContain('--model applies to this run only');
  });
});

describe('Codex E2E hermetic model pin', () => {
  const runner = fs.readFileSync(path.join(ROOT, 'test', 'helpers', 'codex-session-runner.ts'), 'utf8');

  test('copies authentication only and can ignore operator config', () => {
    expect(runner).toContain("for (const entry of ['auth.json'])");
    expect(runner).toContain("if (ignoreUserConfig) args.push('--ignore-user-config')");
    expect(runner).toContain('CODEX_HOME: tempCodexDir');
    expect(runner).not.toContain("if (entry === 'skills') continue");
  });
});