can1357
b2c94bef1a
- Created new @oh-my-pi/pi-utils workspace package with shared utilities for logging, process management, stream handling, and temporary directory management. - Migrated all packages to use centralized logger from @oh-my-pi/pi-utils instead of local winston implementations. - Replaced custom process spawning and stream reading implementations with standardized cspawn and readLines utilities across all modules. - Converted synchronous file operations and process spawning to async patterns using Bun shell syntax and fs/promises. - Added streaming edit abort functionality with configurable setting to abort on patch preview failures. - Updated test framework from vitest to bun:test across all test suites.
67 lines
1.9 KiB
TypeScript
67 lines
1.9 KiB
TypeScript
import { describe, expect, it, vi } from "bun:test";
|
|
import { createTempDirSync } from "@oh-my-pi/pi-utils";
|
|
import * as pythonExecutor from "../../src/core/python-executor";
|
|
import type { ToolSession } from "../../src/core/tools/index";
|
|
import { PythonTool } from "../../src/core/tools/python";
|
|
|
|
function createSession(cwd: string): ToolSession {
|
|
return {
|
|
cwd,
|
|
hasUI: false,
|
|
getSessionFile: () => "session-file",
|
|
getSessionSpawns: () => "*",
|
|
settings: {
|
|
getImageAutoResize: () => true,
|
|
getLspFormatOnWrite: () => true,
|
|
getLspDiagnosticsOnWrite: () => true,
|
|
getLspDiagnosticsOnEdit: () => false,
|
|
getEditFuzzyMatch: () => true,
|
|
getGitToolEnabled: () => true,
|
|
getBashInterceptorEnabled: () => true,
|
|
getBashInterceptorSimpleLsEnabled: () => true,
|
|
getBashInterceptorRules: () => [],
|
|
getPythonToolMode: () => "ipy-only",
|
|
getPythonKernelMode: () => "per-call",
|
|
},
|
|
};
|
|
}
|
|
|
|
describe("python tool execution", () => {
|
|
it("passes kernel options from settings and args", async () => {
|
|
const tempDir = createTempDirSync("@python-tool-");
|
|
const executeSpy = vi.spyOn(pythonExecutor, "executePython").mockResolvedValue({
|
|
output: "ok",
|
|
exitCode: 0,
|
|
cancelled: false,
|
|
truncated: false,
|
|
displayOutputs: [],
|
|
stdinRequested: false,
|
|
});
|
|
|
|
const tool = new PythonTool(createSession(tempDir.path));
|
|
const result = await tool.execute(
|
|
"call-id",
|
|
{ code: "print('hi')", timeout: 5, workdir: tempDir.path, reset: true },
|
|
undefined,
|
|
undefined,
|
|
undefined,
|
|
);
|
|
|
|
expect(executeSpy).toHaveBeenCalledWith(
|
|
"print('hi')",
|
|
expect.objectContaining({
|
|
cwd: tempDir.path,
|
|
timeout: 5000,
|
|
sessionId: `session:session-file:workdir:${tempDir.path}`,
|
|
kernelMode: "per-call",
|
|
reset: true,
|
|
}),
|
|
);
|
|
const text = result.content.find((item) => item.type === "text")?.text;
|
|
expect(text).toBe("ok");
|
|
|
|
executeSpy.mockRestore();
|
|
tempDir.remove();
|
|
});
|
|
});
|