wehub-resource-sync
66 lines
2.3 KiB
TypeScript
66 lines
2.3 KiB
TypeScript
import { beforeAll, describe, expect, it } from "bun:test";
|
|
import type { AgentMessage } from "@oh-my-pi/pi-agent-core";
|
|
import { TreeSelectorComponent } from "@oh-my-pi/pi-coding-agent/modes/components/tree-selector";
|
|
import * as themeModule from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import type { SessionEntry, SessionTreeNode } from "@oh-my-pi/pi-coding-agent/session/session-entries";
|
|
|
|
let counter = 0;
|
|
function makeMessageNode(message: AgentMessage, parentId: string | null = null, label?: string): SessionTreeNode {
|
|
const id = `entry-${counter++}`;
|
|
const entry: SessionEntry = {
|
|
type: "message",
|
|
id,
|
|
parentId,
|
|
timestamp: new Date().toISOString(),
|
|
message,
|
|
};
|
|
return { entry, children: [], label };
|
|
}
|
|
|
|
function render(tree: SessionTreeNode[], width = 120): string {
|
|
const selector = new TreeSelectorComponent(
|
|
tree,
|
|
tree[tree.length - 1]?.entry.id ?? null,
|
|
60,
|
|
() => {},
|
|
() => {},
|
|
);
|
|
return Bun.stripANSI(selector.render(width).join("\n"));
|
|
}
|
|
|
|
describe("TreeSelectorComponent developer message rendering", () => {
|
|
beforeAll(async () => {
|
|
await themeModule.initTheme(false, undefined, undefined, "dark", "light");
|
|
});
|
|
|
|
it("renders developer messages with their content, not just [developer]", () => {
|
|
const planContent = "## Plan\n\n1. Fix the tree selector\n2. Update the HTML export";
|
|
const root = makeMessageNode({ role: "user", content: "/plan", timestamp: 1 });
|
|
const developer = makeMessageNode(
|
|
{ role: "developer", content: [{ type: "text", text: planContent }], timestamp: 2 },
|
|
root.entry.id,
|
|
);
|
|
root.children.push(developer);
|
|
|
|
const rendered = render([root]);
|
|
|
|
expect(rendered).toContain("developer:");
|
|
expect(rendered).toContain("Fix the tree selector");
|
|
expect(rendered).toContain("Update the HTML export");
|
|
expect(rendered).not.toMatch(/^\s*\[developer\]\s*$/m);
|
|
});
|
|
|
|
it("matches developer messages in search (content is searchable)", () => {
|
|
const planContent = "ZZZ_UNIQUE_PLAN_TOKEN approved plan body";
|
|
const root = makeMessageNode({ role: "user", content: "/plan", timestamp: 1 });
|
|
const developer = makeMessageNode(
|
|
{ role: "developer", content: [{ type: "text", text: planContent }], timestamp: 2 },
|
|
root.entry.id,
|
|
);
|
|
root.children.push(developer);
|
|
|
|
const rendered = render([root]);
|
|
expect(rendered).toContain("ZZZ_UNIQUE_PLAN_TOKEN");
|
|
});
|
|
});
|