Files
oh-my-pi/packages/coding-agent/test/core/python-executor-streaming.test.ts
T
can1357 cf60e6df51 feat(coding-agent): implemented eval framework and replaced python tool
- Added a unified eval framework with parser grammar, backend interfaces, and JS/Python execution result types.
- Added eval tool docs and updated prompts for fenced cells, `eval.py`/`eval.js`, and fallback behavior.
- Replaced the built-in `python` tool with `eval` across registry, rendering, interactive modes, and tool settings.
- Migrated Python execution runtime from `src/ipy` to `src/eval/py`, renamed state fields, and removed legacy introspection.
- Refactored browser tooling from in-process VM helpers to worker-managed tab supervisors and protocol transport.
- Added eval parser fallback and JS tool-bridge tests, updated imports, and removed obsolete python-mode suites.
2026-04-30 18:08:37 +02:00

42 lines
1.6 KiB
TypeScript

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("Command timed out after 2 seconds");
});
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");
});
});