diff options
Diffstat (limited to 'packages/api/tests')
| -rw-r--r-- | packages/api/tests/agent-manager.test.ts | 23 | ||||
| -rw-r--r-- | packages/api/tests/routes.test.ts | 23 |
2 files changed, 46 insertions, 0 deletions
diff --git a/packages/api/tests/agent-manager.test.ts b/packages/api/tests/agent-manager.test.ts index 56cb818..2111b0e 100644 --- a/packages/api/tests/agent-manager.test.ts +++ b/packages/api/tests/agent-manager.test.ts @@ -18,6 +18,15 @@ vi.mock("@dispatch/core", () => ({ yield { type: "status", status: "idle" } as const; } }, + PermissionService: class MockPermissionService { + ask(_request: unknown, _rulesets: unknown[]) { + return Promise.resolve("once"); + } + reply(_id: string, _reply: unknown) {} + getPending() { + return []; + } + }, createReadFileTool(_wd: string): ToolDefinition { return { name: "read_file", @@ -42,6 +51,20 @@ vi.mock("@dispatch/core", () => ({ execute: async () => ["file1.ts"], }; }, + createRunShellTool(_wd: string): ToolDefinition { + return { + name: "run_shell", + description: "run shell command", + parameters: { _type: "z.ZodObject", shape: {} } as unknown as ToolDefinition["parameters"], + execute: async () => ({ stdout: "", stderr: "", exitCode: 0 }), + }; + }, + loadConfig(_dir: string) { + return { permissions: {} }; + }, + configToRuleset(_config: unknown) { + return []; + }, })); // Import after mock is defined (Vitest hoists vi.mock automatically) diff --git a/packages/api/tests/routes.test.ts b/packages/api/tests/routes.test.ts index 9f852ee..87ff436 100644 --- a/packages/api/tests/routes.test.ts +++ b/packages/api/tests/routes.test.ts @@ -19,6 +19,15 @@ vi.mock("@dispatch/core", () => ({ yield { type: "status", status: "idle" } as const; } }, + PermissionService: class MockPermissionService { + ask(_request: unknown, _rulesets: unknown[]) { + return Promise.resolve("once"); + } + reply(_id: string, _reply: unknown) {} + getPending() { + return []; + } + }, createReadFileTool(_wd: string): ToolDefinition { return { name: "read_file", @@ -43,6 +52,20 @@ vi.mock("@dispatch/core", () => ({ execute: async () => ["file1.ts"], }; }, + createRunShellTool(_wd: string): ToolDefinition { + return { + name: "run_shell", + description: "run shell command", + parameters: { _type: "z.ZodObject", shape: {} } as unknown as ToolDefinition["parameters"], + execute: async () => ({ stdout: "", stderr: "", exitCode: 0 }), + }; + }, + loadConfig(_dir: string) { + return { permissions: {} }; + }, + configToRuleset(_config: unknown) { + return []; + }, })); const { app } = await import("../src/app.js"); |