/**
* Unit tests for codebase_retrieval tool
*/
import { jest, describe, it, expect, beforeEach } from '@jest/globals';
import {
handleCodebaseRetrieval,
codebaseRetrievalTool,
} from '../../src/mcp/tools/codebaseRetrieval.js';
import { SearchResult } from '../../src/mcp/serviceClient.js';
describe('codebase_retrieval Tool', () => {
let mockServiceClient: any;
beforeEach(() => {
jest.clearAllMocks();
mockServiceClient = {
semanticSearch: jest.fn(),
getIndexStatus: jest.fn(() => ({
workspace: '/tmp/workspace',
lastIndexed: '2024-01-01T00:00:00.000Z',
status: 'idle',
fileCount: 10,
isStale: false,
})),
};
});
it('validates input query', async () => {
await expect(
handleCodebaseRetrieval({ query: '' } as any, mockServiceClient as any)
).rejects.toThrow(/invalid query/i);
});
it('returns JSON string with expected structure', async () => {
const mockResults: SearchResult[] = [
{ path: 'src/a.ts', content: 'code a', lines: '1-5', relevanceScore: 0.9 },
];
mockServiceClient.semanticSearch.mockResolvedValue(mockResults);
const result = await handleCodebaseRetrieval({ query: 'test' }, mockServiceClient as any);
const parsed = JSON.parse(result);
expect(parsed).toHaveProperty('results');
expect(Array.isArray(parsed.results)).toBe(true);
expect(parsed.results[0].file).toBe('src/a.ts');
expect(parsed.results[0].score).toBeCloseTo(0.9);
expect(parsed).toHaveProperty('metadata');
expect(parsed.metadata.workspace).toBe('/tmp/workspace');
});
it('respects top_k parameter and delegates to semanticSearch', async () => {
mockServiceClient.semanticSearch.mockResolvedValue([]);
await handleCodebaseRetrieval({ query: 'test', top_k: 5 }, mockServiceClient as any);
expect(mockServiceClient.semanticSearch).toHaveBeenCalledWith('test', 5);
});
it('adds reason text for each result', async () => {
const mockResults: SearchResult[] = [
{ path: 'src/b.ts', content: 'code b', relevanceScore: 0.5 },
];
mockServiceClient.semanticSearch.mockResolvedValue(mockResults);
const result = await handleCodebaseRetrieval({ query: 'reason' }, mockServiceClient as any);
const parsed = JSON.parse(result);
expect(parsed.results[0].reason).toMatch(/Semantic match/);
});
it('exposes correct tool schema', () => {
expect(codebaseRetrievalTool.name).toBe('codebase_retrieval');
expect(codebaseRetrievalTool.inputSchema.required).toContain('query');
});
});
