102 lines
3.9 KiB
TypeScript
102 lines
3.9 KiB
TypeScript
import { describe, expect, test } from "bun:test";
|
|
import { stripHtmlComments, isMemoryFilePath, getLargeMemoryFiles } from "../../src/utils/claudemd";
|
|
import { buildEffectiveSystemPrompt } from "../../src/utils/systemPrompt";
|
|
import { createTempDir, cleanupTempDir, writeTempFile } from "../mocks/file-system";
|
|
|
|
// ─── CLAUDE.md Integration with System Prompt ─────────────────────────
|
|
|
|
describe("Context build: CLAUDE.md + system prompt integration", () => {
|
|
test("buildEffectiveSystemPrompt passes through default prompt", () => {
|
|
const result = buildEffectiveSystemPrompt({
|
|
defaultSystemPrompt: "You are Claude.",
|
|
});
|
|
// Result is an array of strings (may be split differently)
|
|
const joined = Array.from(result).join("");
|
|
expect(joined).toBe("You are Claude.");
|
|
});
|
|
|
|
test("buildEffectiveSystemPrompt handles empty prompts", () => {
|
|
const result = buildEffectiveSystemPrompt({
|
|
defaultSystemPrompt: "",
|
|
});
|
|
const joined = Array.from(result).join("");
|
|
expect(joined).toBe("");
|
|
});
|
|
|
|
test("buildEffectiveSystemPrompt with overrideSystemPrompt replaces everything", () => {
|
|
const result = buildEffectiveSystemPrompt({
|
|
defaultSystemPrompt: "Default",
|
|
overrideSystemPrompt: "Override",
|
|
});
|
|
const joined = Array.from(result).join("");
|
|
expect(joined).toBe("Override");
|
|
});
|
|
|
|
test("buildEffectiveSystemPrompt with customSystemPrompt replaces default", () => {
|
|
const result = buildEffectiveSystemPrompt({
|
|
defaultSystemPrompt: "Default",
|
|
customSystemPrompt: "Custom",
|
|
});
|
|
const joined = Array.from(result).join("");
|
|
expect(joined).toBe("Custom");
|
|
});
|
|
|
|
test("buildEffectiveSystemPrompt with appendSystemPrompt includes both", () => {
|
|
const result = buildEffectiveSystemPrompt({
|
|
defaultSystemPrompt: "Main prompt",
|
|
appendSystemPrompt: "Appended",
|
|
});
|
|
const joined = Array.from(result).join("");
|
|
expect(joined).toContain("Main prompt");
|
|
expect(joined).toContain("Appended");
|
|
// Appended should come after main
|
|
expect(joined.indexOf("Main prompt")).toBeLessThan(
|
|
joined.indexOf("Appended")
|
|
);
|
|
});
|
|
});
|
|
|
|
// ─── CLAUDE.md Discovery with Real File System ───────────────────────
|
|
|
|
describe("Context build: CLAUDE.md file system integration", () => {
|
|
let tempDir: string;
|
|
|
|
test("strips HTML comments from CLAUDE.md content", () => {
|
|
const input = "<!-- this is a comment -->Actual content";
|
|
const { content, stripped } = stripHtmlComments(input);
|
|
expect(content).toBe("Actual content");
|
|
expect(stripped).toBe(true);
|
|
});
|
|
|
|
test("preserves code blocks when stripping HTML comments", () => {
|
|
const input = "```\n<!-- not a real comment -->\n```\nReal text";
|
|
const { content } = stripHtmlComments(input);
|
|
expect(content).toContain("<!-- not a real comment -->");
|
|
expect(content).toContain("Real text");
|
|
});
|
|
|
|
test("isMemoryFilePath correctly identifies CLAUDE.md paths", () => {
|
|
expect(isMemoryFilePath("/project/CLAUDE.md")).toBe(true);
|
|
expect(isMemoryFilePath("/project/CLAUDE.local.md")).toBe(true);
|
|
expect(isMemoryFilePath("/project/.claude/rules/file.md")).toBe(true);
|
|
expect(isMemoryFilePath("/project/README.md")).toBe(false);
|
|
expect(isMemoryFilePath("/project/src/index.ts")).toBe(false);
|
|
});
|
|
});
|
|
|
|
// ─── Large Memory File Filtering ──────────────────────────────────────
|
|
|
|
describe("Context build: large memory file filtering", () => {
|
|
test("getLargeMemoryFiles returns empty for empty input", () => {
|
|
expect(getLargeMemoryFiles([])).toEqual([]);
|
|
});
|
|
|
|
test("getLargeMemoryFiles returns empty when all files are small", () => {
|
|
const files = [
|
|
{ path: "/a/CLAUDE.md", content: "small" },
|
|
{ path: "/b/CLAUDE.md", content: "also small" },
|
|
];
|
|
expect(getLargeMemoryFiles(files)).toEqual([]);
|
|
});
|
|
});
|