chore: import upstream snapshot with attribution

2026-07-13 12:22:59 +08:00
commit 069dc98253
5392 changed files with 1751174 additions and 0 deletions
@@ -0,0 +1,41 @@
+import { describe, expect, it } from "bun:test";
+import { executePythonWithKernel } from "@oh-my-pi/pi-coding-agent/eval/py/executor";
+import { DEFAULT_MAX_BYTES } from "@oh-my-pi/pi-coding-agent/session/streaming-output";
+import { FakeKernel } from "./helpers";
+
+describe("executePythonWithKernel streaming", () => {
+	it("truncates large output and tracks totals", async () => {
+		const largeOutput = "a".repeat(DEFAULT_MAX_BYTES + 128);
+		const kernel = new FakeKernel(
+			{ status: "ok", cancelled: false, timedOut: false, stdinRequested: false },
+			options => options?.onChunk?.(largeOutput),
+		);
+
+		const result = await executePythonWithKernel(kernel, "print('hi')");
+
+		expect(result.truncated).toBe(true);
+		expect(result.output.length).toBeLessThan(largeOutput.length);
+		expect(result.totalBytes).toBeGreaterThan(result.outputBytes);
+	});
+
+	it("annotates timed out runs", async () => {
+		const kernel = new FakeKernel({ status: "ok", cancelled: true, timedOut: true, stdinRequested: false }, () => {});
+
+		const result = await executePythonWithKernel(kernel, "sleep", { timeoutMs: 2000 });
+
+		expect(result.cancelled).toBe(true);
+		expect(result.exitCode).toBeUndefined();
+		expect(result.output).toContain("eval cell timed out after 2s");
+	});
+
+	it("sanitizes ANSI and carriage returns", async () => {
+		const kernel = new FakeKernel(
+			{ status: "ok", cancelled: false, timedOut: false, stdinRequested: false },
+			options => options?.onChunk?.("\u001b[31mhello\r\n"),
+		);
+
+		const result = await executePythonWithKernel(kernel, "print('hello')");
+
+		expect(result.output).toBe("hello\n");
+	});
+});