chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,41 @@
|
||||
import { describe, expect, it } from "bun:test";
|
||||
import { executePythonWithKernel, type PythonKernelExecutor } from "@oh-my-pi/pi-coding-agent/eval/py/executor";
|
||||
import type { KernelExecuteOptions, KernelExecuteResult } from "@oh-my-pi/pi-coding-agent/eval/py/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)", { timeoutMs: 5000 });
|
||||
|
||||
expect(result.cancelled).toBe(true);
|
||||
expect(result.exitCode).toBeUndefined();
|
||||
expect(result.output).toContain("eval cell timed out after 5s");
|
||||
});
|
||||
|
||||
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);
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user