Land #1647 onto current main: callers/callees/query (CLI + MCP) now say when --limit hid matches, with totals in JSON and a widening hint. Also covers #1639. Co-authored-by: Colby McHenry <colbymchenry@users.noreply.github.com>
This commit is contained in:
co-authored by
Colby McHenry
parent
43271f3cd3
commit
85550eb2ce
@@ -0,0 +1,101 @@
|
||||
import { afterEach, beforeEach, describe, expect, it } from 'vitest';
|
||||
import { spawnSync } from 'child_process';
|
||||
import * as fs from 'fs';
|
||||
import * as os from 'os';
|
||||
import * as path from 'path';
|
||||
import { CodeGraph } from '../src';
|
||||
|
||||
const BIN = path.resolve(__dirname, '../dist/bin/codegraph.js');
|
||||
|
||||
function runCli(cwd: string, args: string[]) {
|
||||
return spawnSync(process.execPath, [BIN, ...args, '-p', cwd], {
|
||||
encoding: 'utf-8',
|
||||
env: { ...process.env, CODEGRAPH_NO_DAEMON: '1', CODEGRAPH_WASM_RELAUNCHED: '1', NO_COLOR: '1' },
|
||||
});
|
||||
}
|
||||
|
||||
describe('CLI truncation reporting (#1639)', () => {
|
||||
let tempDir: string;
|
||||
|
||||
beforeEach(async () => {
|
||||
tempDir = fs.mkdtempSync(path.join(os.tmpdir(), 'codegraph-cli-truncation-'));
|
||||
fs.writeFileSync(
|
||||
path.join(tempDir, 'lib.ts'),
|
||||
[
|
||||
'export function target() {}',
|
||||
'export function helperA() {}',
|
||||
'export function helperB() {}',
|
||||
'export function helperC() {}',
|
||||
'export function source() { helperA(); helperB(); helperC(); }',
|
||||
'export function TargetHitOne() {}',
|
||||
'export function TargetHitTwo() {}',
|
||||
'export function TargetHitThree() {}',
|
||||
].join('\n'),
|
||||
);
|
||||
for (let i = 0; i < 3; i++) {
|
||||
fs.writeFileSync(
|
||||
path.join(tempDir, `caller-${i}.ts`),
|
||||
`import { target } from './lib';\nexport function caller${i}() { target(); }\n`,
|
||||
);
|
||||
}
|
||||
const cg = CodeGraph.initSync(tempDir);
|
||||
await cg.indexAll();
|
||||
cg.close();
|
||||
});
|
||||
|
||||
afterEach(() => {
|
||||
fs.rmSync(tempDir, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
it('reports exact callers metadata in JSON and human output', () => {
|
||||
const jsonRun = runCli(tempDir, ['callers', 'target', '--limit', '2', '--json']);
|
||||
expect(jsonRun.status).toBe(0);
|
||||
const parsed = JSON.parse(jsonRun.stdout);
|
||||
expect(parsed.callers).toHaveLength(2);
|
||||
expect(parsed.total).toBeGreaterThan(2);
|
||||
expect(parsed.limit).toBe(2);
|
||||
expect(parsed.truncated).toBe(true);
|
||||
|
||||
const humanRun = runCli(tempDir, ['callers', 'target', '--limit', '2']);
|
||||
expect(humanRun.stdout).toMatch(/Callers of "target" \(2 of \d+\):/);
|
||||
expect(humanRun.stdout).toMatch(/Showing 2 of \d+; pass --limit to widen\./);
|
||||
|
||||
const complete = JSON.parse(runCli(tempDir, ['callers', 'target', '--limit', '100', '--json']).stdout);
|
||||
expect(complete.total).toBe(complete.callers.length);
|
||||
expect(complete.limit).toBe(100);
|
||||
expect(complete.truncated).toBe(false);
|
||||
});
|
||||
|
||||
it('reports exact callees metadata in JSON and human output', () => {
|
||||
const jsonRun = runCli(tempDir, ['callees', 'source', '--limit', '2', '--json']);
|
||||
expect(jsonRun.status).toBe(0);
|
||||
const parsed = JSON.parse(jsonRun.stdout);
|
||||
expect(parsed.callees).toHaveLength(2);
|
||||
expect(parsed.total).toBeGreaterThan(2);
|
||||
expect(parsed.limit).toBe(2);
|
||||
expect(parsed.truncated).toBe(true);
|
||||
|
||||
const humanRun = runCli(tempDir, ['callees', 'source', '--limit', '2']);
|
||||
expect(humanRun.stdout).toMatch(/Callees of "source" \(2 of \d+\):/);
|
||||
expect(humanRun.stdout).toMatch(/Showing 2 of \d+; pass --limit to widen\./);
|
||||
|
||||
const complete = JSON.parse(runCli(tempDir, ['callees', 'source', '--limit', '100', '--json']).stdout);
|
||||
expect(complete.total).toBe(complete.callees.length);
|
||||
expect(complete.limit).toBe(100);
|
||||
expect(complete.truncated).toBe(false);
|
||||
});
|
||||
|
||||
it('keeps query --json as an array and reports truncation on stderr', () => {
|
||||
const jsonRun = runCli(tempDir, ['query', 'TargetHit', '--limit', '1', '--json']);
|
||||
expect(jsonRun.status).toBe(0);
|
||||
expect(JSON.parse(jsonRun.stdout)).toHaveLength(1);
|
||||
expect(jsonRun.stderr).toContain('Results truncated at 1; pass --limit to widen.');
|
||||
|
||||
const humanRun = runCli(tempDir, ['query', 'TargetHit', '--limit', '1']);
|
||||
expect(humanRun.stdout).toContain('Results truncated at 1; pass --limit to widen.');
|
||||
|
||||
const complete = runCli(tempDir, ['query', 'TargetHit', '--limit', '100', '--json']);
|
||||
expect(Array.isArray(JSON.parse(complete.stdout))).toBe(true);
|
||||
expect(complete.stderr).not.toContain('Results truncated');
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user