aryx/tests/renderer/toolCallSummary.test.ts

import { describe, expect, test } from 'bun:test';

import {
  formatToolCallSummary,
  formatToolArgumentValue,
  getDisplayableArguments,
  formatToolCallPrimaryLabel,
  formatToolGroupLabel,
  extractToolCallSnippet,
} from '@renderer/lib/toolCallSummary';

describe('formatToolCallSummary', () => {
  test('returns undefined when toolName is missing', () => {
    expect(formatToolCallSummary(undefined, { command: 'ls' })).toBeUndefined();
  });

  test('returns undefined when toolArguments is missing', () => {
    expect(formatToolCallSummary('powershell', undefined)).toBeUndefined();
  });

  test('returns undefined when toolArguments is empty', () => {
    expect(formatToolCallSummary('powershell', {})).toBeUndefined();
  });

  test('extracts command for powershell', () => {
    expect(formatToolCallSummary('powershell', { command: 'git status' })).toBe('git status');
  });

  test('truncates long powershell commands', () => {
    const longCommand = 'a'.repeat(100);
    const result = formatToolCallSummary('powershell', { command: longCommand });
    expect(result!.length).toBeLessThanOrEqual(81); // 80 + ellipsis
    expect(result!.endsWith('…')).toBe(true);
  });

  test('extracts path for view tool', () => {
    expect(formatToolCallSummary('view', { path: '/src/index.ts' })).toBe('/src/index.ts');
  });

  test('includes view range when present', () => {
    expect(formatToolCallSummary('view', { path: '/src/index.ts', view_range: [10, 25] }))
      .toBe('/src/index.ts:10-25');
  });

  test('supports viewRange camelCase variant', () => {
    expect(formatToolCallSummary('view', { path: '/src/index.ts', viewRange: [1, 50] }))
      .toBe('/src/index.ts:1-50');
  });

  test('extracts path for edit tool', () => {
    expect(formatToolCallSummary('edit', { path: '/src/utils.ts', old_str: 'foo' }))
      .toBe('/src/utils.ts');
  });

  test('extracts path for create tool', () => {
    expect(formatToolCallSummary('create', { path: '/new-file.ts', file_text: 'content' }))
      .toBe('/new-file.ts');
  });

  test('extracts pattern for grep', () => {
    expect(formatToolCallSummary('grep', { pattern: 'TODO', path: '/src' })).toBe('TODO');
  });

  test('extracts pattern for glob', () => {
    expect(formatToolCallSummary('glob', { pattern: '**/*.ts' })).toBe('**/*.ts');
  });

  test('extracts operation and file for lsp', () => {
    expect(formatToolCallSummary('lsp', { operation: 'goToDefinition', file: '/src/app.ts' }))
      .toBe('goToDefinition /src/app.ts');
  });

  test('extracts just operation when file is missing for lsp', () => {
    expect(formatToolCallSummary('lsp', { operation: 'workspaceSymbol', query: 'Foo' }))
      .toBe('workspaceSymbol');
  });

  test('extracts url for web_fetch', () => {
    expect(formatToolCallSummary('web_fetch', { url: 'https://example.com' }))
      .toBe('https://example.com');
  });

  test('extracts description for sql', () => {
    expect(formatToolCallSummary('sql', { description: 'Insert todos', query: 'INSERT ...' }))
      .toBe('Insert todos');
  });

  test('extracts description for task', () => {
    expect(formatToolCallSummary('task', { description: 'Run tests' })).toBe('Run tests');
  });

  test('extracts question for ask_user', () => {
    expect(formatToolCallSummary('ask_user', { question: 'Which database?' })).toBe('Which database?');
  });

  test('extracts intent for report_intent', () => {
    expect(formatToolCallSummary('report_intent', { intent: 'Exploring codebase' }))
      .toBe('Exploring codebase');
  });

  test('summarizes github tools with query', () => {
    expect(formatToolCallSummary('github-search_code', { query: 'FunctionCallContent' }))
      .toBe('FunctionCallContent');
  });

  test('summarizes github tools with owner/repo', () => {
    expect(formatToolCallSummary('github-get_file_contents', { owner: 'octocat', repo: 'hello-world' }))
      .toBe('octocat/hello-world');
  });

  test('falls back to first string value for unknown tools', () => {
    expect(formatToolCallSummary('unknown_tool', { target: 'production', count: 42 }))
      .toBe('production');
  });

  test('skips [truncated] values in fallback', () => {
    expect(formatToolCallSummary('unknown_tool', { data: '[truncated]', label: 'test' }))
      .toBe('test');
  });

  test('uses first line only for multiline commands', () => {
    const result = formatToolCallSummary('powershell', { command: 'echo hello\necho world' });
    expect(result).toBe('echo hello');
  });
});

describe('formatToolArgumentValue', () => {
  test('formats string values directly', () => {
    expect(formatToolArgumentValue('hello')).toBe('hello');
  });

  test('formats numbers', () => {
    expect(formatToolArgumentValue(42)).toBe('42');
  });

  test('formats booleans', () => {
    expect(formatToolArgumentValue(true)).toBe('true');
  });

  test('formats null as empty string', () => {
    expect(formatToolArgumentValue(null)).toBe('');
  });

  test('formats objects as pretty JSON', () => {
    const result = formatToolArgumentValue({ a: 1, b: 'two' });
    expect(result).toContain('"a": 1');
    expect(result).toContain('"b": "two"');
  });

  test('formats arrays as pretty JSON', () => {
    const result = formatToolArgumentValue([1, 2, 3]);
    expect(result).toContain('1');
    expect(result).toContain('3');
  });
});

