import { describe, expect, it } from 'vitest';

import { findModelById, getCanonicalModelId, modelIdsMatch } from '@/lib/ai/model-aliases';

describe('model aliases', () => {
  it('canonicalizes GPT-5.6 Sol only for OpenAI', () => {
    expect(getCanonicalModelId('openai', 'gpt-5.6-sol')).toBe('gpt-5.6');
    expect(getCanonicalModelId('openrouter', 'gpt-5.6-sol')).toBe('gpt-5.6-sol');
  });

  it('leaves canonical and unrelated model IDs unchanged', () => {
    expect(getCanonicalModelId('openai', 'gpt-5.6')).toBe('gpt-5.6');
    expect(getCanonicalModelId('openai', 'gpt-5.6-terra')).toBe('gpt-5.6-terra');
  });

  it('matches and finds an alias through its canonical model ID', () => {
    const models = [
      { id: 'gpt-5.6-sol', vision: false },
      { id: 'gpt-5.6', vision: true },
    ];

    expect(modelIdsMatch('openai', 'gpt-5.6', 'gpt-5.6-sol')).toBe(true);
    expect(findModelById('openai', models, 'gpt-5.6-sol')).toBe(models[1]);
    expect(findModelById('openrouter', [models[1]], 'gpt-5.6-sol')).toBeUndefined();
  });
});