import { describe, it, expect, vi } from 'vitest' import { providers, getAllProviders, getProvider, safeDiscoverSessions, discoverAllSessions } from '../src/providers/index.js' import type { Provider } from '../src/providers/types.js' function fakeProvider(name: string, discover: Provider['discoverSessions']): Provider { return { name, displayName: name, modelDisplayName: (m: string) => m, toolDisplayName: (t: string) => t, discoverSessions: discover, } as unknown as Provider } describe('provider registry', () => { it('has core providers registered synchronously', () => { expect(providers.map(p => p.name)).toEqual(['claude', 'cline', 'codebuff', 'codex', 'copilot', 'devin', 'droid', 'gemini', 'hermes', 'ibm-bob', 'kilo-code', 'kiro', 'kimi', 'lingtai-tui', 'mistral-vibe', 'mux', 'openclaw', 'open-design', 'pi', 'omp', 'qwen', 'roo-code', 'zerostack', 'grok']) }) it('codebuff tool display names normalize codebuff-native names to canonical set', () => { const codebuff = providers.find(p => p.name === 'codebuff')! expect(codebuff.toolDisplayName('read_files')).toBe('Read') expect(codebuff.toolDisplayName('code_search')).toBe('Grep') expect(codebuff.toolDisplayName('str_replace')).toBe('Edit') expect(codebuff.toolDisplayName('run_terminal_command')).toBe('Bash') expect(codebuff.toolDisplayName('spawn_agents')).toBe('Agent') expect(codebuff.toolDisplayName('write_todos')).toBe('TodoWrite') expect(codebuff.toolDisplayName('unknown_tool')).toBe('unknown_tool') }) it('codebuff model display names cover known agent tiers', () => { const codebuff = providers.find(p => p.name === 'codebuff')! expect(codebuff.modelDisplayName('codebuff')).toBe('Codebuff') expect(codebuff.modelDisplayName('codebuff-base2')).toBe('Codebuff Base 2') expect(codebuff.modelDisplayName('some-future-model')).toBe('some-future-model') }) it('includes sqlite providers after async load', async () => { const all = await getAllProviders() const names = all.map(p => p.name) expect(names).toContain('claude') expect(names).toContain('codex') expect(names).toContain('forge') expect(names).toContain('warp') expect(names.length).toBeGreaterThanOrEqual(2) }) it('forge is available through async provider loading', async () => { const forge = await getProvider('forge') expect(forge).toBeDefined() expect(forge!.name).toBe('forge') }) it('warp model and tool display names are normalized', async () => { const warp = await getProvider('warp') expect(warp).toBeDefined() expect(warp!.modelDisplayName('warp-auto-efficient')).toBe('Warp Auto (efficient)') expect(warp!.modelDisplayName('gpt-5.3-codex')).toBe('GPT-5.3 Codex') expect(warp!.toolDisplayName('run_command')).toBe('Bash') }) it('opencode model display names strip provider prefix', async () => { const all = await getAllProviders() const oc = all.find(p => p.name === 'opencode') if (!oc) return expect(oc.modelDisplayName('anthropic/claude-opus-4-6-20260205')).toBe('Opus 4.6') expect(oc.modelDisplayName('google/gemini-2.5-pro')).toBe('Gemini 2.5 Pro') }) it('opencode tool display names normalize builtins', async () => { const all = await getAllProviders() const oc = all.find(p => p.name === 'opencode') if (!oc) return expect(oc.toolDisplayName('bash')).toBe('Bash') expect(oc.toolDisplayName('edit')).toBe('Edit') expect(oc.toolDisplayName('task')).toBe('Agent') expect(oc.toolDisplayName('unknown_tool')).toBe('unknown_tool') }) it('claude tool display names are identity', () => { const claude = providers.find(p => p.name === 'claude')! expect(claude.toolDisplayName('Bash')).toBe('Bash') expect(claude.toolDisplayName('Read')).toBe('Read') }) it('codex tool display names are normalized', () => { const codex = providers.find(p => p.name === 'codex')! expect(codex.toolDisplayName('exec_command')).toBe('Bash') expect(codex.toolDisplayName('read_file')).toBe('Read') expect(codex.toolDisplayName('write_file')).toBe('Edit') expect(codex.toolDisplayName('spawn_agent')).toBe('Agent') }) it('codex model display names are human-readable', () => { const codex = providers.find(p => p.name === 'codex')! expect(codex.modelDisplayName('gpt-5.4')).toBe('GPT-5.4') expect(codex.modelDisplayName('gpt-5.4-mini')).toBe('GPT-5.4 Mini') expect(codex.modelDisplayName('gpt-5.3-codex')).toBe('GPT-5.3 Codex') expect(codex.modelDisplayName('gpt-5.5')).toBe('GPT-5.5') }) it('claude model display names are human-readable', () => { const claude = providers.find(p => p.name === 'claude')! expect(claude.modelDisplayName('claude-opus-4-6-20260205')).toBe('Opus 4.6') expect(claude.modelDisplayName('claude-sonnet-4-6')).toBe('Sonnet 4.6') }) it('kimi model and tool display names are normalized', () => { const kimi = providers.find(p => p.name === 'kimi')! expect(kimi.modelDisplayName('kimi-auto')).toBe('Kimi (auto)') expect(kimi.modelDisplayName('kimi-k2-thinking-turbo')).toBe('Kimi K2 Thinking Turbo') expect(kimi.toolDisplayName('Shell')).toBe('Bash') expect(kimi.toolDisplayName('WriteFile')).toBe('Write') }) it('lingtai-tui model display names are normalized', () => { const lingtai = providers.find(p => p.name === 'lingtai-tui')! expect(lingtai.displayName).toBe('LingTai TUI') expect(lingtai.modelDisplayName('claude-sonnet-4-6')).toBe('Sonnet 4.6') expect(lingtai.toolDisplayName('custom_tool')).toBe('custom_tool') }) it('cursor model display names handle auto mode', async () => { const all = await getAllProviders() const cursor = all.find(p => p.name === 'cursor')! expect(cursor.modelDisplayName('cursor-auto')).toBe('Cursor (auto)') expect(cursor.modelDisplayName('claude-4.5-opus-high-thinking')).toBe('Opus 4.5 (Thinking)') expect(cursor.modelDisplayName('grok-code-fast-1')).toBe('Grok Code Fast') expect(cursor.modelDisplayName('unknown-model')).toBe('unknown-model') }) describe('provider-discovery isolation', () => { it('safeDiscoverSessions returns [] and warns once instead of propagating', async () => { const warn = vi.spyOn(process.stderr, 'write').mockReturnValue(true) const boom = fakeProvider('boom-helper', async () => { throw new Error('crafted file blew up') }) try { await expect(safeDiscoverSessions(boom)).resolves.toEqual([]) expect(warn.mock.calls.length).toBeGreaterThanOrEqual(1) expect(String(warn.mock.calls[0]![0])).toContain('boom-helper') // Deduped on repeat within the same run: no additional warning. const afterFirst = warn.mock.calls.length await safeDiscoverSessions(boom) expect(warn.mock.calls.length).toBe(afterFirst) } finally { warn.mockRestore() } }) it('discoverAllSessions drops a throwing provider but keeps the healthy ones', async () => { const warn = vi.spyOn(process.stderr, 'write').mockReturnValue(true) const boom = fakeProvider('boom-loop', async () => { throw new Error('kaboom') }) const ok1 = fakeProvider('ok1', async () => [{ path: '/a.jsonl', project: 'p1', provider: 'ok1' }]) const ok2 = fakeProvider('ok2', async () => [{ path: '/b.jsonl', project: 'p2', provider: 'ok2' }]) try { // A throwing provider in the middle must not abort the loop. const sources = await discoverAllSessions('all', [ok1, boom, ok2]) expect(sources.map(s => s.path)).toEqual(['/a.jsonl', '/b.jsonl']) expect(warn.mock.calls.some(c => String(c[0]).includes('boom-loop'))).toBe(true) } finally { warn.mockRestore() } }) it('discoverAllSessions honors the provider filter', async () => { const ok1 = fakeProvider('keep', async () => [{ path: '/keep.jsonl', project: 'k', provider: 'keep' }]) const ok2 = fakeProvider('drop', async () => [{ path: '/drop.jsonl', project: 'd', provider: 'drop' }]) const sources = await discoverAllSessions('keep', [ok1, ok2]) expect(sources.map(s => s.path)).toEqual(['/keep.jsonl']) }) }) })