@didim365/agent-cli-core 0.1.1 → 0.1.3

package/dist/src/tools/mcp-tool.test.js

@@ -1,657 +0,0 @@
-/**
- * @license
- * Copyright 2025 Google LLC
- * SPDX-License-Identifier: Apache-2.0
- */
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
-import { safeJsonStringify } from '../utils/safeJsonStringify.js';
-import { DiscoveredMCPTool, generateValidName } from './mcp-tool.js'; // Added getStringifiedResultForDisplay
-import { ToolConfirmationOutcome } from './tools.js'; // Added ToolConfirmationOutcome
-import { ToolErrorType } from './tool-error.js';
-import { createMockMessageBus, getMockMessageBusInstance, } from '../test-utils/mock-message-bus.js';
-// Mock @google/genai mcpToTool and CallableTool
-// We only need to mock the parts of CallableTool that DiscoveredMCPTool uses.
-const mockCallTool = vi.fn();
-const mockToolMethod = vi.fn();
-const mockCallableToolInstance = {
-    tool: mockToolMethod, // Not directly used by DiscoveredMCPTool instance methods
-    callTool: mockCallTool,
-    // Add other methods if DiscoveredMCPTool starts using them
-};
-const createSdkResponse = (toolName, response) => [
-    {
-        functionResponse: {
-            name: toolName,
-            response,
-        },
-    },
-];
-describe('generateValidName', () => {
-    it('should return a valid name for a simple function', () => {
-        expect(generateValidName('myFunction')).toBe('myFunction');
-    });
-    it('should replace invalid characters with underscores', () => {
-        expect(generateValidName('invalid-name with spaces')).toBe('invalid-name_with_spaces');
-    });
-    it('should truncate long names', () => {
-        expect(generateValidName('x'.repeat(80))).toBe('xxxxxxxxxxxxxxxxxxxxxxxxxxxx___xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx');
-    });
-    it('should handle names with only invalid characters', () => {
-        expect(generateValidName('!@#$%^&*()')).toBe('__________');
-    });
-    it.each([
-        { length: 63, expected: 63, description: 'exactly 63 characters' },
-        { length: 64, expected: 63, description: 'exactly 64 characters' },
-        { length: 80, expected: 63, description: 'longer than 64 characters' },
-    ])('should handle names that are $description long', ({ length, expected }) => {
-        expect(generateValidName('a'.repeat(length)).length).toBe(expected);
-    });
-});
-describe('DiscoveredMCPTool', () => {
-    const serverName = 'mock-mcp-server';
-    const serverToolName = 'actual-server-tool-name';
-    const baseDescription = 'A test MCP tool.';
-    const inputSchema = {
-        type: 'object',
-        properties: { param: { type: 'string' } },
-        required: ['param'],
-    };
-    let tool;
-    beforeEach(() => {
-        mockCallTool.mockClear();
-        mockToolMethod.mockClear();
-        const bus = createMockMessageBus();
-        getMockMessageBusInstance(bus).defaultToolDecision = 'ask_user';
-        tool = new DiscoveredMCPTool(mockCallableToolInstance, serverName, serverToolName, baseDescription, inputSchema, bus);
-        // Clear allowlist before each relevant test, especially for shouldConfirmExecute
-        const invocation = tool.build({ param: 'mock' });
-        invocation.constructor.allowlist.clear();
-    });
-    afterEach(() => {
-        vi.restoreAllMocks();
-    });
-    describe('constructor', () => {
-        it('should set properties correctly', () => {
-            expect(tool.name).toBe(serverToolName);
-            expect(tool.schema.name).toBe(serverToolName);
-            expect(tool.schema.description).toBe(baseDescription);
-            expect(tool.schema.parameters).toBeUndefined();
-            expect(tool.schema.parametersJsonSchema).toEqual(inputSchema);
-            expect(tool.serverToolName).toBe(serverToolName);
-        });
-    });
-    describe('execute', () => {
-        it('should call mcpTool.callTool with correct parameters and format display output', async () => {
-            const params = { param: 'testValue' };
-            const mockToolSuccessResultObject = {
-                success: true,
-                details: 'executed',
-            };
-            const mockFunctionResponseContent = [
-                {
-                    type: 'text',
-                    text: JSON.stringify(mockToolSuccessResultObject),
-                },
-            ];
-            const mockMcpToolResponseParts = [
-                {
-                    functionResponse: {
-                        name: serverToolName,
-                        response: { content: mockFunctionResponseContent },
-                    },
-                },
-            ];
-            mockCallTool.mockResolvedValue(mockMcpToolResponseParts);
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(mockCallTool).toHaveBeenCalledWith([
-                { name: serverToolName, args: params },
-            ]);
-            const stringifiedResponseContent = JSON.stringify(mockToolSuccessResultObject);
-            expect(toolResult.llmContent).toEqual([
-                { text: stringifiedResponseContent },
-            ]);
-            expect(toolResult.returnDisplay).toBe(stringifiedResponseContent);
-        });
-        it('should handle empty result from getStringifiedResultForDisplay', async () => {
-            const params = { param: 'testValue' };
-            const mockMcpToolResponsePartsEmpty = [];
-            mockCallTool.mockResolvedValue(mockMcpToolResponsePartsEmpty);
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.returnDisplay).toBe('```json\n[]\n```');
-            expect(toolResult.llmContent).toEqual([
-                { text: '[Error: Could not parse tool response]' },
-            ]);
-        });
-        it('should propagate rejection if mcpTool.callTool rejects', async () => {
-            const params = { param: 'failCase' };
-            const expectedError = new Error('MCP call failed');
-            mockCallTool.mockRejectedValue(expectedError);
-            const invocation = tool.build(params);
-            await expect(invocation.execute(new AbortController().signal)).rejects.toThrow(expectedError);
-        });
-        it.each([
-            { isErrorValue: true, description: 'true (bool)' },
-            { isErrorValue: 'true', description: '"true" (str)' },
-        ])('should return a structured error if MCP tool reports an error', async ({ isErrorValue }) => {
-            const tool = new DiscoveredMCPTool(mockCallableToolInstance, serverName, serverToolName, baseDescription, inputSchema, createMockMessageBus(), undefined, undefined, undefined, undefined, undefined);
-            const params = { param: 'isErrorTrueCase' };
-            const functionCall = {
-                name: serverToolName,
-                args: params,
-            };
-            const errorResponse = { isError: isErrorValue };
-            const mockMcpToolResponseParts = [
-                {
-                    functionResponse: {
-                        name: serverToolName,
-                        response: { error: errorResponse },
-                    },
-                },
-            ];
-            mockCallTool.mockResolvedValue(mockMcpToolResponseParts);
-            const expectedErrorMessage = `MCP tool '${serverToolName}' reported tool error for function call: ${safeJsonStringify(functionCall)} with response: ${safeJsonStringify(mockMcpToolResponseParts)}`;
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.MCP_TOOL_ERROR);
-            expect(result.llmContent).toBe(expectedErrorMessage);
-            expect(result.returnDisplay).toContain(`Error: MCP tool '${serverToolName}' reported an error.`);
-        });
-        it('should return a structured error if MCP tool reports a top-level isError (spec compliant)', async () => {
-            const tool = new DiscoveredMCPTool(mockCallableToolInstance, serverName, serverToolName, baseDescription, inputSchema, createMockMessageBus(), undefined, undefined, undefined, undefined, undefined);
-            const params = { param: 'isErrorTopLevelCase' };
-            const functionCall = {
-                name: serverToolName,
-                args: params,
-            };
-            // Spec compliant error response: { isError: true } at the top level of content (or response object in this mapping)
-            const errorResponse = { isError: true };
-            const mockMcpToolResponseParts = [
-                {
-                    functionResponse: {
-                        name: serverToolName,
-                        response: errorResponse,
-                    },
-                },
-            ];
-            mockCallTool.mockResolvedValue(mockMcpToolResponseParts);
-            const expectedErrorMessage = `MCP tool '${serverToolName}' reported tool error for function call: ${safeJsonStringify(functionCall)} with response: ${safeJsonStringify(mockMcpToolResponseParts)}`;
-            const invocation = tool.build(params);
-            const result = await invocation.execute(new AbortController().signal);
-            expect(result.error?.type).toBe(ToolErrorType.MCP_TOOL_ERROR);
-            expect(result.llmContent).toBe(expectedErrorMessage);
-            expect(result.returnDisplay).toContain(`Error: MCP tool '${serverToolName}' reported an error.`);
-        });
-        it.each([
-            { isErrorValue: false, description: 'false (bool)' },
-            { isErrorValue: 'false', description: '"false" (str)' },
-        ])('should consider a ToolResult with isError ${description} to be a success', async ({ isErrorValue }) => {
-            const tool = new DiscoveredMCPTool(mockCallableToolInstance, serverName, serverToolName, baseDescription, inputSchema, createMockMessageBus(), undefined, undefined, undefined, undefined, undefined);
-            const params = { param: 'isErrorFalseCase' };
-            const mockToolSuccessResultObject = {
-                success: true,
-                details: 'executed',
-            };
-            const mockFunctionResponseContent = [
-                {
-                    type: 'text',
-                    text: JSON.stringify(mockToolSuccessResultObject),
-                },
-            ];
-            const errorResponse = { isError: isErrorValue };
-            const mockMcpToolResponseParts = [
-                {
-                    functionResponse: {
-                        name: serverToolName,
-                        response: {
-                            error: errorResponse,
-                            content: mockFunctionResponseContent,
-                        },
-                    },
-                },
-            ];
-            mockCallTool.mockResolvedValue(mockMcpToolResponseParts);
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            const stringifiedResponseContent = JSON.stringify(mockToolSuccessResultObject);
-            expect(toolResult.llmContent).toEqual([
-                { text: stringifiedResponseContent },
-            ]);
-            expect(toolResult.returnDisplay).toBe(stringifiedResponseContent);
-        });
-        it('should handle a simple text response correctly', async () => {
-            const params = { param: 'test' };
-            const successMessage = 'This is a success message.';
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [{ type: 'text', text: successMessage }],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            // 1. Assert that the llmContent sent to the scheduler is a clean Part array.
-            expect(toolResult.llmContent).toEqual([{ text: successMessage }]);
-            // 2. Assert that the display output is the simple text message.
-            expect(toolResult.returnDisplay).toBe(successMessage);
-            // 3. Verify that the underlying callTool was made correctly.
-            expect(mockCallTool).toHaveBeenCalledWith([
-                { name: serverToolName, args: params },
-            ]);
-        });
-        it('should handle an AudioBlock response', async () => {
-            const params = { param: 'play' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    {
-                        type: 'audio',
-                        data: 'BASE64_AUDIO_DATA',
-                        mimeType: 'audio/mp3',
-                    },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([
-                {
-                    text: `[Tool '${serverToolName}' provided the following audio data with mime-type: audio/mp3]`,
-                },
-                {
-                    inlineData: {
-                        mimeType: 'audio/mp3',
-                        data: 'BASE64_AUDIO_DATA',
-                    },
-                },
-            ]);
-            expect(toolResult.returnDisplay).toBe('[Audio: audio/mp3]');
-        });
-        it('should handle a ResourceLinkBlock response', async () => {
-            const params = { param: 'get' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    {
-                        type: 'resource_link',
-                        uri: 'file:///path/to/thing',
-                        name: 'resource-name',
-                        title: 'My Resource',
-                    },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([
-                {
-                    text: 'Resource Link: My Resource at file:///path/to/thing',
-                },
-            ]);
-            expect(toolResult.returnDisplay).toBe('[Link to My Resource: file:///path/to/thing]');
-        });
-        it('should handle an embedded text ResourceBlock response', async () => {
-            const params = { param: 'get' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    {
-                        type: 'resource',
-                        resource: {
-                            uri: 'file:///path/to/text.txt',
-                            text: 'This is the text content.',
-                            mimeType: 'text/plain',
-                        },
-                    },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([
-                { text: 'This is the text content.' },
-            ]);
-            expect(toolResult.returnDisplay).toBe('This is the text content.');
-        });
-        it('should handle an embedded binary ResourceBlock response', async () => {
-            const params = { param: 'get' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    {
-                        type: 'resource',
-                        resource: {
-                            uri: 'file:///path/to/data.bin',
-                            blob: 'BASE64_BINARY_DATA',
-                            mimeType: 'application/octet-stream',
-                        },
-                    },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([
-                {
-                    text: `[Tool '${serverToolName}' provided the following embedded resource with mime-type: application/octet-stream]`,
-                },
-                {
-                    inlineData: {
-                        mimeType: 'application/octet-stream',
-                        data: 'BASE64_BINARY_DATA',
-                    },
-                },
-            ]);
-            expect(toolResult.returnDisplay).toBe('[Embedded Resource: application/octet-stream]');
-        });
-        it('should handle a mix of content block types', async () => {
-            const params = { param: 'complex' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    { type: 'text', text: 'First part.' },
-                    {
-                        type: 'image',
-                        data: 'BASE64_IMAGE_DATA',
-                        mimeType: 'image/jpeg',
-                    },
-                    { type: 'text', text: 'Second part.' },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([
-                { text: 'First part.' },
-                {
-                    text: `[Tool '${serverToolName}' provided the following image data with mime-type: image/jpeg]`,
-                },
-                {
-                    inlineData: {
-                        mimeType: 'image/jpeg',
-                        data: 'BASE64_IMAGE_DATA',
-                    },
-                },
-                { text: 'Second part.' },
-            ]);
-            expect(toolResult.returnDisplay).toBe('First part.\n[Image: image/jpeg]\nSecond part.');
-        });
-        it('should ignore unknown content block types', async () => {
-            const params = { param: 'test' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    { type: 'text', text: 'Valid part.' },
-                    { type: 'future_block', data: 'some-data' },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([{ text: 'Valid part.' }]);
-            expect(toolResult.returnDisplay).toBe('Valid part.\n[Unknown content type: future_block]');
-        });
-        it('should handle a complex mix of content block types', async () => {
-            const params = { param: 'super-complex' };
-            mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                content: [
-                    { type: 'text', text: 'Here is a resource.' },
-                    {
-                        type: 'resource_link',
-                        uri: 'file:///path/to/resource',
-                        name: 'resource-name',
-                        title: 'My Resource',
-                    },
-                    {
-                        type: 'resource',
-                        resource: {
-                            uri: 'file:///path/to/text.txt',
-                            text: 'Embedded text content.',
-                            mimeType: 'text/plain',
-                        },
-                    },
-                    {
-                        type: 'image',
-                        data: 'BASE64_IMAGE_DATA',
-                        mimeType: 'image/jpeg',
-                    },
-                ],
-            }));
-            const invocation = tool.build(params);
-            const toolResult = await invocation.execute(new AbortController().signal);
-            expect(toolResult.llmContent).toEqual([
-                { text: 'Here is a resource.' },
-                {
-                    text: 'Resource Link: My Resource at file:///path/to/resource',
-                },
-                { text: 'Embedded text content.' },
-                {
-                    text: `[Tool '${serverToolName}' provided the following image data with mime-type: image/jpeg]`,
-                },
-                {
-                    inlineData: {
-                        mimeType: 'image/jpeg',
-                        data: 'BASE64_IMAGE_DATA',
-                    },
-                },
-            ]);
-            expect(toolResult.returnDisplay).toBe('Here is a resource.\n[Link to My Resource: file:///path/to/resource]\nEmbedded text content.\n[Image: image/jpeg]');
-        });
-        describe('AbortSignal support', () => {
-            const MOCK_TOOL_DELAY = 1000;
-            const ABORT_DELAY = 50;
-            it('should abort immediately if signal is already aborted', async () => {
-                const params = { param: 'test' };
-                const controller = new AbortController();
-                controller.abort();
-                const invocation = tool.build(params);
-                await expect(invocation.execute(controller.signal)).rejects.toThrow('Tool call aborted');
-                // Tool should not be called if signal is already aborted
-                expect(mockCallTool).not.toHaveBeenCalled();
-            });
-            it('should abort during tool execution', async () => {
-                const params = { param: 'test' };
-                const controller = new AbortController();
-                // Mock a delayed response to simulate long-running tool
-                mockCallTool.mockImplementation(() => new Promise((resolve) => {
-                    setTimeout(() => {
-                        resolve([
-                            {
-                                functionResponse: {
-                                    name: serverToolName,
-                                    response: {
-                                        content: [{ type: 'text', text: 'Success' }],
-                                    },
-                                },
-                            },
-                        ]);
-                    }, MOCK_TOOL_DELAY);
-                }));
-                const invocation = tool.build(params);
-                const promise = invocation.execute(controller.signal);
-                // Abort after a short delay to simulate cancellation during execution
-                setTimeout(() => controller.abort(), ABORT_DELAY);
-                await expect(promise).rejects.toThrow('Tool call aborted');
-            });
-            it('should complete successfully if not aborted', async () => {
-                const params = { param: 'test' };
-                const controller = new AbortController();
-                mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                    content: [{ type: 'text', text: 'Success' }],
-                }));
-                const invocation = tool.build(params);
-                const result = await invocation.execute(controller.signal);
-                expect(result.llmContent).toEqual([{ text: 'Success' }]);
-                expect(result.returnDisplay).toBe('Success');
-                expect(mockCallTool).toHaveBeenCalledWith([
-                    { name: serverToolName, args: params },
-                ]);
-            });
-            it('should handle tool error even when abort signal is provided', async () => {
-                const params = { param: 'test' };
-                const controller = new AbortController();
-                mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, { error: { isError: true } }));
-                const invocation = tool.build(params);
-                const result = await invocation.execute(controller.signal);
-                expect(result.error?.type).toBe(ToolErrorType.MCP_TOOL_ERROR);
-                expect(result.returnDisplay).toContain(`Error: MCP tool '${serverToolName}' reported an error.`);
-            });
-            it('should handle callTool rejection with abort signal', async () => {
-                const params = { param: 'test' };
-                const controller = new AbortController();
-                const expectedError = new Error('Network error');
-                mockCallTool.mockRejectedValue(expectedError);
-                const invocation = tool.build(params);
-                await expect(invocation.execute(controller.signal)).rejects.toThrow(expectedError);
-            });
-            it.each([
-                {
-                    name: 'successful completion',
-                    setup: () => {
-                        mockCallTool.mockResolvedValue(createSdkResponse(serverToolName, {
-                            content: [{ type: 'text', text: 'Success' }],
-                        }));
-                    },
-                    expectError: false,
-                },
-                {
-                    name: 'error',
-                    setup: () => {
-                        mockCallTool.mockRejectedValue(new Error('Tool execution failed'));
-                    },
-                    expectError: true,
-                },
-            ])('should cleanup event listeners properly on $name', async ({ setup, expectError }) => {
-                const params = { param: 'test' };
-                const controller = new AbortController();
-                setup();
-                const invocation = tool.build(params);
-                if (expectError) {
-                    try {
-                        await invocation.execute(controller.signal);
-                    }
-                    catch (_error) {
-                        // Expected error
-                    }
-                }
-                else {
-                    await invocation.execute(controller.signal);
-                }
-                // Verify cleanup by aborting after execution
-                controller.abort();
-                expect(controller.signal.aborted).toBe(true);
-            });
-        });
-    });
-    describe('shouldConfirmExecute', () => {
-        it('should return false if trust is true', async () => {
-            const trustedTool = new DiscoveredMCPTool(mockCallableToolInstance, serverName, serverToolName, baseDescription, inputSchema, createMockMessageBus(), true, undefined, { isTrustedFolder: () => true }, undefined, undefined);
-            const invocation = trustedTool.build({ param: 'mock' });
-            expect(await invocation.shouldConfirmExecute(new AbortController().signal)).toBe(false);
-        });
-        it('should return false if server is allowlisted', async () => {
-            const invocation = tool.build({ param: 'mock' });
-            invocation.constructor.allowlist.add(serverName);
-            expect(await invocation.shouldConfirmExecute(new AbortController().signal)).toBe(false);
-        });
-        it('should return false if tool is allowlisted', async () => {
-            const toolAllowlistKey = `${serverName}.${serverToolName}`;
-            const invocation = tool.build({ param: 'mock' });
-            invocation.constructor.allowlist.add(toolAllowlistKey);
-            expect(await invocation.shouldConfirmExecute(new AbortController().signal)).toBe(false);
-        });
-        it('should return confirmation details if not trusted and not allowlisted', async () => {
-            const invocation = tool.build({ param: 'mock' });
-            const confirmation = await invocation.shouldConfirmExecute(new AbortController().signal);
-            expect(confirmation).not.toBe(false);
-            if (confirmation && confirmation.type === 'mcp') {
-                // Type guard for ToolMcpConfirmationDetails
-                expect(confirmation.type).toBe('mcp');
-                expect(confirmation.serverName).toBe(serverName);
-                expect(confirmation.toolName).toBe(serverToolName);
-            }
-            else if (confirmation) {
-                // Handle other possible confirmation types if necessary, or strengthen test if only MCP is expected
-                throw new Error('Confirmation was not of expected type MCP or was false');
-            }
-            else {
-                throw new Error('Confirmation details not in expected format or was false');
-            }
-        });
-        it.each([
-            {
-                outcome: ToolConfirmationOutcome.ProceedAlwaysServer,
-                description: 'add server to allowlist on ProceedAlwaysServer',
-                shouldAddServer: true,
-                shouldAddTool: false,
-            },
-            {
-                outcome: ToolConfirmationOutcome.ProceedAlwaysTool,
-                description: 'add tool to allowlist on ProceedAlwaysTool',
-                shouldAddServer: false,
-                shouldAddTool: true,
-            },
-            {
-                outcome: ToolConfirmationOutcome.Cancel,
-                description: 'handle Cancel confirmation outcome',
-                shouldAddServer: false,
-                shouldAddTool: false,
-            },
-            {
-                outcome: ToolConfirmationOutcome.ProceedOnce,
-                description: 'handle ProceedOnce confirmation outcome',
-                shouldAddServer: false,
-                shouldAddTool: false,
-            },
-        ])('should $description', async ({ outcome, shouldAddServer, shouldAddTool }) => {
-            const toolAllowlistKey = `${serverName}.${serverToolName}`;
-            const invocation = tool.build({ param: 'mock' });
-            const confirmation = await invocation.shouldConfirmExecute(new AbortController().signal);
-            expect(confirmation).not.toBe(false);
-            if (confirmation &&
-                typeof confirmation === 'object' &&
-                'onConfirm' in confirmation &&
-                typeof confirmation.onConfirm === 'function') {
-                await confirmation.onConfirm(outcome);
-                expect(invocation.constructor.allowlist.has(serverName)).toBe(shouldAddServer);
-                expect(invocation.constructor.allowlist.has(toolAllowlistKey)).toBe(shouldAddTool);
-            }
-            else {
-                throw new Error('Confirmation details or onConfirm not in expected format');
-            }
-        });
-    });
-    describe('shouldConfirmExecute with folder trust', () => {
-        const mockConfig = (isTrusted) => ({
-            isTrustedFolder: () => isTrusted,
-        });
-        it.each([
-            {
-                trust: true,
-                isTrusted: true,
-                shouldConfirm: false,
-                description: 'return false if trust is true and folder is trusted',
-            },
-            {
-                trust: true,
-                isTrusted: false,
-                shouldConfirm: true,
-                description: 'return confirmation details if trust is true but folder is not trusted',
-            },
-            {
-                trust: false,
-                isTrusted: true,
-                shouldConfirm: true,
-                description: 'return confirmation details if trust is false, even if folder is trusted',
-            },
-        ])('should $description', async ({ trust, isTrusted, shouldConfirm }) => {
-            const bus = createMockMessageBus();
-            getMockMessageBusInstance(bus).defaultToolDecision = 'ask_user';
-            const testTool = new DiscoveredMCPTool(mockCallableToolInstance, serverName, serverToolName, baseDescription, inputSchema, bus, trust, undefined, mockConfig(isTrusted), undefined, undefined);
-            const invocation = testTool.build({ param: 'mock' });
-            const confirmation = await invocation.shouldConfirmExecute(new AbortController().signal);
-            if (shouldConfirm) {
-                expect(confirmation).not.toBe(false);
-                expect(confirmation).toHaveProperty('type', 'mcp');
-            }
-            else {
-                expect(confirmation).toBe(false);
-            }
-        });
-    });
-    describe('DiscoveredMCPToolInvocation', () => {
-        it('should return the stringified params from getDescription', () => {
-            const params = { param: 'testValue', param2: 'anotherOne' };
-            const invocation = tool.build(params);
-            const description = invocation.getDescription();
-            expect(description).toBe('{"param":"testValue","param2":"anotherOne"}');
-        });
-    });
-});
-//# sourceMappingURL=mcp-tool.test.js.map