wehub-resource-sync
89 lines
2.8 KiB
TypeScript
89 lines
2.8 KiB
TypeScript
import { beforeAll, describe, expect, it } from "bun:test";
|
|
import { resetSettingsForTest, Settings } from "@oh-my-pi/pi-coding-agent/config/settings";
|
|
import type { EvalStatusEvent, EvalToolDetails } from "@oh-my-pi/pi-coding-agent/eval/types";
|
|
import { ToolExecutionComponent } from "@oh-my-pi/pi-coding-agent/modes/components/tool-execution";
|
|
import { initTheme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import type { TUI } from "@oh-my-pi/pi-tui";
|
|
|
|
const uiStub = { requestRender() {}, requestComponentRender() {} } as unknown as TUI;
|
|
|
|
function makeEvalComponent() {
|
|
return new ToolExecutionComponent("eval", { code: "parallel([...])", language: "python" }, {}, undefined, uiStub);
|
|
}
|
|
|
|
/** Build an eval result whose `details.cells` carry agent-fan-out progress. */
|
|
function evalAgentResult(events: EvalStatusEvent[], text = "") {
|
|
const details: EvalToolDetails = {
|
|
language: "python",
|
|
languages: ["python"],
|
|
cells: [
|
|
{
|
|
index: 0,
|
|
title: "Investigate",
|
|
code: "results = parallel([...])",
|
|
language: "python",
|
|
output: text,
|
|
status: "running",
|
|
statusEvents: events,
|
|
},
|
|
],
|
|
};
|
|
return { content: [{ type: "text" as const, text }], details };
|
|
}
|
|
|
|
function expectLive(component: ToolExecutionComponent): void {
|
|
expect(component.isTranscriptBlockFinalized()).toBe(false);
|
|
expect(component.getNativeScrollbackLiveRegionStart()).toBe(0);
|
|
}
|
|
|
|
function expectFinal(component: ToolExecutionComponent): void {
|
|
expect(component.isTranscriptBlockFinalized()).toBe(true);
|
|
expect(component.getNativeScrollbackLiveRegionStart()).toBeUndefined();
|
|
}
|
|
|
|
describe("eval tool transcript finalization", () => {
|
|
beforeAll(async () => {
|
|
resetSettingsForTest();
|
|
await Settings.init({ inMemory: true });
|
|
await initTheme();
|
|
});
|
|
|
|
it("keeps partial eval results in the native-scrollback live region", () => {
|
|
const component = makeEvalComponent();
|
|
|
|
component.updateResult(evalAgentResult([{ op: "agent", id: "a1", status: "running" }]), true);
|
|
|
|
expectLive(component);
|
|
});
|
|
|
|
it("moves the block out of the live region as soon as the eval result settles", () => {
|
|
const component = makeEvalComponent();
|
|
component.updateResult(evalAgentResult([{ op: "agent", id: "a1", status: "running" }]), true);
|
|
expectLive(component);
|
|
|
|
component.updateResult({ content: [{ type: "text" as const, text: "done\n" }] }, false);
|
|
|
|
expectFinal(component);
|
|
});
|
|
|
|
it("stays in the live region across agent-progress shape churn while partial", () => {
|
|
const component = makeEvalComponent();
|
|
|
|
component.updateResult(
|
|
evalAgentResult([{ op: "agent", id: "a1", status: "running", currentTool: "read" }]),
|
|
true,
|
|
);
|
|
expectLive(component);
|
|
|
|
component.updateResult(
|
|
evalAgentResult([
|
|
{ op: "agent", id: "a1", status: "running" },
|
|
{ op: "agent", id: "a2", status: "running", currentTool: "bash" },
|
|
]),
|
|
true,
|
|
);
|
|
|
|
expectLive(component);
|
|
});
|
|
});
|