diff options
Diffstat (limited to 'packages/api/tests')
| -rw-r--r-- | packages/api/tests/agent-manager.test.ts | 29 | ||||
| -rw-r--r-- | packages/api/tests/routes.test.ts | 8 |
2 files changed, 37 insertions, 0 deletions
diff --git a/packages/api/tests/agent-manager.test.ts b/packages/api/tests/agent-manager.test.ts index 014022a..970ac1d 100644 --- a/packages/api/tests/agent-manager.test.ts +++ b/packages/api/tests/agent-manager.test.ts @@ -419,6 +419,14 @@ vi.mock("@dispatch/core", () => ({ execute: async () => "mock", }; }, + createSearchCodeTool(_wd: string) { + return { + name: "search_code", + description: "search code", + parameters: { _type: "z.ZodObject", shape: {} }, + execute: async () => "mock", + }; + }, createYoutubeTranscribeTool() { return { name: "youtube_transcribe", @@ -1441,6 +1449,27 @@ describe("AgentManager", () => { }); }); + describe("search_code permission gating", () => { + // Reuses the parent-path tool construction to confirm the perm flag wires + // the search_code tool on/off correctly. + async function toolsForPerms(tabId: string, perms: Record<string, string>): Promise<string[]> { + for (const [k, v] of Object.entries(perms)) setFakeSetting(k, v); + const manager = new AgentManager(); + await manager.processMessage(tabId, "go"); + return constructedAgents.at(-1)?.toolNames ?? []; + } + + it("grants search_code when perm_search_code is allowed", async () => { + const tools = await toolsForPerms("tab-cs-on", { perm_search_code: "allow" }); + expect(tools).toContain("search_code"); + }); + + it("omits search_code when perm_search_code is not allowed", async () => { + const tools = await toolsForPerms("tab-cs-off", {}); + expect(tools).not.toContain("search_code"); + }); + }); + // ─── Usage side-channel persistence ────────────────────────────── // // `usage` AgentEvents (one per LLM round-trip) are persisted as invisible diff --git a/packages/api/tests/routes.test.ts b/packages/api/tests/routes.test.ts index a8db5ce..c85d43d 100644 --- a/packages/api/tests/routes.test.ts +++ b/packages/api/tests/routes.test.ts @@ -283,6 +283,14 @@ vi.mock("@dispatch/core", () => ({ execute: async () => "mock", }; }, + createSearchCodeTool(_wd: string) { + return { + name: "search_code", + description: "search code", + parameters: { _type: "z.ZodObject", shape: {} }, + execute: async () => "mock", + }; + }, createYoutubeTranscribeTool() { return { name: "youtube_transcribe", |
