mirror of
https://github.com/affaan-m/ECC.git
synced 2026-11-04 15:15:49 +00:00
343 lines
11 KiB
JavaScript
343 lines
11 KiB
JavaScript
/**
|
|
* Tests for scripts/gemini-adapt-agents.js
|
|
*/
|
|
|
|
const assert = require('assert');
|
|
const fs = require('fs');
|
|
const os = require('os');
|
|
const path = require('path');
|
|
const { execFileSync } = require('child_process');
|
|
|
|
const SCRIPT = path.join(__dirname, '..', '..', 'scripts', 'gemini-adapt-agents.js');
|
|
|
|
function run(args = [], options = {}) {
|
|
try {
|
|
const stdout = execFileSync('node', [SCRIPT, ...args], {
|
|
encoding: 'utf8',
|
|
stdio: ['pipe', 'pipe', 'pipe'],
|
|
cwd: options.cwd,
|
|
timeout: 10000,
|
|
});
|
|
return { code: 0, stdout, stderr: '' };
|
|
} catch (error) {
|
|
return {
|
|
code: error.status || 1,
|
|
stdout: error.stdout || '',
|
|
stderr: error.stderr || '',
|
|
};
|
|
}
|
|
}
|
|
|
|
function test(name, fn) {
|
|
try {
|
|
fn();
|
|
console.log(` ✓ ${name}`);
|
|
return true;
|
|
} catch (error) {
|
|
console.log(` ✗ ${name}`);
|
|
console.log(` Error: ${error.message}`);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
function createTempDir() {
|
|
return fs.mkdtempSync(path.join(os.tmpdir(), 'ecc-gemini-adapt-'));
|
|
}
|
|
|
|
function cleanupTempDir(dirPath) {
|
|
fs.rmSync(dirPath, { recursive: true, force: true });
|
|
}
|
|
|
|
function writeAgent(dirPath, name, body) {
|
|
fs.mkdirSync(dirPath, { recursive: true });
|
|
fs.writeFileSync(path.join(dirPath, name), body);
|
|
}
|
|
|
|
function runTests() {
|
|
console.log('\n=== Testing gemini-adapt-agents.js ===\n');
|
|
|
|
let passed = 0;
|
|
let failed = 0;
|
|
|
|
if (test('shows help with an explicit help flag', () => {
|
|
const result = run(['--help']);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
assert.ok(result.stdout.includes('Adapt ECC agent frontmatter for Gemini CLI'));
|
|
assert.ok(result.stdout.includes('Usage:'));
|
|
})) passed++; else failed++;
|
|
|
|
if (test('adapts Claude Code tool names and strips unsupported color metadata', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'gan-planner.md',
|
|
[
|
|
'---',
|
|
'name: gan-planner',
|
|
'description: Planner agent',
|
|
'tools: [Read, Write, Edit, Bash, Grep, Glob, WebSearch, WebFetch, mcp__context7__resolve-library-id]',
|
|
'model: opus',
|
|
'color: purple',
|
|
'---',
|
|
'',
|
|
'Body'
|
|
].join('\n')
|
|
);
|
|
|
|
const result = run([agentsDir]);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
assert.ok(result.stdout.includes('Updated 1 agent file(s)'));
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'gan-planner.md'), 'utf8');
|
|
assert.ok(updated.includes('tools: ["read_file", "write_file", "replace", "run_shell_command", "grep_search", "glob", "google_web_search", "web_fetch", "mcp_context7_resolve_library_id"]'));
|
|
assert.ok(!updated.includes('color: purple'));
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('adapts comma-separated scalar Claude Code tools', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'docs-lookup.md',
|
|
[
|
|
'---',
|
|
'name: docs-lookup',
|
|
'description: Documentation lookup agent',
|
|
'tools: Read, Grep, mcp__context7__resolve-library-id, mcp__context7__query-docs',
|
|
'model: sonnet',
|
|
'---',
|
|
'',
|
|
'Body'
|
|
].join('\n')
|
|
);
|
|
|
|
const result = run([agentsDir]);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'docs-lookup.md'), 'utf8');
|
|
assert.ok(updated.includes('tools: ["read_file", "grep_search", "mcp_context7_resolve_library_id", "mcp_context7_query_docs"]'));
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('defaults to the cwd .gemini/agents directory', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'architect.md',
|
|
[
|
|
'---',
|
|
'name: architect',
|
|
'description: Architect agent',
|
|
'tools: ["Read", "Grep", "Glob"]',
|
|
'model: opus',
|
|
'---',
|
|
'',
|
|
'Body'
|
|
].join('\n')
|
|
);
|
|
|
|
const result = run([], { cwd: tempDir });
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'architect.md'), 'utf8');
|
|
assert.ok(updated.includes('tools: ["read_file", "grep_search", "glob"]'));
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('deduplicates MultiEdit/Edit and ReadFile/NotebookRead/Read', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'code-simplifier.md',
|
|
[
|
|
'---',
|
|
'name: code-simplifier',
|
|
'description: Refactoring agent',
|
|
'tools: Read, ReadFile, NotebookRead, Edit, MultiEdit, Write',
|
|
'---',
|
|
'',
|
|
'Body'
|
|
].join('\n')
|
|
);
|
|
|
|
const result = run([agentsDir]);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'code-simplifier.md'), 'utf8');
|
|
assert.ok(updated.includes('tools: ["read_file", "replace", "write_file"]'));
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('adapts a CRLF agent on Windows and preserves its line endings', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'windows.md',
|
|
['---', 'name: windows', 'tools: Read, Bash', 'color: purple', '---', '', 'Body line'].join('\r\n')
|
|
);
|
|
|
|
const result = run([agentsDir]);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
assert.ok(result.stdout.includes('Updated 1 agent file(s)'), result.stdout);
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'windows.md'), 'utf8');
|
|
assert.ok(updated.includes('tools: ["read_file", "run_shell_command"]'));
|
|
assert.ok(!updated.includes('color: purple'));
|
|
assert.ok(updated.includes('\r\n'), 'CRLF endings were lost');
|
|
assert.ok(!/[^\r]\n/.test(updated), 'found a bare LF in a CRLF file');
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('rejects an unknown flag without modifying any file', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'architect.md',
|
|
['---', 'name: architect', 'tools: Read, Bash', '---', '', 'Body'].join('\n')
|
|
);
|
|
const before = fs.readFileSync(path.join(agentsDir, 'architect.md'), 'utf8');
|
|
|
|
const result = run(['--dry-run', agentsDir]);
|
|
assert.strictEqual(result.code, 1);
|
|
assert.ok(result.stderr.includes('Unknown option: --dry-run'), result.stderr);
|
|
assert.strictEqual(fs.readFileSync(path.join(agentsDir, 'architect.md'), 'utf8'), before);
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('adapts multi-line YAML block list tools and is idempotent on rerun', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'block-tools.md',
|
|
[
|
|
'---',
|
|
'name: block-tools',
|
|
'description: Block list tools',
|
|
'tools:',
|
|
' - Read',
|
|
' # comment inside list',
|
|
' - Grep',
|
|
'model: sonnet',
|
|
'---',
|
|
'',
|
|
'Body'
|
|
].join('\n')
|
|
);
|
|
|
|
const first = run([agentsDir]);
|
|
assert.strictEqual(first.code, 0, first.stderr);
|
|
assert.ok(first.stdout.includes('Updated 1 agent file(s)'));
|
|
|
|
const afterFirst = fs.readFileSync(path.join(agentsDir, 'block-tools.md'), 'utf8');
|
|
assert.ok(afterFirst.includes('tools: ["read_file", "grep_search"]'));
|
|
assert.ok(!afterFirst.includes('- Read'));
|
|
|
|
const second = run([agentsDir]);
|
|
assert.strictEqual(second.code, 0, second.stderr);
|
|
assert.ok(second.stdout.includes('Updated 0 agent file(s)'));
|
|
assert.ok(second.stdout.includes('1 already compatible'));
|
|
assert.strictEqual(fs.readFileSync(path.join(agentsDir, 'block-tools.md'), 'utf8'), afterFirst);
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('strips unquoted inline YAML comments on block-list items while preserving quoted # literals', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'inline-comments.md',
|
|
[
|
|
'---',
|
|
'name: inline-comments',
|
|
'description: Inline comment handling',
|
|
'tools: # allowed tools',
|
|
' - Read # inspect files',
|
|
' - "Read # explanation" # trailing comment',
|
|
' - \'Grep # literal\' # another comment',
|
|
'---',
|
|
'',
|
|
'Body'
|
|
].join('\n')
|
|
);
|
|
|
|
const result = run([agentsDir]);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'inline-comments.md'), 'utf8');
|
|
assert.ok(
|
|
updated.includes('tools: ["read_file", "Read # explanation", "Grep # literal"]'),
|
|
updated
|
|
);
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
if (test('preserves mixed line endings in the body verbatim when adapting frontmatter', () => {
|
|
const tempDir = createTempDir();
|
|
const agentsDir = path.join(tempDir, '.gemini', 'agents');
|
|
const mixedBody = '\nFirst LF line\nPasted CRLF line\r\nFinal LF line\n';
|
|
|
|
try {
|
|
writeAgent(
|
|
agentsDir,
|
|
'mixed-endings.md',
|
|
`---\nname: mixed-endings\ntools: Read, Bash\ncolor: purple\n---\n${mixedBody}`
|
|
);
|
|
|
|
const result = run([agentsDir]);
|
|
assert.strictEqual(result.code, 0, result.stderr);
|
|
|
|
const updated = fs.readFileSync(path.join(agentsDir, 'mixed-endings.md'), 'utf8');
|
|
assert.strictEqual(
|
|
updated,
|
|
`---\nname: mixed-endings\ntools: ["read_file", "run_shell_command"]\n---\n${mixedBody}`
|
|
);
|
|
} finally {
|
|
cleanupTempDir(tempDir);
|
|
}
|
|
})) passed++; else failed++;
|
|
|
|
console.log(`\nResults: Passed: ${passed}, Failed: ${failed}`);
|
|
process.exit(failed > 0 ? 1 : 0);
|
|
}
|
|
|
|
runTests();
|