hapi/shared/src/cursorCliSku.test.ts

import { describe, expect, it } from 'vitest';
import {
    cursorCliSkuBaseId,
    findBestCliSkuForAcpWire,
    isCursorAcpWireModelId,
    matchCliSkuToAcpWireId
} from './cursorCliSku';

describe('cursorCliSkuBaseId', () => {
    it('strips effort/speed suffixes from CLI skus', () => {
        expect(cursorCliSkuBaseId('gpt-5.5-high-fast')).toBe('gpt-5.5');
        expect(cursorCliSkuBaseId('composer-2.5-fast')).toBe('composer-2.5');
        expect(cursorCliSkuBaseId('gpt-5.3-codex-xhigh-fast')).toBe('gpt-5.3-codex');
    });

    it('keeps wire base ids unchanged', () => {
        expect(cursorCliSkuBaseId('composer-2.5[fast=true]')).toBe('composer-2.5');
    });
});

describe('matchCliSkuToAcpWireId', () => {
    const available = [
        { modelId: 'composer-2.5[fast=true]' },
        { modelId: 'composer-2.5[fast=false]' },
        { modelId: 'gpt-5.5[context=272k,reasoning=medium,fast=false]' }
    ];

    it('returns exact wire matches', () => {
        expect(matchCliSkuToAcpWireId('composer-2.5[fast=false]', available)).toBe('composer-2.5[fast=false]');
    });

    it('maps CLI skus onto the matching ACP wire for the same base', () => {
        expect(matchCliSkuToAcpWireId('composer-2.5-fast', available)).toBe('composer-2.5[fast=true]');
        expect(matchCliSkuToAcpWireId('gpt-5.5-medium', available)).toBe('gpt-5.5[context=272k,reasoning=medium,fast=false]');
    });

    it('maps base-only SKU to fast=false when fast variants exist (cursor CLI convention)', () => {
        expect(matchCliSkuToAcpWireId('composer-2.5', available)).toBe('composer-2.5[fast=false]');
    });

    it('still maps base-only SKU when only one variant exists', () => {
        expect(matchCliSkuToAcpWireId('composer-2.5', [{ modelId: 'composer-2.5[fast=true]' }])).toBe(
            'composer-2.5[fast=true]'
        );
    });
});

describe('findBestCliSkuForAcpWire', () => {
    it('picks the sku that best matches wire params, not the first partial match', () => {
        const wire = 'gpt-5.5[context=272k,reasoning=medium,fast=false]';
        const best = findBestCliSkuForAcpWire(wire, [
            'gpt-5.5-high-fast',
            'gpt-5.5-medium',
            'gpt-5.5-low'
        ]);
        expect(best).toBe('gpt-5.5-medium');
    });

    it('prefers base-only sku for fast=false wire over -fast sku', () => {
        const wire = 'composer-2.5[fast=false]';
        const best = findBestCliSkuForAcpWire(wire, ['composer-2.5', 'composer-2.5-fast']);
        expect(best).toBe('composer-2.5');
    });

    it('prefers -fast sku for fast=true wire over base-only sku', () => {
        const wire = 'composer-2.5[fast=true]';
        const best = findBestCliSkuForAcpWire(wire, ['composer-2.5', 'composer-2.5-fast']);
        expect(best).toBe('composer-2.5-fast');
    });
});

describe('round-trip (regression for #883: "selected but no response")', () => {
    const acpWires = [
        { modelId: 'composer-2.5[fast=true]' },
        { modelId: 'composer-2.5[fast=false]' }
    ];
    const pickerSkus = ['composer-2.5', 'composer-2.5-fast'];

    function simulateRoundTrip(clickedSku: string): { sessionModel: string; radioOn: string | null } {
        // CLI side: applyCursorAcpModel → resolveCursorAcpWireId → matchCliSkuToAcpWireId
        const sessionModel = matchCliSkuToAcpWireId(clickedSku, acpWires);
        if (!sessionModel) {
            throw new Error('CLI rejected sku');
        }
        // Web side after refetch: cursorVariantSelectValue uses findBestCliSkuForAcpWire
        const radioOn = findBestCliSkuForAcpWire(sessionModel, pickerSkus);
        return { sessionModel, radioOn };
    }

    it('clicking composer-2.5 (slow) lands on the slow radio, not the fast one', () => {
        const result = simulateRoundTrip('composer-2.5');
        expect(result.sessionModel).toBe('composer-2.5[fast=false]');
        expect(result.radioOn).toBe('composer-2.5');
    });

    it('clicking composer-2.5-fast lands on the fast radio', () => {
        const result = simulateRoundTrip('composer-2.5-fast');
        expect(result.sessionModel).toBe('composer-2.5[fast=true]');
        expect(result.radioOn).toBe('composer-2.5-fast');
    });

    it('clicking each picker option lands on a distinct session model (no collapse)', () => {
        const slow = simulateRoundTrip('composer-2.5').sessionModel;
        const fast = simulateRoundTrip('composer-2.5-fast').sessionModel;
        expect(slow).not.toBe(fast);
    });
});

describe('isCursorAcpWireModelId', () => {
    it('detects wire ids', () => {
        expect(isCursorAcpWireModelId('gpt-5.5[fast=false]')).toBe(true);
        expect(isCursorAcpWireModelId('gpt-5.5-high-fast')).toBe(false);
    });
});