can1357
779ca4872b
- Removed Prettier configuration files (.prettierignore and .prettierrc) and migrated formatting to Biome. - Updated Biome configuration from version 2.3.11 to 2.3.12 and changed arrowParentheses rule from 'always' to 'asNeeded'. - Pinned @biomejs/biome dependency to exact version 2.3.12 in package.json and bun.lock. - Applied consistent arrow function formatting across 489 files by removing unnecessary parentheses around single parameters. - Removed blank lines after comment blocks and reorganized imports for consistency across the codebase.
105 lines
4.1 KiB
TypeScript
105 lines
4.1 KiB
TypeScript
import { afterEach, describe, expect, it } from "bun:test";
|
|
import * as fs from "node:fs/promises";
|
|
import * as path from "node:path";
|
|
import {
|
|
discoverPythonModules,
|
|
loadPythonModules,
|
|
type PythonModuleExecutor,
|
|
} from "@oh-my-pi/pi-coding-agent/ipy/modules";
|
|
import { TempDir } from "@oh-my-pi/pi-utils";
|
|
|
|
const fixturesDir = path.resolve(import.meta.dir, "../../test/fixtures/python-modules");
|
|
|
|
const readFixture = (name: string): Promise<string> => Bun.file(path.join(fixturesDir, name)).text();
|
|
|
|
const writeModule = async (dir: string, name: string, tag: string) => {
|
|
await fs.mkdir(dir, { recursive: true });
|
|
const base = await readFixture(name);
|
|
await Bun.write(path.join(dir, name), `${base}\n# ${tag}`);
|
|
};
|
|
|
|
describe("python modules", () => {
|
|
let tempRoot: TempDir | null = null;
|
|
|
|
afterEach(() => {
|
|
if (tempRoot) {
|
|
tempRoot.removeSync();
|
|
}
|
|
tempRoot = null;
|
|
});
|
|
|
|
it("discovers modules with project override and sorted order", async () => {
|
|
tempRoot = TempDir.createSync("@omp-python-modules-");
|
|
const homeDir = path.join(tempRoot.path(), "home");
|
|
const cwd = path.join(tempRoot.path(), "project");
|
|
|
|
await writeModule(path.join(homeDir, ".omp", "agent", "modules"), "alpha.py", "user-omp");
|
|
await writeModule(path.join(homeDir, ".pi", "agent", "modules"), "beta.py", "user-pi");
|
|
await writeModule(path.join(homeDir, ".pi", "agent", "modules"), "delta.py", "user-pi");
|
|
|
|
await writeModule(path.join(cwd, ".omp", "modules"), "alpha.py", "project-omp");
|
|
await writeModule(path.join(cwd, ".omp", "modules"), "beta.py", "project-omp");
|
|
await writeModule(path.join(cwd, ".pi", "modules"), "gamma.py", "project-pi");
|
|
|
|
const modules = await discoverPythonModules({ cwd, homeDir });
|
|
const names = modules.map(module => path.basename(module.path));
|
|
expect(names).toEqual(["alpha.py", "beta.py", "delta.py", "gamma.py"]);
|
|
expect(modules.map(module => ({ name: path.basename(module.path), source: module.source }))).toEqual([
|
|
{ name: "alpha.py", source: "project" },
|
|
{ name: "beta.py", source: "project" },
|
|
{ name: "delta.py", source: "user" },
|
|
{ name: "gamma.py", source: "project" },
|
|
]);
|
|
expect(modules.find(module => module.path.endsWith("alpha.py"))?.content).toContain("project-omp");
|
|
expect(modules.find(module => module.path.endsWith("delta.py"))?.content).toContain("user-pi");
|
|
});
|
|
|
|
it("loads modules in sorted order with silent execution", async () => {
|
|
tempRoot = TempDir.createSync("@omp-python-modules-");
|
|
const homeDir = path.join(tempRoot.path(), "home");
|
|
const cwd = path.join(tempRoot.path(), "project");
|
|
|
|
await writeModule(path.join(homeDir, ".omp", "agent", "modules"), "beta.py", "user-omp");
|
|
await writeModule(path.join(homeDir, ".omp", "agent", "modules"), "alpha.py", "user-omp");
|
|
|
|
const calls: Array<{ name: string; options?: { silent?: boolean; storeHistory?: boolean } }> = [];
|
|
const executor: PythonModuleExecutor = {
|
|
execute: async (code: string, options?: { silent?: boolean; storeHistory?: boolean }) => {
|
|
const name = code.includes("def alpha") ? "alpha" : "beta";
|
|
calls.push({ name, options });
|
|
return { status: "ok", cancelled: false };
|
|
},
|
|
};
|
|
|
|
await loadPythonModules(executor, { cwd, homeDir });
|
|
expect(calls.map(call => call.name)).toEqual(["alpha", "beta"]);
|
|
for (const call of calls) {
|
|
expect(call.options).toEqual({ silent: true, storeHistory: false });
|
|
}
|
|
});
|
|
|
|
it("fails fast when a module fails to execute", async () => {
|
|
tempRoot = TempDir.createSync("@omp-python-modules-");
|
|
const homeDir = path.join(tempRoot.path(), "home");
|
|
const cwd = path.join(tempRoot.path(), "project");
|
|
|
|
await writeModule(path.join(homeDir, ".omp", "agent", "modules"), "alpha.py", "user-omp");
|
|
await writeModule(path.join(cwd, ".omp", "modules"), "beta.py", "project-omp");
|
|
|
|
const executor: PythonModuleExecutor = {
|
|
execute: async (code: string) => {
|
|
if (code.includes("def beta")) {
|
|
return {
|
|
status: "error",
|
|
cancelled: false,
|
|
error: { name: "Error", value: "boom", traceback: [] },
|
|
};
|
|
}
|
|
return { status: "ok", cancelled: false };
|
|
},
|
|
};
|
|
|
|
await expect(loadPythonModules(executor, { cwd, homeDir })).rejects.toThrow("Failed to load Python module");
|
|
});
|
|
});
|