Files
oh-my-pi/packages/coding-agent/src/core/python-executor-mapping.test.ts
T
can1357 1d414c1383 feat: added IPython-backed Python tool with streaming output and Jupyter integration
- Added IPython-backed Python tool with streaming output and image/JSON rendering.
- Implemented Jupyter kernel gateway integration with WebSocket communication.
- Added Python prelude with 30+ shell-like utility functions for file operations.
- Migrated environment variables from PI_ to OMP_ prefix with automatic migration.
- Added streaming output system with automatic spill-to-disk for large outputs.
- Reorganized settings interface into behavior, tools, display, voice, status, lsp, and exa tabs.
2026-01-18 19:32:27 +01:00

42 lines
1.4 KiB
TypeScript

import { describe, expect, it } from "bun:test";
import { executePythonWithKernel, type PythonKernelExecutor } from "./python-executor";
import type { KernelExecuteOptions, KernelExecuteResult } from "./python-kernel";
class FakeKernel implements PythonKernelExecutor {
constructor(
private result: KernelExecuteResult,
private onExecute: (options?: KernelExecuteOptions) => void = () => {},
) {}
async execute(_code: string, options?: KernelExecuteOptions): Promise<KernelExecuteResult> {
this.onExecute(options);
return this.result;
}
}
describe("executePythonWithKernel mapping", () => {
it("annotates timeout cancellations", async () => {
const kernel = new FakeKernel({ status: "ok", cancelled: true, timedOut: true, stdinRequested: false });
const result = await executePythonWithKernel(kernel, "sleep(10)", { timeout: 5000 });
expect(result.cancelled).toBe(true);
expect(result.exitCode).toBeUndefined();
expect(result.output).toContain("Command timed out after 5 seconds");
});
it("maps error status to non-zero exit code", async () => {
const kernel = new FakeKernel(
{ status: "error", cancelled: false, timedOut: false, stdinRequested: false },
(options) => {
options?.onChunk?.("traceback\n");
},
);
const result = await executePythonWithKernel(kernel, "1 / 0");
expect(result.exitCode).toBe(1);
expect(result.output).toContain("traceback");
expect(result.stdinRequested).toBe(false);
});
});