File size: 4,326 Bytes
7a1ad33 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 | /**
* @license
* Copyright 2026 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, expect, it } from 'vitest';
import {
buildToolRegistry,
resolveToolName,
getToolsByCategory,
type ToolCategory,
} from '../utils/tool-registry.js';
describe('tool-registry', () => {
const registry = buildToolRegistry();
describe('buildToolRegistry', () => {
it('includes all canonical built-in tools', () => {
expect(registry.totalTools).toBeGreaterThanOrEqual(26);
});
it('every tool has a valid category', () => {
for (const [name, entry] of registry.tools) {
expect(entry.category).toBeTruthy();
expect(entry.name).toBe(name);
}
});
it('byCategory entries match tools map', () => {
let categoryTotal = 0;
for (const [, entries] of registry.byCategory) {
for (const entry of entries) {
expect(registry.tools.get(entry.name)).toBe(entry);
}
categoryTotal += entries.length;
}
expect(categoryTotal).toBe(registry.totalTools);
});
it('aliasLookup covers every canonical name', () => {
for (const name of registry.tools.keys()) {
expect(registry.aliasLookup.get(name)).toBe(name);
}
});
it('aliasLookup covers every legacy alias', () => {
for (const [, entry] of registry.tools) {
for (const alias of entry.aliases) {
expect(registry.aliasLookup.get(alias)).toBe(entry.name);
}
}
});
it('is deterministic across calls', () => {
const second = buildToolRegistry();
expect([...second.tools.keys()]).toEqual([...registry.tools.keys()]);
expect(second.totalTools).toBe(registry.totalTools);
});
});
describe('resolveToolName', () => {
it('resolves canonical names to themselves', () => {
expect(resolveToolName(registry, 'grep_search')).toBe('grep_search');
expect(resolveToolName(registry, 'run_shell_command')).toBe(
'run_shell_command',
);
});
it('resolves legacy alias to canonical name', () => {
expect(resolveToolName(registry, 'search_file_content')).toBe(
'grep_search',
);
});
it('returns undefined for unknown tool names', () => {
expect(resolveToolName(registry, 'nonexistent_tool')).toBeUndefined();
});
it('returns undefined for empty string', () => {
expect(resolveToolName(registry, '')).toBeUndefined();
});
});
describe('getToolsByCategory', () => {
it('returns file-system tools', () => {
const tools = getToolsByCategory(registry, 'file-system');
const names = tools.map((t) => t.name);
expect(names).toContain('glob');
expect(names).toContain('grep_search');
expect(names).toContain('read_file');
expect(names).toContain('write_file');
expect(names).toContain('replace');
});
it('returns task-tracker tools', () => {
const tools = getToolsByCategory(registry, 'task-tracker');
const names = tools.map((t) => t.name);
expect(names).toContain('tracker_create_task');
expect(names).toContain('tracker_update_task');
expect(names).toContain('tracker_get_task');
expect(names).toContain('tracker_list_tasks');
expect(names).toContain('tracker_add_dependency');
expect(names).toContain('tracker_visualize');
expect(names).toHaveLength(6);
});
it('returns agent tools', () => {
const tools = getToolsByCategory(registry, 'agent');
const names = tools.map((t) => t.name);
expect(names).toContain('invoke_agent');
expect(names).toContain('complete_task');
expect(names).toContain('update_topic');
});
it('returns empty array for unknown category', () => {
expect(
getToolsByCategory(registry, 'nonexistent' as ToolCategory),
).toEqual([]);
});
it('every defined category has at least one tool', () => {
const expectedCategories: ToolCategory[] = [
'file-system',
'shell',
'web',
'planning',
'user-interaction',
'skills',
'task-tracker',
'agent',
'mcp',
];
for (const cat of expectedCategories) {
expect(getToolsByCategory(registry, cat).length).toBeGreaterThan(0);
}
});
});
});
|