54 lines
2.2 KiB
TypeScript
54 lines
2.2 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import { buildBootstrapContextFiles, DEFAULT_BOOTSTRAP_MAX_CHARS } from "./pi-embedded-helpers.js";
|
|
import { DEFAULT_AGENTS_FILENAME } from "./workspace.js";
|
|
|
|
const makeFile = (overrides: Partial<WorkspaceBootstrapFile>): WorkspaceBootstrapFile => ({
|
|
name: DEFAULT_AGENTS_FILENAME,
|
|
path: "/tmp/AGENTS.md",
|
|
content: "",
|
|
missing: false,
|
|
...overrides,
|
|
});
|
|
describe("buildBootstrapContextFiles", () => {
|
|
it("keeps missing markers", () => {
|
|
const files = [makeFile({ missing: true, content: undefined })];
|
|
expect(buildBootstrapContextFiles(files)).toEqual([
|
|
{
|
|
path: DEFAULT_AGENTS_FILENAME,
|
|
content: "[MISSING] Expected at: /tmp/AGENTS.md",
|
|
},
|
|
]);
|
|
});
|
|
it("skips empty or whitespace-only content", () => {
|
|
const files = [makeFile({ content: " \n " })];
|
|
expect(buildBootstrapContextFiles(files)).toEqual([]);
|
|
});
|
|
it("truncates large bootstrap content", () => {
|
|
const head = `HEAD-${"a".repeat(600)}`;
|
|
const tail = `${"b".repeat(300)}-TAIL`;
|
|
const long = `${head}${tail}`;
|
|
const files = [makeFile({ name: "TOOLS.md", content: long })];
|
|
const warnings: string[] = [];
|
|
const maxChars = 200;
|
|
const expectedTailChars = Math.floor(maxChars * 0.2);
|
|
const [result] = buildBootstrapContextFiles(files, {
|
|
maxChars,
|
|
warn: (message) => warnings.push(message),
|
|
});
|
|
expect(result?.content).toContain("[...truncated, read TOOLS.md for full content...]");
|
|
expect(result?.content.length).toBeLessThan(long.length);
|
|
expect(result?.content.startsWith(long.slice(0, 120))).toBe(true);
|
|
expect(result?.content.endsWith(long.slice(-expectedTailChars))).toBe(true);
|
|
expect(warnings).toHaveLength(1);
|
|
expect(warnings[0]).toContain("TOOLS.md");
|
|
expect(warnings[0]).toContain("limit 200");
|
|
});
|
|
it("keeps content under the default limit", () => {
|
|
const long = "a".repeat(DEFAULT_BOOTSTRAP_MAX_CHARS - 10);
|
|
const files = [makeFile({ content: long })];
|
|
const [result] = buildBootstrapContextFiles(files);
|
|
expect(result?.content).toBe(long);
|
|
expect(result?.content).not.toContain("[...truncated, read AGENTS.md for full content...]");
|
|
});
|
|
});
|