wehub-resource-sync
50 lines
1.6 KiB
TypeScript
50 lines
1.6 KiB
TypeScript
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 {
|
|
private result: KernelExecuteResult;
|
|
private onExecute?: (options?: KernelExecuteOptions) => void;
|
|
|
|
constructor(result: KernelExecuteResult, onExecute?: (options?: KernelExecuteOptions) => void) {
|
|
this.result = result;
|
|
this.onExecute = onExecute;
|
|
}
|
|
|
|
async execute(_code: string, options?: KernelExecuteOptions): Promise<KernelExecuteResult> {
|
|
this.onExecute?.(options);
|
|
return this.result;
|
|
}
|
|
}
|
|
|
|
describe("executePythonWithKernel result mapping", () => {
|
|
it("adds timeout annotation when cancelled", async () => {
|
|
const kernel = new FakeKernel({
|
|
status: "ok",
|
|
cancelled: true,
|
|
timedOut: true,
|
|
stdinRequested: false,
|
|
});
|
|
|
|
const result = await executePythonWithKernel(kernel, "sleep()", { timeoutMs: 5000 });
|
|
|
|
expect(result.exitCode).toBeUndefined();
|
|
expect(result.cancelled).toBe(true);
|
|
expect(result.output).toContain("eval cell timed out after 5s");
|
|
});
|
|
|
|
it("maps kernel error status to exit code 1", async () => {
|
|
const kernel = new FakeKernel(
|
|
{ status: "error", cancelled: false, timedOut: false, stdinRequested: false },
|
|
options => {
|
|
options?.onChunk?.("Traceback...\n");
|
|
},
|
|
);
|
|
|
|
const result = await executePythonWithKernel(kernel, "raise ValueError('boom')");
|
|
|
|
expect(result.exitCode).toBe(1);
|
|
expect(result.output).toContain("Traceback");
|
|
});
|
|
});
|