mirror of
https://github.com/abhigyanpatwari/GitNexus.git
synced 2026-09-07 08:26:11 +00:00
81 lines
2.8 KiB
TypeScript
81 lines
2.8 KiB
TypeScript
import { describe, it, expect, beforeAll } from 'vitest';
|
|
import path from 'path';
|
|
import {
|
|
FIXTURES,
|
|
getRelationships,
|
|
getNodesByLabel,
|
|
getNodesByLabelFull,
|
|
runPipelineFromRepo,
|
|
type PipelineResult,
|
|
} from './helpers.js';
|
|
|
|
describe('Python @mcp.tool() detection', () => {
|
|
let result: PipelineResult;
|
|
|
|
beforeAll(async () => {
|
|
result = await runPipelineFromRepo(path.join(FIXTURES, 'python-mcp-tools'), () => {});
|
|
}, 60000);
|
|
|
|
it('creates Tool nodes for @mcp.tool() decorated functions', () => {
|
|
const tools = getNodesByLabel(result, 'Tool');
|
|
expect(tools).toContain('get_weather');
|
|
expect(tools).toContain('search_docs');
|
|
expect(tools).toContain('explicit_tool');
|
|
});
|
|
|
|
it('uses handler functions for HANDLES_TOOL edges', () => {
|
|
const edges = getRelationships(result, 'HANDLES_TOOL');
|
|
expect(edges.length).toBeGreaterThanOrEqual(2);
|
|
|
|
const weatherEdge = edges.find((e) => e.target === 'get_weather');
|
|
expect(weatherEdge).toBeDefined();
|
|
expect(weatherEdge!.source).toBe('get_weather');
|
|
expect(weatherEdge!.sourceLabel).toBe('Function');
|
|
|
|
const searchEdge = edges.find((e) => e.target === 'search_docs');
|
|
expect(searchEdge).toBeDefined();
|
|
expect(searchEdge!.source).toBe('search_docs');
|
|
expect(searchEdge!.sourceLabel).toBe('Function');
|
|
});
|
|
|
|
it('detects exactly 3 tools from the fixture', () => {
|
|
const tools = getNodesByLabel(result, 'Tool');
|
|
expect(tools).toHaveLength(3);
|
|
});
|
|
|
|
it('uses Python handler docstrings as tool descriptions', () => {
|
|
const tools = getNodesByLabelFull(result, 'Tool');
|
|
const descriptions = new Map(tools.map((tool) => [tool.name, tool.properties.description]));
|
|
|
|
expect(descriptions.get('get_weather')).toBe('Get weather for a city.');
|
|
expect(descriptions.get('search_docs')).toBe('Search documentation.');
|
|
expect(descriptions.get('explicit_tool')).toBe('Explicit description');
|
|
});
|
|
|
|
it('links each tool only to flows rooted at its handler', () => {
|
|
const edges = getRelationships(result, 'ENTRY_POINT_OF').filter(
|
|
(e) => e.sourceLabel === 'Tool' && e.targetLabel === 'Process',
|
|
);
|
|
|
|
const flowsByTool = new Map<string, string[]>();
|
|
for (const edge of edges) {
|
|
let flows = flowsByTool.get(edge.source);
|
|
if (flows === undefined) {
|
|
flows = [];
|
|
flowsByTool.set(edge.source, flows);
|
|
}
|
|
flows.push(edge.target);
|
|
}
|
|
|
|
const weatherFlows = flowsByTool.get('get_weather') ?? [];
|
|
const searchFlows = flowsByTool.get('search_docs') ?? [];
|
|
|
|
expect(weatherFlows).toHaveLength(1);
|
|
expect(weatherFlows[0]).toContain('Get_weather');
|
|
expect(weatherFlows[0]).toContain('_format_weather');
|
|
expect(searchFlows).toHaveLength(1);
|
|
expect(searchFlows[0]).toContain('Search_docs');
|
|
expect(searchFlows[0]).toContain('_rank_docs');
|
|
expect(weatherFlows).not.toEqual(searchFlows);
|
|
});
|
|
});
|