can1357
2e33e436b4
- Added `sanitizeText` function to pi-natives that strips ANSI escape sequences, removes control characters and lone surrogates, and normalizes line endings. - Moved `sanitizeText` function from `@oh-my-pi/pi-utils` to `@oh-my-pi/pi-natives` for better code organization and native performance. - Added line length clamping (4000 characters) to bash and Python execution output to prevent excessively long lines. - Replaced internal `#normalizeOutput` methods with `sanitizeText` utility function in bash and Python execution components. - Fixed bash interactive tool to gracefully handle malformed output chunks by normalizing them with `sanitizeText`. - Simplified documentation by removing WASM terminology from package descriptions and comments.
47 lines
1.5 KiB
TypeScript
47 lines
1.5 KiB
TypeScript
import { describe, expect, it } from "bun:test";
|
|
import { getThemeByName } from "@oh-my-pi/pi-coding-agent/modes/theme/theme";
|
|
import { pythonToolRenderer } from "@oh-my-pi/pi-coding-agent/tools/python";
|
|
import { sanitizeText } from "@oh-my-pi/pi-natives";
|
|
|
|
describe("pythonToolRenderer", () => {
|
|
it("renders truncated output when collapsed and full output when expanded", async () => {
|
|
const theme = await getThemeByName("dark");
|
|
expect(theme).toBeDefined();
|
|
const uiTheme = theme!;
|
|
|
|
const fullOutput = ["line 1", "line 2", "line 3", "line 4"].join("\n");
|
|
|
|
const result = {
|
|
content: [{ type: "text", text: fullOutput }],
|
|
details: {
|
|
cells: [
|
|
{
|
|
index: 0,
|
|
title: "run",
|
|
code: "print('hello')",
|
|
output: fullOutput,
|
|
status: "complete" as const,
|
|
durationMs: 12,
|
|
},
|
|
],
|
|
},
|
|
};
|
|
|
|
const collapsed = pythonToolRenderer.renderResult(
|
|
result,
|
|
{ expanded: false, isPartial: false, renderContext: { previewLines: 2 } },
|
|
uiTheme,
|
|
);
|
|
const collapsedLines = sanitizeText(collapsed.render(80).join("\n"));
|
|
expect(collapsedLines).toContain("line 4");
|
|
expect(collapsedLines).not.toContain("line 1");
|
|
expect(collapsedLines).toContain("more lines");
|
|
|
|
const expanded = pythonToolRenderer.renderResult(result, { expanded: true, isPartial: false }, uiTheme);
|
|
const expandedLines = sanitizeText(expanded.render(80).join("\n"));
|
|
expect(expandedLines).toContain("line 1");
|
|
expect(expandedLines).toContain("line 4");
|
|
expect(expandedLines).not.toContain("more lines");
|
|
});
|
|
});
|