describe('getDisplayableArguments', () => {
  test('returns empty array when toolArguments is undefined', () => {
    expect(getDisplayableArguments(undefined)).toEqual([]);
  });

  test('filters out null and undefined values', () => {
    const result = getDisplayableArguments({ a: 'value', b: null, c: undefined, d: 'ok' });
    expect(result).toEqual([['a', 'value'], ['d', 'ok']]);
  });

  test('filters out empty string values', () => {
    const result = getDisplayableArguments({ a: '', b: 'value' });
    expect(result).toEqual([['b', 'value']]);
  });

  test('filters out description key', () => {
    const result = getDisplayableArguments({ description: 'Insert todos', query: 'INSERT ...' });
    expect(result).toEqual([['query', 'INSERT ...']]);
  });
});

/* ── formatToolCallPrimaryLabel ────────────────────────────── */

describe('formatToolCallPrimaryLabel', () => {
  test('produces verb-based label for view with path', () => {
    expect(formatToolCallPrimaryLabel('view', { path: '/src/components/ChatPane.tsx' }))
      .toBe('Viewed `ChatPane.tsx`');
  });

  test('includes view range', () => {
    expect(formatToolCallPrimaryLabel('view', { path: '/src/index.ts', view_range: [10, 25] }))
      .toBe('Viewed `index.ts:10-25`');
  });

  test('falls back to tool name when view has no args', () => {
    expect(formatToolCallPrimaryLabel('view', {})).toBe('view');
  });

  test('produces verb-based label for edit', () => {
    expect(formatToolCallPrimaryLabel('edit', { path: '/src/utils.ts', old_str: 'foo' }))
      .toBe('Edited `utils.ts`');
  });

  test('produces verb-based label for create', () => {
    expect(formatToolCallPrimaryLabel('create', { path: '/new-file.ts' }))
      .toBe('Created `new-file.ts`');
  });

  test('produces verb-based label for grep', () => {
    expect(formatToolCallPrimaryLabel('grep', { pattern: 'TODO', path: '/src' }))
      .toBe('Searched for `TODO`');
  });

  test('produces verb-based label for powershell', () => {
    expect(formatToolCallPrimaryLabel('powershell', { command: 'npm run build' }))
      .toBe('Ran `npm run build`');
  });

  test('produces verb-based label for task', () => {
    expect(formatToolCallPrimaryLabel('task', { description: 'Explore codebase' }))
      .toBe('Launched agent: Explore codebase');
  });

  test('produces verb-based label for web_fetch with hostname', () => {
    expect(formatToolCallPrimaryLabel('web_fetch', { url: 'https://example.com/page' }))
      .toBe('Fetched `example.com`');
  });

  test('produces verb-based label for sql', () => {
    expect(formatToolCallPrimaryLabel('sql', { description: 'Insert todos', query: 'INSERT ...' }))
      .toBe('SQL: Insert todos');
  });

  test('uses fallback string arg when specific key is missing', () => {
    expect(formatToolCallPrimaryLabel('view', { file_path: '/src/foo.ts' }))
      .toBe('Viewed `/src/foo.ts`');
  });

  test('handles GitHub tools', () => {
    const result = formatToolCallPrimaryLabel('github-mcp-server-search_code', { query: 'auth' });
    expect(result).toContain('search code');
    expect(result).toContain('auth');
  });

  test('falls back to "Used <tool>" for unknown tools', () => {
    expect(formatToolCallPrimaryLabel('custom_tool', { data: 'test' }))
      .toBe('Used custom_tool: test');
  });

  test('returns "Tool call" for undefined toolName', () => {
    expect(formatToolCallPrimaryLabel(undefined, {})).toBe('Tool call');
  });
});

/* ── formatToolGroupLabel ─────────────────────────────────── */

describe('formatToolGroupLabel', () => {
  test('pluralizes correctly for view', () => {
    expect(formatToolGroupLabel('view', 1)).toBe('Viewed 1 file');
    expect(formatToolGroupLabel('view', 4)).toBe('Viewed 4 files');
  });

  test('pluralizes correctly for grep', () => {
    expect(formatToolGroupLabel('grep', 1)).toBe('Searched 1 pattern');
    expect(formatToolGroupLabel('grep', 3)).toBe('Searched 3 patterns');
  });

  test('uses generic format for unknown tools', () => {
    expect(formatToolGroupLabel('custom', 2)).toBe('2 custom calls');
  });
});

/* ── extractToolCallSnippet ───────────────────────────────── */

describe('extractToolCallSnippet', () => {
  test('extracts file name for view', () => {
    expect(extractToolCallSnippet('view', { path: '/src/components/ChatPane.tsx' }))
      .toBe('ChatPane.tsx');
  });

  test('extracts pattern for grep', () => {
    expect(extractToolCallSnippet('grep', { pattern: 'TODO' })).toBe('TODO');
  });

  test('extracts command for powershell', () => {
    expect(extractToolCallSnippet('powershell', { command: 'npm test' })).toBe('npm test');
  });

  test('returns undefined for unknown tool', () => {
    expect(extractToolCallSnippet('custom', {})).toBeUndefined();
  });

  test('falls back to first string arg when specific key is missing', () => {
    expect(extractToolCallSnippet('view', { file_path: '/src/foo.ts' })).toBe('/src/foo.ts');
  });
});