mirror of https://github.com/openclaw/openclaw.git
105 lines
2.8 KiB
TypeScript
105 lines
2.8 KiB
TypeScript
import type { AgentTool } from "@mariozechner/pi-agent-core";
|
|
import { Type } from "@sinclair/typebox";
|
|
import { describe, expect, it } from "vitest";
|
|
import {
|
|
buildEmbeddedSandboxInfo,
|
|
splitSdkTools,
|
|
} from "./pi-embedded-runner.js";
|
|
import type { SandboxContext } from "./sandbox.js";
|
|
|
|
describe("buildEmbeddedSandboxInfo", () => {
|
|
it("returns undefined when sandbox is missing", () => {
|
|
expect(buildEmbeddedSandboxInfo()).toBeUndefined();
|
|
});
|
|
|
|
it("maps sandbox context into prompt info", () => {
|
|
const sandbox = {
|
|
enabled: true,
|
|
sessionKey: "session:test",
|
|
workspaceDir: "/tmp/clawdbot-sandbox",
|
|
agentWorkspaceDir: "/tmp/clawdbot-workspace",
|
|
workspaceAccess: "none",
|
|
containerName: "clawdbot-sbx-test",
|
|
containerWorkdir: "/workspace",
|
|
docker: {
|
|
image: "clawdbot-sandbox:bookworm-slim",
|
|
containerPrefix: "clawdbot-sbx-",
|
|
workdir: "/workspace",
|
|
readOnlyRoot: true,
|
|
tmpfs: ["/tmp"],
|
|
network: "none",
|
|
user: "1000:1000",
|
|
capDrop: ["ALL"],
|
|
env: { LANG: "C.UTF-8" },
|
|
},
|
|
tools: {
|
|
allow: ["bash"],
|
|
deny: ["browser"],
|
|
},
|
|
browser: {
|
|
controlUrl: "http://localhost:9222",
|
|
noVncUrl: "http://localhost:6080",
|
|
containerName: "clawdbot-sbx-browser-test",
|
|
},
|
|
} satisfies SandboxContext;
|
|
|
|
expect(buildEmbeddedSandboxInfo(sandbox)).toEqual({
|
|
enabled: true,
|
|
workspaceDir: "/tmp/clawdbot-sandbox",
|
|
workspaceAccess: "none",
|
|
agentWorkspaceMount: undefined,
|
|
browserControlUrl: "http://localhost:9222",
|
|
browserNoVncUrl: "http://localhost:6080",
|
|
});
|
|
});
|
|
});
|
|
|
|
function createStubTool(name: string): AgentTool {
|
|
return {
|
|
name,
|
|
label: name,
|
|
description: "",
|
|
parameters: Type.Object({}),
|
|
execute: async () => ({ content: [], details: {} }),
|
|
};
|
|
}
|
|
|
|
describe("splitSdkTools", () => {
|
|
const tools = [
|
|
createStubTool("read"),
|
|
createStubTool("bash"),
|
|
createStubTool("edit"),
|
|
createStubTool("write"),
|
|
createStubTool("browser"),
|
|
];
|
|
|
|
it("routes built-ins to custom tools when sandboxed", () => {
|
|
const { builtInTools, customTools } = splitSdkTools({
|
|
tools,
|
|
sandboxEnabled: true,
|
|
});
|
|
expect(builtInTools).toEqual([]);
|
|
expect(customTools.map((tool) => tool.name)).toEqual([
|
|
"read",
|
|
"bash",
|
|
"edit",
|
|
"write",
|
|
"browser",
|
|
]);
|
|
});
|
|
|
|
it("keeps built-ins as SDK tools when not sandboxed", () => {
|
|
const { builtInTools, customTools } = splitSdkTools({
|
|
tools,
|
|
sandboxEnabled: false,
|
|
});
|
|
expect(builtInTools.map((tool) => tool.name)).toEqual([
|
|
"read",
|
|
"bash",
|
|
"edit",
|
|
"write",
|
|
]);
|
|
expect(customTools.map((tool) => tool.name)).toEqual(["browser"]);
|
|
});
|
|
});
|