wehub-resource-sync
107 lines
4.1 KiB
TypeScript
107 lines
4.1 KiB
TypeScript
import { beforeAll, describe, expect, it } from "bun:test";
|
|
import { getThemeByName, initTheme } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import { previewWindowRows } from "@oh-my-pi/pi-coding-agent/tools/render-utils";
|
|
import { sshToolRenderer } from "@oh-my-pi/pi-coding-agent/tools/ssh";
|
|
import { sanitizeText } from "@oh-my-pi/pi-utils";
|
|
|
|
beforeAll(async () => {
|
|
await initTheme();
|
|
});
|
|
|
|
describe("sshToolRenderer", () => {
|
|
it("keeps the status header on one line when the command spans multiple lines", async () => {
|
|
const uiTheme = (await getThemeByName("dark"))!;
|
|
expect(uiTheme).toBeDefined();
|
|
const command = ["set -e", "cat > /etc/apt/sources.list <<'EOF'", "# mirrors", "EOF"].join("\n");
|
|
const component = sshToolRenderer.renderResult(
|
|
{ content: [{ type: "text", text: "" }] },
|
|
{ expanded: false, isPartial: false },
|
|
uiTheme,
|
|
{ host: "router", command },
|
|
);
|
|
const rendered = component.render(120);
|
|
const sanitized = rendered.map(line => sanitizeText(line));
|
|
// First visible row is the status header, and it MUST remain a single line.
|
|
// It carries the host but NOT the command — the command lives in the body.
|
|
const header = sanitized[0]!;
|
|
expect(header).toContain("SSH");
|
|
expect(header).toContain("[router]");
|
|
expect(header).not.toContain("set -e");
|
|
expect(header).not.toContain("EOF");
|
|
// Every command line still appears, inside the framed body.
|
|
const body = sanitized.slice(1).join("\n");
|
|
expect(body).toContain("$ set -e");
|
|
expect(body).toContain("cat > /etc/apt/sources.list <<'EOF'");
|
|
expect(body).toContain("# mirrors");
|
|
expect(body).toContain("EOF");
|
|
});
|
|
|
|
it("keeps the pending-call header on one line for multiline commands", async () => {
|
|
const uiTheme = (await getThemeByName("dark"))!;
|
|
expect(uiTheme).toBeDefined();
|
|
const command = "set -e\ndo-something";
|
|
const component = sshToolRenderer.renderCall(
|
|
{ host: "router", command },
|
|
{ expanded: false, isPartial: true },
|
|
uiTheme,
|
|
);
|
|
const rendered = component.render(120);
|
|
const sanitized = rendered.map(line => sanitizeText(line));
|
|
const header = sanitized[0]!;
|
|
expect(header).toContain("SSH");
|
|
expect(header).toContain("[router]");
|
|
expect(header).not.toContain("set -e");
|
|
expect(header).not.toContain("do-something");
|
|
const body = sanitized.slice(1).join("\n");
|
|
expect(body).toContain("$ set -e");
|
|
expect(body).toContain("do-something");
|
|
});
|
|
|
|
it("keeps partial results pending until the final SSH result", async () => {
|
|
const uiTheme = (await getThemeByName("dark"))!;
|
|
expect(uiTheme).toBeDefined();
|
|
const renderHeader = (isPartial: boolean) =>
|
|
sanitizeText(
|
|
sshToolRenderer
|
|
.renderResult(
|
|
{ content: [{ type: "text", text: isPartial ? "streaming" : "done" }] },
|
|
{ expanded: false, isPartial },
|
|
uiTheme,
|
|
{ host: "sccpu", command: "uptime" },
|
|
)
|
|
.render(120)[0]!,
|
|
);
|
|
|
|
expect(renderHeader(true)).toContain("⏳ SSH: [sccpu]");
|
|
expect(renderHeader(false)).toContain("⇄ SSH: [sccpu]");
|
|
});
|
|
it("renders the collapsed command as a viewport tail window in every state — no stream→final expansion", async () => {
|
|
const uiTheme = (await getThemeByName("dark"))!;
|
|
expect(uiTheme).toBeDefined();
|
|
const total = previewWindowRows() + 5;
|
|
const command = Array.from({ length: total }, (_, i) => `step_${i}`).join("\n");
|
|
const render = (opts: { expanded: boolean; isPartial: boolean }) =>
|
|
sanitizeText(
|
|
sshToolRenderer
|
|
.renderResult({ content: [{ type: "text", text: "" }] }, opts, uiTheme, { host: "router", command })
|
|
.render(120)
|
|
.join("\n"),
|
|
);
|
|
|
|
// Identical tail window streaming and final: the end stays visible, the
|
|
// head is elided behind an "earlier lines" marker. Only ctrl+o uncaps.
|
|
for (const rendered of [
|
|
render({ expanded: false, isPartial: true }),
|
|
render({ expanded: false, isPartial: false }),
|
|
]) {
|
|
expect(rendered).toContain(`step_${total - 1}`);
|
|
expect(rendered).toContain("earlier line");
|
|
expect(rendered).not.toContain("step_0");
|
|
}
|
|
|
|
const expandedFinal = render({ expanded: true, isPartial: false });
|
|
expect(expandedFinal).toContain("$ step_0");
|
|
expect(expandedFinal).not.toContain("earlier line");
|
|
});
|
|
});
|