alex
2f198ea0d2
BREAKING: Heuristic fallback removed — LLM client now required Changes: - Remove extractFileHeuristic, extractPackageHeuristic, all helpers - extractFileLLM / extractPackageLLM now throw LLMError when client missing - Add hybrid binary detection: extension blacklist + content sniffing - Increase file size limit: 50KB → 500KB (text files only) - Restructure src/ into subdirectories: - src/llm/ — all LLM clients, extract, batch, cache, error - src/ast/ — AST extraction - src/cli/ — CLI entry point - Update all imports across codebase and tests - Add tests/mock-llm.ts helper for deterministic mock clients - Update all tests to use mock LLM clients (no heuristics dependency) - All 52 tests passing (including 8 real LLM integration tests)
128 lines
3.9 KiB
TypeScript
128 lines
3.9 KiB
TypeScript
import { describe, it, expect, beforeEach, afterEach } from "vitest";
|
|
import {
|
|
mkdtempSync,
|
|
writeFileSync,
|
|
mkdirSync,
|
|
rmSync,
|
|
readFileSync,
|
|
} from "fs";
|
|
import { join } from "path";
|
|
import { tmpdir } from "os";
|
|
import { initProject } from "../src/init.js";
|
|
import { patchFile } from "../src/patch.js";
|
|
import { validateMaps } from "../src/validate.js";
|
|
import { createMockFileClient, createMockPackageClient } from "./mock-llm.js";
|
|
|
|
describe("integration", () => {
|
|
let dir: string;
|
|
|
|
beforeEach(() => {
|
|
dir = mkdtempSync(join(tmpdir(), "pi-map-int-"));
|
|
});
|
|
|
|
afterEach(() => {
|
|
rmSync(dir, { recursive: true });
|
|
});
|
|
|
|
it("init creates .pi-map.md files", async () => {
|
|
mkdirSync(join(dir, "src"));
|
|
writeFileSync(join(dir, "src", "index.ts"), `export function foo() {}\n`);
|
|
const client = createMockFileClient();
|
|
await initProject(dir, { llmClient: client, verbose: false });
|
|
|
|
const map = readFileSync(join(dir, "src", ".pi-map.md"), "utf8");
|
|
expect(map).toContain("# src");
|
|
});
|
|
|
|
it("patch updates a file entry", async () => {
|
|
mkdirSync(join(dir, "src"));
|
|
writeFileSync(join(dir, "src", "index.ts"), `export function foo() {}\n`);
|
|
writeFileSync(join(dir, "src", "utils.ts"), `export const bar = 1;\n`);
|
|
const client = createMockFileClient();
|
|
await initProject(dir, { llmClient: client, verbose: false });
|
|
|
|
// Modify a file
|
|
writeFileSync(
|
|
join(dir, "src", "index.ts"),
|
|
`export function foo() {}\nexport function baz() {}\n`,
|
|
);
|
|
await patchFile(join(dir, "src", "index.ts"), client, dir);
|
|
|
|
const map = readFileSync(join(dir, "src", ".pi-map.md"), "utf8");
|
|
expect(map).toContain("baz");
|
|
});
|
|
|
|
it("patch adds dirty marker for large packages", async () => {
|
|
mkdirSync(join(dir, "src"));
|
|
// Create 11 files so it's a "large" package
|
|
for (let i = 0; i < 11; i++) {
|
|
writeFileSync(
|
|
join(dir, "src", `file${i}.ts`),
|
|
`export const x${i} = ${i};\n`,
|
|
);
|
|
}
|
|
const client = createMockFileClient();
|
|
await initProject(dir, { llmClient: client, verbose: false });
|
|
|
|
// Modify a file
|
|
writeFileSync(
|
|
join(dir, "src", "file0.ts"),
|
|
`export const x0 = 0;\nexport const y = 99;\n`,
|
|
);
|
|
await patchFile(join(dir, "src", "file0.ts"), client, dir);
|
|
|
|
const map = readFileSync(join(dir, "src", ".pi-map.md"), "utf8");
|
|
expect(map).toContain("y");
|
|
expect(map).toContain("dirty");
|
|
expect(map).toContain("patched");
|
|
});
|
|
|
|
it("validate detects new files", async () => {
|
|
mkdirSync(join(dir, "src"));
|
|
writeFileSync(join(dir, "src", "a.ts"), `export const a = 1;\n`);
|
|
const client = createMockFileClient();
|
|
await initProject(dir, { llmClient: client, verbose: false });
|
|
|
|
// Add new file
|
|
writeFileSync(join(dir, "src", "b.ts"), `export const b = 2;\n`);
|
|
|
|
const result = await validateMaps(dir);
|
|
expect(result.clean).toBe(false);
|
|
expect(result.discrepancies.some((d) => d.type === "missing")).toBe(true);
|
|
});
|
|
|
|
it("validate detects deleted files", async () => {
|
|
mkdirSync(join(dir, "src"));
|
|
writeFileSync(join(dir, "src", "a.ts"), `export const a = 1;\n`);
|
|
writeFileSync(join(dir, "src", "b.ts"), `export const b = 2;\n`);
|
|
const client = createMockFileClient();
|
|
await initProject(dir, { llmClient: client, verbose: false });
|
|
|
|
// Delete a file
|
|
rmSync(join(dir, "src", "b.ts"));
|
|
|
|
const result = await validateMaps(dir);
|
|
expect(result.clean).toBe(false);
|
|
expect(result.discrepancies.some((d) => d.type === "orphaned")).toBe(true);
|
|
});
|
|
|
|
it("validate detects changed signatures", async () => {
|
|
mkdirSync(join(dir, "src"));
|
|
writeFileSync(join(dir, "src", "a.ts"), `export const a = 1;\n`);
|
|
const client = createMockFileClient();
|
|
await initProject(dir, { llmClient: client, verbose: false });
|
|
|
|
// Change exports
|
|
writeFileSync(
|
|
join(dir, "src", "a.ts"),
|
|
`export const a = 1;\nexport const c = 3;\n`,
|
|
);
|
|
|
|
const result = await validateMaps(dir);
|
|
expect(result.clean).toBe(false);
|
|
expect(result.discrepancies.some((d) => d.type === "stale-signature")).toBe(
|
|
true,
|
|
);
|
|
});
|
|
});
|