| 44 | |
| 45 | // Helper function to create a mock DiscoveredMCPTool |
| 46 | const createMockMCPTool = ( |
| 47 | name: string, |
| 48 | serverName: string, |
| 49 | description?: string, |
| 50 | ) => |
| 51 | new DiscoveredMCPTool( |
| 52 | { |
| 53 | callTool: vi.fn(), |
| 54 | tool: vi.fn(), |
| 55 | } as unknown as CallableTool, |
| 56 | serverName, |
| 57 | name, |
| 58 | description || `Description for ${name}`, |
| 59 | { type: Type.OBJECT, properties: {} }, |
| 60 | name, // serverToolName same as name for simplicity |
| 61 | ); |
| 62 | |
| 63 | describe('mcpCommand', () => { |
| 64 | let mockContext: ReturnType<typeof createMockCommandContext>; |