Jack Amadeo
99 lines
3.0 KiB
TypeScript
99 lines
3.0 KiB
TypeScript
import { describe, expect, it } from "vitest";
|
|
import {
|
|
buildProjectSystemPrompt,
|
|
composeSystemPrompt,
|
|
getProjectArtifactRoots,
|
|
getProjectFolderName,
|
|
getProjectFolderOption,
|
|
} from "./chatProjectContext";
|
|
|
|
describe("chatProjectContext", () => {
|
|
it("builds project instructions from stored project settings", () => {
|
|
const systemPrompt = buildProjectSystemPrompt({
|
|
id: "project-1",
|
|
name: "Goose2",
|
|
description: "Desktop app",
|
|
prompt: "Always read AGENTS.md before editing.",
|
|
icon: "folder",
|
|
color: "#000000",
|
|
preferredProvider: "goose",
|
|
preferredModel: "claude-sonnet-4",
|
|
workingDirs: ["/Users/wesb/dev/goose2"],
|
|
useWorktrees: true,
|
|
order: 0,
|
|
archivedAt: null,
|
|
createdAt: "now",
|
|
updatedAt: "now",
|
|
});
|
|
|
|
expect(systemPrompt).toContain("<project-settings>");
|
|
expect(systemPrompt).toContain("Project name: Goose2");
|
|
expect(systemPrompt).toContain(
|
|
"Working directories: /Users/wesb/dev/goose2",
|
|
);
|
|
expect(systemPrompt).toContain(
|
|
"Default working directory: /Users/wesb/dev/goose2",
|
|
);
|
|
expect(systemPrompt).toContain("Preferred provider: goose");
|
|
expect(systemPrompt).toContain(
|
|
"Use git worktrees for branch isolation: yes",
|
|
);
|
|
expect(systemPrompt).toContain("<project-file-policy>");
|
|
expect(systemPrompt).toContain(
|
|
"Use /Users/wesb/dev/goose2 as the default working directory for this project.",
|
|
);
|
|
expect(systemPrompt).toContain("<project-instructions>");
|
|
expect(systemPrompt).toContain("Always read AGENTS.md before editing.");
|
|
});
|
|
|
|
it("combines persona and project prompts without empty sections", () => {
|
|
expect(
|
|
composeSystemPrompt("Persona prompt", undefined, "Project prompt"),
|
|
).toBe("Persona prompt\n\nProject prompt");
|
|
});
|
|
|
|
it("extracts the folder name from a path", () => {
|
|
expect(getProjectFolderName("/Users/wesb/dev/goose2")).toBe("goose2");
|
|
expect(getProjectFolderName("C:\\Users\\wesb\\goose2\\")).toBe("goose2");
|
|
});
|
|
|
|
it("creates folder options from the project's working directories", () => {
|
|
expect(
|
|
getProjectFolderOption({
|
|
workingDirs: ["/Users/wesb/dev/goose2", "/Users/wesb/dev/other"],
|
|
}),
|
|
).toEqual([
|
|
{
|
|
id: "/Users/wesb/dev/goose2",
|
|
name: "goose2",
|
|
path: "/Users/wesb/dev/goose2",
|
|
},
|
|
{
|
|
id: "/Users/wesb/dev/other",
|
|
name: "other",
|
|
path: "/Users/wesb/dev/other",
|
|
},
|
|
]);
|
|
});
|
|
|
|
it("returns an empty array when workingDirs is empty", () => {
|
|
expect(
|
|
getProjectFolderOption({
|
|
workingDirs: [],
|
|
}),
|
|
).toEqual([]);
|
|
});
|
|
|
|
it("returns an empty array when project is null", () => {
|
|
expect(getProjectFolderOption(null)).toEqual([]);
|
|
});
|
|
|
|
it("returns working dirs unchanged", () => {
|
|
expect(
|
|
getProjectArtifactRoots({
|
|
workingDirs: ["/Users/wesb/dev/goose2", "/Users/wesb/dev/other"],
|
|
}),
|
|
).toEqual(["/Users/wesb/dev/goose2", "/Users/wesb/dev/other"]);
|
|
});
|
|
});
|