mirror of
https://github.com/tiennm99/ccs.git
synced 2026-07-16 22:16:41 +00:00
Tam Nhu Tran
ce1915366d
- Change spawn stdio from piped to 'ignore' preventing buffer deadlock - Replace setInterval with sequential setTimeout polling - Fix TOCTOU in SIGKILL escalation (send directly without probing) - Move CursorConfig interface to types.ts - Change detectProvider default from 'openai' to 'unknown' - Remove redundant removePidFile() when PID is null
100 lines
2.9 KiB
TypeScript
100 lines
2.9 KiB
TypeScript
/**
|
|
* Unit tests for Cursor models module
|
|
*/
|
|
|
|
import { describe, it, expect } from 'bun:test';
|
|
import {
|
|
DEFAULT_CURSOR_MODELS,
|
|
DEFAULT_CURSOR_PORT,
|
|
DEFAULT_CURSOR_MODEL,
|
|
getDefaultModel,
|
|
detectProvider,
|
|
formatModelName,
|
|
fetchModelsFromDaemon,
|
|
} from '../../../src/cursor/cursor-models';
|
|
|
|
describe('DEFAULT_CURSOR_MODELS', () => {
|
|
it('contains models from multiple providers', () => {
|
|
const providers = new Set(DEFAULT_CURSOR_MODELS.map((m) => m.provider));
|
|
expect(providers.has('anthropic')).toBe(true);
|
|
expect(providers.has('openai')).toBe(true);
|
|
expect(providers.has('google')).toBe(true);
|
|
});
|
|
|
|
it('has exactly one default model', () => {
|
|
const defaults = DEFAULT_CURSOR_MODELS.filter((m) => m.isDefault);
|
|
expect(defaults).toHaveLength(1);
|
|
expect(defaults[0].id).toBe(DEFAULT_CURSOR_MODEL);
|
|
});
|
|
});
|
|
|
|
describe('DEFAULT_CURSOR_PORT', () => {
|
|
it('is 4242', () => {
|
|
expect(DEFAULT_CURSOR_PORT).toBe(4242);
|
|
});
|
|
});
|
|
|
|
describe('DEFAULT_CURSOR_MODEL', () => {
|
|
it('is gpt-4.1', () => {
|
|
expect(DEFAULT_CURSOR_MODEL).toBe('gpt-4.1');
|
|
});
|
|
});
|
|
|
|
describe('getDefaultModel', () => {
|
|
it('returns the default model constant', () => {
|
|
expect(getDefaultModel()).toBe(DEFAULT_CURSOR_MODEL);
|
|
});
|
|
});
|
|
|
|
describe('detectProvider', () => {
|
|
it('detects anthropic models', () => {
|
|
expect(detectProvider('claude-sonnet-4')).toBe('anthropic');
|
|
expect(detectProvider('claude-opus-4')).toBe('anthropic');
|
|
});
|
|
|
|
it('detects openai models', () => {
|
|
expect(detectProvider('gpt-4.1')).toBe('openai');
|
|
expect(detectProvider('gpt-5-mini')).toBe('openai');
|
|
expect(detectProvider('o3-mini')).toBe('openai');
|
|
});
|
|
|
|
it('detects o1 and o4 models as openai', () => {
|
|
expect(detectProvider('o1')).toBe('openai');
|
|
expect(detectProvider('o1-preview')).toBe('openai');
|
|
expect(detectProvider('o4-mini')).toBe('openai');
|
|
});
|
|
|
|
it('detects google models', () => {
|
|
expect(detectProvider('gemini-2.5-pro')).toBe('google');
|
|
});
|
|
|
|
it('detects cursor models', () => {
|
|
expect(detectProvider('cursor-small')).toBe('cursor');
|
|
});
|
|
|
|
it('defaults to unknown for unrecognized models', () => {
|
|
expect(detectProvider('unknown-model')).toBe('unknown');
|
|
});
|
|
});
|
|
|
|
describe('formatModelName', () => {
|
|
it('returns catalog name for known models', () => {
|
|
expect(formatModelName('claude-sonnet-4')).toBe('Claude Sonnet 4');
|
|
expect(formatModelName('gpt-4.1')).toBe('GPT-4.1');
|
|
});
|
|
|
|
it('converts kebab-case to title case for unknown models', () => {
|
|
expect(formatModelName('my-custom-model')).toBe('My Custom Model');
|
|
});
|
|
});
|
|
|
|
describe('fetchModelsFromDaemon', () => {
|
|
it('falls back to DEFAULT_CURSOR_MODELS when daemon is unreachable', async () => {
|
|
// Use a port that nothing is listening on
|
|
const unreachablePort = 9999;
|
|
const models = await fetchModelsFromDaemon(unreachablePort);
|
|
|
|
expect(models).toEqual(DEFAULT_CURSOR_MODELS);
|
|
});
|
|
});
|