symbol-context.test.ts — mcp Source File

Source Code

import { describe, it, expect, afterAll } from '@jest/globals';
import { promises as fs } from 'fs';
import * as path from 'path';
import * as os from 'os';
import { buildIndexes } from '../cache/graph-cache';
import { findSymbol, renderSymbolContext, languageFromExtension } from './symbol-context';
import { MAX_SOURCE_LINES } from '../constants';

// ── Helpers ──

/** Build a minimal IndexedGraph from inline nodes + relationships */
function buildGraph(
  nodes: Array<{ id: string; labels: string[]; properties: Record<string, unknown> }>,
  relationships: Array<{ id: string; type: string; startNode: string; endNode: string; properties?: Record<string, unknown> }> = [],
) {
  const raw: any = {
    repo: 'test-repo',
    graph: { nodes, relationships: relationships.map(r => ({ ...r, properties: r.properties ?? {} })) },
  };
  return buildIndexes(raw, 'test-key');
}

// ── findSymbol tests (Issue #107) ──

describe('findSymbol', () => {
  it('query "filter" does NOT match symbol "f" (bidirectional bug regression)', () => {
    const graph = buildGraph([
      { id: 'f1', labels: ['Function'], properties: { name: 'f', filePath: 'a.py', startLine: 1 } },
      { id: 'f2', labels: ['Function'], properties: { name: 'filter', filePath: 'b.py', startLine: 1 } },
    ]);

    const results = findSymbol(graph, 'filter');
    const names = results.map(n => n.properties?.name);
    expect(names).toContain('filter');
    expect(names).not.toContain('f');
  });

  it('single-char query "f" finds exact match via Strategy 1, substring returns empty', () => {
    const graph = buildGraph([
      { id: 'f1', labels: ['Function'], properties: { name: 'f', filePath: 'a.py', startLine: 1 } },
      { id: 'f2', labels: ['Function'], properties: { name: 'filter', filePath: 'b.py', startLine: 1 } },
    ]);

    // Exact match works
    const exactResults = findSymbol(graph, 'f');
    expect(exactResults.length).toBe(1);
    expect(exactResults[0].properties?.name).toBe('f');

    // A single-char query that has no exact match returns nothing (no substring for len < 2)
    const noMatchResults = findSymbol(graph, 'x');
    expect(noMatchResults.length).toBe(0);
  });

  it('two functions named "filter" — one with 3 callers sorts before one with 0', () => {
    const graph = buildGraph(
      [
        { id: 'popular', labels: ['Function'], properties: { name: 'filter', filePath: 'a.py', startLine: 1 } },
        { id: 'lonely', labels: ['Function'], properties: { name: 'filter', filePath: 'b.py', startLine: 1 } },
        { id: 'c1', labels: ['Function'], properties: { name: 'caller1', filePath: 'c.py', startLine: 1 } },
        { id: 'c2', labels: ['Function'], properties: { name: 'caller2', filePath: 'c.py', startLine: 10 } },
// ... (138 more lines)