ai-cli-mcp 2.0.0

package/dist/__tests__/server.test.js

@@ -0,0 +1,665 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { spawn } from 'node:child_process';
+import { existsSync } from 'node:fs';
+import { homedir } from 'node:os';
+import { resolve as pathResolve } from 'node:path';
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { EventEmitter } from 'node:events';
+// Mock dependencies
+vi.mock('node:child_process');
+vi.mock('node:fs');
+vi.mock('node:os');
+vi.mock('node:path', () => ({
+    resolve: vi.fn((path) => path),
+    join: vi.fn((...args) => args.join('/')),
+    isAbsolute: vi.fn((path) => path.startsWith('/'))
+}));
+vi.mock('@modelcontextprotocol/sdk/server/stdio.js');
+vi.mock('@modelcontextprotocol/sdk/types.js', () => ({
+    ListToolsRequestSchema: { name: 'listTools' },
+    CallToolRequestSchema: { name: 'callTool' },
+    ErrorCode: {
+        InternalError: 'InternalError',
+        MethodNotFound: 'MethodNotFound',
+        InvalidParams: 'InvalidParams'
+    },
+    McpError: vi.fn().mockImplementation((code, message) => {
+        const error = new Error(message);
+        error.code = code;
+        return error;
+    })
+}));
+vi.mock('@modelcontextprotocol/sdk/server/index.js', () => ({
+    Server: vi.fn().mockImplementation(function () {
+        this.setRequestHandler = vi.fn();
+        this.connect = vi.fn();
+        this.close = vi.fn();
+        this.onerror = undefined;
+        return this;
+    }),
+}));
+// Mock package.json
+vi.mock('../../package.json', () => ({
+    default: { version: '1.0.0-test' }
+}));
+// Re-import after mocks
+const mockExistsSync = vi.mocked(existsSync);
+const mockSpawn = vi.mocked(spawn);
+const mockHomedir = vi.mocked(homedir);
+const mockPathResolve = vi.mocked(pathResolve);
+// Module loading will happen in tests
+describe('ClaudeCodeServer Unit Tests', () => {
+    let consoleErrorSpy;
+    let consoleWarnSpy;
+    let originalEnv;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        vi.resetModules();
+        vi.unmock('../server.js');
+        consoleErrorSpy = vi.spyOn(console, 'error').mockImplementation(() => { });
+        consoleWarnSpy = vi.spyOn(console, 'warn').mockImplementation(() => { });
+        originalEnv = { ...process.env };
+        // Reset env
+        process.env = { ...originalEnv };
+    });
+    afterEach(() => {
+        consoleErrorSpy.mockRestore();
+        consoleWarnSpy.mockRestore();
+        process.env = originalEnv;
+    });
+    describe('debugLog function', () => {
+        it('should log when debug mode is enabled', async () => {
+            process.env.MCP_CLAUDE_DEBUG = 'true';
+            const module = await import('../server.js');
+            // @ts-ignore - accessing private function for testing
+            const { debugLog } = module;
+            debugLog('Test message');
+            expect(consoleErrorSpy).toHaveBeenCalledWith('Test message');
+        });
+        it('should not log when debug mode is disabled', async () => {
+            // Reset modules to clear cache
+            vi.resetModules();
+            consoleErrorSpy.mockClear();
+            process.env.MCP_CLAUDE_DEBUG = 'false';
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { debugLog } = module;
+            debugLog('Test message');
+            expect(consoleErrorSpy).not.toHaveBeenCalled();
+        });
+    });
+    describe('findClaudeCli function', () => {
+        it('should return local path when it exists', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockImplementation((path) => {
+                // Mock returns true for real CLI path
+                if (path === '/home/user/.claude/local/claude')
+                    return true;
+                return false;
+            });
+            const module = await import('../server.js');
+            // @ts-ignore
+            const findClaudeCli = module.default?.findClaudeCli || module.findClaudeCli;
+            const result = findClaudeCli();
+            expect(result).toBe('/home/user/.claude/local/claude');
+        });
+        it('should fallback to PATH when local does not exist', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(false);
+            const module = await import('../server.js');
+            // @ts-ignore
+            const findClaudeCli = module.default?.findClaudeCli || module.findClaudeCli;
+            const result = findClaudeCli();
+            expect(result).toBe('claude');
+            expect(consoleWarnSpy).toHaveBeenCalledWith(expect.stringContaining('Claude CLI not found at ~/.claude/local/claude'));
+        });
+        it('should use custom name from CLAUDE_CLI_NAME', async () => {
+            process.env.CLAUDE_CLI_NAME = 'my-claude';
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(false);
+            const module = await import('../server.js');
+            // @ts-ignore
+            const findClaudeCli = module.default?.findClaudeCli || module.findClaudeCli;
+            const result = findClaudeCli();
+            expect(result).toBe('my-claude');
+        });
+        it('should use absolute path from CLAUDE_CLI_NAME', async () => {
+            process.env.CLAUDE_CLI_NAME = '/absolute/path/to/claude';
+            const module = await import('../server.js');
+            // @ts-ignore
+            const findClaudeCli = module.default?.findClaudeCli || module.findClaudeCli;
+            const result = findClaudeCli();
+            expect(result).toBe('/absolute/path/to/claude');
+        });
+        it('should throw error for relative paths in CLAUDE_CLI_NAME', async () => {
+            process.env.CLAUDE_CLI_NAME = './relative/path/claude';
+            const module = await import('../server.js');
+            // @ts-ignore
+            const findClaudeCli = module.default?.findClaudeCli || module.findClaudeCli;
+            expect(() => findClaudeCli()).toThrow('Invalid CLAUDE_CLI_NAME: Relative paths are not allowed');
+        });
+        it('should throw error for paths with ../ in CLAUDE_CLI_NAME', async () => {
+            process.env.CLAUDE_CLI_NAME = '../relative/path/claude';
+            const module = await import('../server.js');
+            // @ts-ignore
+            const findClaudeCli = module.default?.findClaudeCli || module.findClaudeCli;
+            expect(() => findClaudeCli()).toThrow('Invalid CLAUDE_CLI_NAME: Relative paths are not allowed');
+        });
+    });
+    describe('spawnAsync function', () => {
+        let mockProcess;
+        beforeEach(() => {
+            // Create a mock process
+            mockProcess = new EventEmitter();
+            mockProcess.stdout = new EventEmitter();
+            mockProcess.stderr = new EventEmitter();
+            mockProcess.stdout.on = vi.fn((event, handler) => {
+                mockProcess.stdout[event] = handler;
+            });
+            mockProcess.stderr.on = vi.fn((event, handler) => {
+                mockProcess.stderr[event] = handler;
+            });
+            mockSpawn.mockReturnValue(mockProcess);
+        });
+        it('should execute command successfully', async () => {
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { spawnAsync } = module;
+            // mockProcess is already defined in the outer scope
+            // Start the async operation
+            const promise = spawnAsync('echo', ['test']);
+            // Simulate successful execution
+            setTimeout(() => {
+                mockProcess.stdout['data']('test output');
+                mockProcess.stderr['data']('');
+                mockProcess.emit('close', 0);
+            }, 10);
+            const result = await promise;
+            expect(result).toEqual({
+                stdout: 'test output',
+                stderr: ''
+            });
+        });
+        it('should handle command failure', async () => {
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { spawnAsync } = module;
+            // mockProcess is already defined in the outer scope
+            // Start the async operation
+            const promise = spawnAsync('false', []);
+            // Simulate failed execution
+            setTimeout(() => {
+                mockProcess.stderr['data']('error output');
+                mockProcess.emit('close', 1);
+            }, 10);
+            await expect(promise).rejects.toThrow('Command failed with exit code 1');
+        });
+        it('should handle spawn error', async () => {
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { spawnAsync } = module;
+            // mockProcess is already defined in the outer scope
+            // Start the async operation
+            const promise = spawnAsync('nonexistent', []);
+            // Simulate spawn error
+            setTimeout(() => {
+                const error = new Error('spawn error');
+                error.code = 'ENOENT';
+                error.path = 'nonexistent';
+                error.syscall = 'spawn';
+                mockProcess.emit('error', error);
+            }, 10);
+            await expect(promise).rejects.toThrow('Spawn error');
+        });
+        it('should respect timeout option', async () => {
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { spawnAsync } = module;
+            const result = spawnAsync('sleep', ['10'], { timeout: 100 });
+            expect(mockSpawn).toHaveBeenCalledWith('sleep', ['10'], expect.objectContaining({
+                timeout: 100
+            }));
+        });
+        it('should use provided cwd option', async () => {
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { spawnAsync } = module;
+            const result = spawnAsync('ls', [], { cwd: '/tmp' });
+            expect(mockSpawn).toHaveBeenCalledWith('ls', [], expect.objectContaining({
+                cwd: '/tmp'
+            }));
+        });
+    });
+    describe('ClaudeCodeServer class', () => {
+        it('should initialize with correct settings', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock before resetting modules
+            vi.mocked(Server).mockImplementation(function () {
+                this.setRequestHandler = vi.fn();
+                this.connect = vi.fn();
+                this.close = vi.fn();
+                this.onerror = undefined;
+                return this;
+            });
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            expect(consoleErrorSpy).toHaveBeenCalledWith(expect.stringContaining('[Setup] Using Claude CLI command/path:'));
+        });
+        it('should set up tool handlers', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            const { Server } = await import('@modelcontextprotocol/sdk/server/index.js');
+            const mockSetRequestHandler = vi.fn();
+            vi.mocked(Server).mockImplementation(function () {
+                this.setRequestHandler = mockSetRequestHandler;
+                this.connect = vi.fn();
+                this.close = vi.fn();
+                this.onerror = undefined;
+                return this;
+            });
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            expect(mockSetRequestHandler).toHaveBeenCalled();
+        });
+        it('should set up error handler', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            const { Server } = await import('@modelcontextprotocol/sdk/server/index.js');
+            let errorHandler = null;
+            vi.mocked(Server).mockImplementation(function () {
+                this.setRequestHandler = vi.fn();
+                this.connect = vi.fn();
+                this.close = vi.fn();
+                Object.defineProperty(this, 'onerror', {
+                    get() { return errorHandler; },
+                    set(handler) { errorHandler = handler; },
+                    enumerable: true,
+                    configurable: true
+                });
+                return this;
+            });
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            // Test error handler
+            errorHandler(new Error('Test error'));
+            expect(consoleErrorSpy).toHaveBeenCalledWith('[Error]', expect.any(Error));
+        });
+        it('should handle SIGINT', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock first
+            vi.mocked(Server).mockImplementation(function () {
+                this.setRequestHandler = vi.fn();
+                this.connect = vi.fn();
+                this.close = vi.fn();
+                this.onerror = undefined;
+                return this;
+            });
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const exitSpy = vi.spyOn(process, 'exit').mockImplementation(() => undefined);
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Emit SIGINT
+            const sigintHandler = process.listeners('SIGINT').slice(-1)[0];
+            await sigintHandler();
+            expect(mockServerInstance.close).toHaveBeenCalled();
+            expect(exitSpy).toHaveBeenCalledWith(0);
+            exitSpy.mockRestore();
+        });
+    });
+    describe('Tool handler implementation', () => {
+        // Define setupServerMock for this describe block
+        let errorHandler = null;
+        function setupServerMock() {
+            errorHandler = null;
+            vi.mocked(Server).mockImplementation(function () {
+                this.setRequestHandler = vi.fn();
+                this.connect = vi.fn();
+                this.close = vi.fn();
+                Object.defineProperty(this, 'onerror', {
+                    get() { return errorHandler; },
+                    set(handler) { errorHandler = handler; },
+                    enumerable: true,
+                    configurable: true
+                });
+                return this;
+            });
+        }
+        it('should handle ListToolsRequest', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Use the setupServerMock function from the beginning of the file
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the ListToolsRequest handler
+            const listToolsCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'listTools');
+            expect(listToolsCall).toBeDefined();
+            // Test the handler
+            const handler = listToolsCall[1];
+            const result = await handler();
+            expect(result.tools).toHaveLength(4);
+            expect(result.tools[0].name).toBe('claude_code');
+            expect(result.tools[0].description).toContain('Claude Code Agent');
+            expect(result.tools[1].name).toBe('list_claude_processes');
+            expect(result.tools[2].name).toBe('get_claude_result');
+            expect(result.tools[3].name).toBe('kill_claude_process');
+        });
+        it('should handle CallToolRequest', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            expect(callToolCall).toBeDefined();
+            // Create a mock process for the tool execution
+            const mockProcess = new EventEmitter();
+            mockProcess.pid = 12345;
+            mockProcess.stdout = new EventEmitter();
+            mockProcess.stderr = new EventEmitter();
+            mockProcess.stdout.on = vi.fn();
+            mockProcess.stderr.on = vi.fn();
+            mockProcess.kill = vi.fn();
+            mockSpawn.mockReturnValue(mockProcess);
+            // Test the handler
+            const handler = callToolCall[1];
+            const result = await handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt: 'test prompt',
+                        workFolder: '/tmp'
+                    }
+                }
+            });
+            // claude_code now returns PID immediately
+            expect(result.content[0].type).toBe('text');
+            const response = JSON.parse(result.content[0].text);
+            expect(response.pid).toBe(12345);
+            expect(response.status).toBe('started');
+        });
+        it('should require workFolder parameter', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Test missing workFolder
+            await expect(handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt: 'test'
+                    }
+                }
+            })).rejects.toThrow('Missing or invalid required parameter: workFolder');
+        });
+        it('should handle non-existent workFolder', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockImplementation((path) => {
+                // Make the CLI path exist but the workFolder not exist
+                if (String(path).includes('.claude'))
+                    return true;
+                if (path === '/nonexistent')
+                    return false;
+                return false;
+            });
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Should throw error for non-existent workFolder
+            await expect(handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt: 'test',
+                        workFolder: '/nonexistent'
+                    }
+                }
+            })).rejects.toThrow('Working folder does not exist');
+        });
+        it('should handle session_id parameter', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Create mock process
+            const mockProcess = new EventEmitter();
+            mockProcess.pid = 12347;
+            mockProcess.stdout = new EventEmitter();
+            mockProcess.stderr = new EventEmitter();
+            mockProcess.stdout.on = vi.fn();
+            mockProcess.stderr.on = vi.fn();
+            mockProcess.kill = vi.fn();
+            mockSpawn.mockReturnValue(mockProcess);
+            const result = await handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt: 'test prompt',
+                        workFolder: '/tmp',
+                        session_id: 'test-session-123'
+                    }
+                }
+            });
+            // Verify spawn was called with -r flag
+            expect(mockSpawn).toHaveBeenCalledWith(expect.any(String), expect.arrayContaining(['-r', 'test-session-123', '-p', 'test prompt']), expect.any(Object));
+        });
+        it('should handle prompt_file parameter', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockImplementation((path) => {
+                if (String(path).includes('.claude'))
+                    return true;
+                if (path === '/tmp')
+                    return true;
+                if (path === '/tmp/prompt.txt')
+                    return true;
+                return false;
+            });
+            // Mock readFileSync
+            const readFileSyncMock = vi.fn().mockReturnValue('Content from file');
+            vi.doMock('node:fs', () => ({
+                existsSync: mockExistsSync,
+                readFileSync: readFileSyncMock
+            }));
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Create mock process
+            const mockProcess = new EventEmitter();
+            mockProcess.pid = 12348;
+            mockProcess.stdout = new EventEmitter();
+            mockProcess.stderr = new EventEmitter();
+            mockProcess.stdout.on = vi.fn();
+            mockProcess.stderr.on = vi.fn();
+            mockProcess.kill = vi.fn();
+            mockSpawn.mockReturnValue(mockProcess);
+            const result = await handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt_file: '/tmp/prompt.txt',
+                        workFolder: '/tmp'
+                    }
+                }
+            });
+            // Verify file was read and spawn was called with content
+            expect(mockSpawn).toHaveBeenCalledWith(expect.any(String), expect.arrayContaining(['-p', 'Content from file']), expect.any(Object));
+        });
+        it('should resolve model aliases when calling claude_code tool', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up spawn mock to return a process
+            const mockProcess = new EventEmitter();
+            mockProcess.stdout = new EventEmitter();
+            mockProcess.stderr = new EventEmitter();
+            mockProcess.pid = 12345;
+            mockSpawn.mockReturnValue(mockProcess);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Test with haiku alias
+            const result = await handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt: 'test prompt',
+                        workFolder: '/tmp',
+                        model: 'haiku'
+                    }
+                }
+            });
+            // Verify spawn was called with resolved model name
+            expect(mockSpawn).toHaveBeenCalledWith(expect.any(String), expect.arrayContaining(['--model', 'claude-3-5-haiku-20241022']), expect.any(Object));
+            // Verify PID is returned
+            expect(result.content[0].text).toContain('"pid": 12345');
+        });
+        it('should pass non-alias model names unchanged', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up spawn mock to return a process
+            const mockProcess = new EventEmitter();
+            mockProcess.stdout = new EventEmitter();
+            mockProcess.stderr = new EventEmitter();
+            mockProcess.pid = 12346;
+            mockSpawn.mockReturnValue(mockProcess);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Test with non-alias model name
+            const result = await handler({
+                params: {
+                    name: 'claude_code',
+                    arguments: {
+                        prompt: 'test prompt',
+                        workFolder: '/tmp',
+                        model: 'sonnet'
+                    }
+                }
+            });
+            // Verify spawn was called with unchanged model name
+            expect(mockSpawn).toHaveBeenCalledWith(expect.any(String), expect.arrayContaining(['--model', 'sonnet']), expect.any(Object));
+        });
+        it('should reject when both prompt and prompt_file are provided', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Test both parameters provided
+            try {
+                await handler({
+                    params: {
+                        name: 'claude_code',
+                        arguments: {
+                            prompt: 'test prompt',
+                            prompt_file: '/tmp/prompt.txt',
+                            workFolder: '/tmp'
+                        }
+                    }
+                });
+                expect.fail('Should have thrown an error');
+            }
+            catch (error) {
+                expect(error.message).toContain('Cannot specify both prompt and prompt_file');
+            }
+        });
+        it('should reject when neither prompt nor prompt_file are provided', async () => {
+            mockHomedir.mockReturnValue('/home/user');
+            mockExistsSync.mockReturnValue(true);
+            // Set up Server mock
+            setupServerMock();
+            const module = await import('../server.js');
+            // @ts-ignore
+            const { ClaudeCodeServer } = module;
+            const server = new ClaudeCodeServer();
+            const mockServerInstance = vi.mocked(Server).mock.results[0].value;
+            // Find the CallToolRequest handler
+            const callToolCall = mockServerInstance.setRequestHandler.mock.calls.find((call) => call[0].name === 'callTool');
+            const handler = callToolCall[1];
+            // Test neither parameter provided
+            try {
+                await handler({
+                    params: {
+                        name: 'claude_code',
+                        arguments: {
+                            workFolder: '/tmp'
+                        }
+                    }
+                });
+                expect.fail('Should have thrown an error');
+            }
+            catch (error) {
+                expect(error.message).toContain('Either prompt or prompt_file must be provided');
+            }
+        });
+    });
+});

package/dist/__tests__/setup.js

@@ -0,0 +1,11 @@
+// Global test setup
+import { beforeAll, afterAll } from 'vitest';
+import { getSharedMock, cleanupSharedMock } from './utils/persistent-mock.js';
+beforeAll(async () => {
+    console.error('[TEST SETUP] Creating shared mock for all tests...');
+    await getSharedMock();
+});
+afterAll(async () => {
+    console.error('[TEST SETUP] Cleaning up shared mock...');
+    await cleanupSharedMock();
+});