141 lines
4.1 KiB
TypeScript
141 lines
4.1 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
|
|
import { getModelColorClass, parseModelString } from '../../../src/shared/utils/modelParser';
|
|
|
|
describe('modelParser', () => {
|
|
describe('parseModelString', () => {
|
|
it('should return null for empty string', () => {
|
|
expect(parseModelString('')).toBeNull();
|
|
});
|
|
|
|
it('should return null for undefined', () => {
|
|
expect(parseModelString(undefined)).toBeNull();
|
|
});
|
|
|
|
it('should return null for synthetic model', () => {
|
|
expect(parseModelString('<synthetic>')).toBeNull();
|
|
});
|
|
|
|
it('should return null for non-claude model', () => {
|
|
expect(parseModelString('gpt-4')).toBeNull();
|
|
});
|
|
|
|
// New format tests: claude-{family}-{major}-{minor}-{date}
|
|
it('should parse new format: claude-sonnet-4-5-20250929', () => {
|
|
const result = parseModelString('claude-sonnet-4-5-20250929');
|
|
expect(result).toEqual({
|
|
name: 'sonnet4.5',
|
|
family: 'sonnet',
|
|
majorVersion: 4,
|
|
minorVersion: 5,
|
|
});
|
|
});
|
|
|
|
it('should parse new format without minor version', () => {
|
|
const result = parseModelString('claude-opus-5-20260101');
|
|
expect(result).toEqual({
|
|
name: 'opus5',
|
|
family: 'opus',
|
|
majorVersion: 5,
|
|
minorVersion: null,
|
|
});
|
|
});
|
|
|
|
it('should parse new format without date: claude-opus-4-6', () => {
|
|
const result = parseModelString('claude-opus-4-6');
|
|
expect(result).toEqual({
|
|
name: 'opus4.6',
|
|
family: 'opus',
|
|
majorVersion: 4,
|
|
minorVersion: 6,
|
|
});
|
|
});
|
|
|
|
it('should parse new format: claude-haiku-3-20240307', () => {
|
|
const result = parseModelString('claude-haiku-3-20240307');
|
|
expect(result).toEqual({
|
|
name: 'haiku3',
|
|
family: 'haiku',
|
|
majorVersion: 3,
|
|
minorVersion: null,
|
|
});
|
|
});
|
|
|
|
// Old format tests: claude-{major}[-{minor}]-{family}-{date}
|
|
it('should parse old format: claude-3-opus-20240229', () => {
|
|
const result = parseModelString('claude-3-opus-20240229');
|
|
expect(result).toEqual({
|
|
name: 'opus3',
|
|
family: 'opus',
|
|
majorVersion: 3,
|
|
minorVersion: null,
|
|
});
|
|
});
|
|
|
|
it('should parse old format with minor: claude-3-5-sonnet-20241022', () => {
|
|
const result = parseModelString('claude-3-5-sonnet-20241022');
|
|
expect(result).toEqual({
|
|
name: 'sonnet3.5',
|
|
family: 'sonnet',
|
|
majorVersion: 3,
|
|
minorVersion: 5,
|
|
});
|
|
});
|
|
|
|
it('should handle case insensitivity', () => {
|
|
const result = parseModelString('CLAUDE-SONNET-4-5-20250929');
|
|
expect(result).toEqual({
|
|
name: 'sonnet4.5',
|
|
family: 'sonnet',
|
|
majorVersion: 4,
|
|
minorVersion: 5,
|
|
});
|
|
});
|
|
|
|
it('should handle whitespace', () => {
|
|
const result = parseModelString(' claude-sonnet-4-5-20250929 ');
|
|
expect(result).toEqual({
|
|
name: 'sonnet4.5',
|
|
family: 'sonnet',
|
|
majorVersion: 4,
|
|
minorVersion: 5,
|
|
});
|
|
});
|
|
|
|
it('should return null for invalid format with only two parts', () => {
|
|
expect(parseModelString('claude-sonnet')).toBeNull();
|
|
});
|
|
|
|
it('should handle unknown model families', () => {
|
|
const result = parseModelString('claude-newmodel-4-5-20250929');
|
|
expect(result).toEqual({
|
|
name: 'newmodel4.5',
|
|
family: 'newmodel',
|
|
majorVersion: 4,
|
|
minorVersion: 5,
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('getModelColorClass', () => {
|
|
it('should return color for opus', () => {
|
|
expect(getModelColorClass('opus')).toBe('text-zinc-400');
|
|
});
|
|
|
|
it('should return color for sonnet', () => {
|
|
expect(getModelColorClass('sonnet')).toBe('text-zinc-400');
|
|
});
|
|
|
|
it('should return color for haiku', () => {
|
|
expect(getModelColorClass('haiku')).toBe('text-zinc-400');
|
|
});
|
|
|
|
it('should return default color for unknown family', () => {
|
|
expect(getModelColorClass('unknown')).toBe('text-zinc-500');
|
|
});
|
|
|
|
it('should return default color for arbitrary string', () => {
|
|
expect(getModelColorClass('newmodel')).toBe('text-zinc-500');
|
|
});
|
|
});
|
|
});
|