alexion
6081e3e325
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: 71023aec1546d6d677a65669c5665c0927ce0dd0
84 lines
2.8 KiB
TypeScript
84 lines
2.8 KiB
TypeScript
import type { Box, TuiMouseEvent } from "@earendil-works/pi-tui";
|
|
import { beforeAll, describe, expect, test } from "vitest";
|
|
import { BranchSummaryMessageComponent } from "../src/modes/interactive/components/branch-summary-message.ts";
|
|
import { CompactionSummaryMessageComponent } from "../src/modes/interactive/components/compaction-summary-message.ts";
|
|
import { SkillInvocationMessageComponent } from "../src/modes/interactive/components/skill-invocation-message.ts";
|
|
import { initTheme } from "../src/modes/interactive/theme/theme.ts";
|
|
import { stripAnsi } from "../src/utils/ansi.ts";
|
|
|
|
const WIDTH = 80;
|
|
|
|
function renderText(component: Box): string {
|
|
return stripAnsi(component.render(WIDTH).join("\n"));
|
|
}
|
|
|
|
function clickRow(component: Box, marker: string): void {
|
|
const lines = component.render(WIDTH);
|
|
const row = lines.findIndex((line) => stripAnsi(line).includes(marker));
|
|
expect(row).toBeGreaterThanOrEqual(0);
|
|
const event: TuiMouseEvent = {
|
|
type: "click",
|
|
button: "left",
|
|
x: 2,
|
|
y: row,
|
|
screenX: 2,
|
|
screenY: row,
|
|
width: WIDTH,
|
|
height: lines.length,
|
|
shift: false,
|
|
alt: false,
|
|
ctrl: false,
|
|
clickCount: 1,
|
|
};
|
|
expect(component.handleMouse(event)?.handled).toBe(true);
|
|
}
|
|
|
|
describe("collapsible message components", () => {
|
|
beforeAll(() => initTheme("dark"));
|
|
|
|
test("toggles a compaction summary when clicked", () => {
|
|
const component = new CompactionSummaryMessageComponent({
|
|
role: "compactionSummary",
|
|
summary: "compaction details",
|
|
tokensBefore: 1234,
|
|
timestamp: Date.now(),
|
|
});
|
|
|
|
expect(renderText(component)).not.toContain("compaction details");
|
|
clickRow(component, "[compaction]");
|
|
expect(renderText(component)).toContain("compaction details");
|
|
clickRow(component, "[compaction]");
|
|
expect(renderText(component)).not.toContain("compaction details");
|
|
});
|
|
|
|
test("toggles a branch summary when clicked", () => {
|
|
const component = new BranchSummaryMessageComponent({
|
|
role: "branchSummary",
|
|
summary: "branch details",
|
|
fromId: "entry-1",
|
|
timestamp: Date.now(),
|
|
});
|
|
|
|
expect(renderText(component)).not.toContain("branch details");
|
|
clickRow(component, "[branch]");
|
|
expect(renderText(component)).toContain("branch details");
|
|
clickRow(component, "[branch]");
|
|
expect(renderText(component)).not.toContain("branch details");
|
|
});
|
|
|
|
test("toggles a skill invocation when clicked", () => {
|
|
const component = new SkillInvocationMessageComponent({
|
|
name: "example-skill",
|
|
location: "/tmp/example-skill.md",
|
|
content: "skill details",
|
|
userMessage: undefined,
|
|
});
|
|
|
|
expect(renderText(component)).not.toContain("skill details");
|
|
clickRow(component, "[skill]");
|
|
expect(renderText(component)).toContain("skill details");
|
|
clickRow(component, "[skill]");
|
|
expect(renderText(component)).not.toContain("skill details");
|
|
});
|
|
});
|