From 9b487ca368caaa99cb3acc71d7bd034125334658 Mon Sep 17 00:00:00 2001
From: JackChen <jack_ai@qq.com>
Date: Wed, 15 Apr 2026 15:10:02 +0800
Subject: [PATCH] test: add disallowedTools blocking custom tool test

---
 tests/orchestrator.test.ts | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/tests/orchestrator.test.ts b/tests/orchestrator.test.ts
index f7b5ce9..0210647 100644
--- a/tests/orchestrator.test.ts
+++ b/tests/orchestrator.test.ts
@@ -205,6 +205,30 @@ describe('OpenMultiAgent', () => {
       expect(toolNames).not.toContain('bash')
     })
 
+    it('customTools can be blocked by disallowedTools', async () => {
+      mockAdapterResponses = ['done']
+
+      const { z } = await import('zod')
+      const { defineTool } = await import('../src/tool/framework.js')
+
+      const myTool = defineTool({
+        name: 'my_custom_tool',
+        description: 'A custom tool for testing',
+        inputSchema: z.object({ query: z.string() }),
+        execute: async ({ query }) => ({ data: query }),
+      })
+
+      const oma = new OpenMultiAgent({ defaultModel: 'mock-model' })
+
+      await oma.runAgent(
+        { ...agentConfig('solo'), customTools: [myTool], disallowedTools: ['my_custom_tool'] },
+        'test',
+      )
+
+      const toolNames = capturedChatOptions[0]?.tools?.map(t => t.name) ?? []
+      expect(toolNames).not.toContain('my_custom_tool')
+    })
+
     it('fires onProgress events', async () => {
       mockAdapterResponses = ['done']