@blackbox_ai/blackbox-cli-core 0.0.4

package/dist/src/tools/shell.test.js

@@ -0,0 +1,632 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { vi, describe, it, expect, beforeEach, afterEach, } from 'vitest';
+const mockShellExecutionService = vi.hoisted(() => vi.fn());
+vi.mock('../services/shellExecutionService.js', () => ({
+    ShellExecutionService: { execute: mockShellExecutionService },
+}));
+vi.mock('fs');
+vi.mock('os');
+vi.mock('crypto');
+vi.mock('../utils/summarizer.js');
+import { isCommandAllowed } from '../utils/shell-utils.js';
+import { ShellTool } from './shell.js';
+import {} from '../config/config.js';
+import {} from '../services/shellExecutionService.js';
+import * as fs from 'node:fs';
+import * as os from 'node:os';
+import { EOL } from 'node:os';
+import * as path from 'node:path';
+import * as crypto from 'node:crypto';
+import * as summarizer from '../utils/summarizer.js';
+import { ToolErrorType } from './tool-error.js';
+import { ToolConfirmationOutcome } from './tools.js';
+import { OUTPUT_UPDATE_INTERVAL_MS } from './shell.js';
+import { createMockWorkspaceContext } from '../test-utils/mockWorkspaceContext.js';
+describe('ShellTool', () => {
+    let shellTool;
+    let mockConfig;
+    let mockShellOutputCallback;
+    let resolveExecutionPromise;
+    beforeEach(() => {
+        vi.clearAllMocks();
+        mockConfig = {
+            getCoreTools: vi.fn().mockReturnValue([]),
+            getExcludeTools: vi.fn().mockReturnValue([]),
+            getDebugMode: vi.fn().mockReturnValue(false),
+            getTargetDir: vi.fn().mockReturnValue('/test/dir'),
+            getSummarizeToolOutputConfig: vi.fn().mockReturnValue(undefined),
+            getWorkspaceContext: () => createMockWorkspaceContext('.'),
+            getGeminiClient: vi.fn(),
+            getGitCoAuthor: vi.fn().mockReturnValue({
+                enabled: true,
+                name: 'BlackboxAI',
+                email: 'code@blackbox.ai',
+            }),
+            getShouldUseNodePtyShell: vi.fn().mockReturnValue(false),
+        };
+        shellTool = new ShellTool(mockConfig);
+        vi.mocked(os.platform).mockReturnValue('linux');
+        vi.mocked(os.tmpdir).mockReturnValue('/tmp');
+        vi.mocked(crypto.randomBytes).mockReturnValue(Buffer.from('abcdef', 'hex'));
+        // Capture the output callback to simulate streaming events from the service
+        mockShellExecutionService.mockImplementation((_cmd, _cwd, callback) => {
+            mockShellOutputCallback = callback;
+            return {
+                pid: 12345,
+                result: new Promise((resolve) => {
+                    resolveExecutionPromise = resolve;
+                }),
+            };
+        });
+    });
+    describe('isCommandAllowed', () => {
+        it('should allow a command if no restrictions are provided', () => {
+            mockConfig.getCoreTools.mockReturnValue(undefined);
+            mockConfig.getExcludeTools.mockReturnValue(undefined);
+            expect(isCommandAllowed('ls -l', mockConfig).allowed).toBe(true);
+        });
+        it('should block a command with command substitution using $()', () => {
+            expect(isCommandAllowed('echo $(rm -rf /)', mockConfig).allowed).toBe(false);
+        });
+    });
+    describe('build', () => {
+        it('should return an invocation for a valid command', () => {
+            const invocation = shellTool.build({
+                command: 'ls -l',
+                is_background: false,
+            });
+            expect(invocation).toBeDefined();
+        });
+        it('should throw an error for an empty command', () => {
+            expect(() => shellTool.build({ command: ' ', is_background: false })).toThrow('Command cannot be empty.');
+        });
+        it('should throw an error for a non-existent directory', () => {
+            vi.mocked(fs.existsSync).mockReturnValue(false);
+            expect(() => shellTool.build({
+                command: 'ls',
+                directory: 'rel/path',
+                is_background: false,
+            })).toThrow("Directory 'rel/path' is not a registered workspace directory.");
+        });
+        it('should include background indicator in description when is_background is true', () => {
+            const invocation = shellTool.build({
+                command: 'npm start',
+                is_background: true,
+            });
+            expect(invocation.getDescription()).toContain('[background]');
+        });
+        it('should not include background indicator in description when is_background is false', () => {
+            const invocation = shellTool.build({
+                command: 'npm test',
+                is_background: false,
+            });
+            expect(invocation.getDescription()).not.toContain('[background]');
+        });
+    });
+    describe('execute', () => {
+        const mockAbortSignal = new AbortController().signal;
+        const resolveShellExecution = (result = {}) => {
+            const fullResult = {
+                rawOutput: Buffer.from(result.output || ''),
+                output: 'Success',
+                exitCode: 0,
+                signal: null,
+                error: null,
+                aborted: false,
+                pid: 12345,
+                executionMethod: 'child_process',
+                ...result,
+            };
+            resolveExecutionPromise(fullResult);
+        };
+        it('should wrap command on linux and parse pgrep output', async () => {
+            const invocation = shellTool.build({
+                command: 'my-command &',
+                is_background: false,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({ pid: 54321 });
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue(`54321${EOL}54322${EOL}`); // Service PID and background PID
+            const result = await promise;
+            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
+            const wrappedCommand = `{ my-command & }; __code=$?; pgrep -g 0 >${tmpFile} 2>&1; exit $__code;`;
+            expect(mockShellExecutionService).toHaveBeenCalledWith(wrappedCommand, expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            expect(result.llmContent).toContain('Background PIDs: 54322');
+            expect(vi.mocked(fs.unlinkSync)).toHaveBeenCalledWith(tmpFile);
+        });
+        it('should add ampersand to command when is_background is true and command does not end with &', async () => {
+            const invocation = shellTool.build({
+                command: 'npm start',
+                is_background: true,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({ pid: 54321 });
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue('54321\n54322\n');
+            await promise;
+            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
+            const wrappedCommand = `{ npm start & }; __code=$?; pgrep -g 0 >${tmpFile} 2>&1; exit $__code;`;
+            expect(mockShellExecutionService).toHaveBeenCalledWith(wrappedCommand, expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+        });
+        it('should not add extra ampersand when is_background is true and command already ends with &', async () => {
+            const invocation = shellTool.build({
+                command: 'npm start &',
+                is_background: true,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({ pid: 54321 });
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue('54321\n54322\n');
+            await promise;
+            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
+            const wrappedCommand = `{ npm start & }; __code=$?; pgrep -g 0 >${tmpFile} 2>&1; exit $__code;`;
+            expect(mockShellExecutionService).toHaveBeenCalledWith(wrappedCommand, expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+        });
+        it('should not add ampersand when is_background is false', async () => {
+            const invocation = shellTool.build({
+                command: 'npm test',
+                is_background: false,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({ pid: 54321 });
+            vi.mocked(fs.existsSync).mockReturnValue(true);
+            vi.mocked(fs.readFileSync).mockReturnValue('54321\n54322\n');
+            await promise;
+            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
+            const wrappedCommand = `{ npm test; }; __code=$?; pgrep -g 0 >${tmpFile} 2>&1; exit $__code;`;
+            expect(mockShellExecutionService).toHaveBeenCalledWith(wrappedCommand, expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+        });
+        it('should not wrap command on windows', async () => {
+            vi.mocked(os.platform).mockReturnValue('win32');
+            const invocation = shellTool.build({
+                command: 'dir',
+                is_background: false,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({
+                rawOutput: Buffer.from(''),
+                output: '',
+                exitCode: 0,
+                signal: null,
+                error: null,
+                aborted: false,
+                pid: 12345,
+                executionMethod: 'child_process',
+            });
+            await promise;
+            expect(mockShellExecutionService).toHaveBeenCalledWith('dir', expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+        });
+        it('should format error messages correctly', async () => {
+            const error = new Error('wrapped command failed');
+            const invocation = shellTool.build({
+                command: 'user-command',
+                is_background: false,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({
+                error,
+                exitCode: 1,
+                output: 'err',
+                rawOutput: Buffer.from('err'),
+                signal: null,
+                aborted: false,
+                pid: 12345,
+                executionMethod: 'child_process',
+            });
+            const result = await promise;
+            expect(result.llmContent).toContain('Error: wrapped command failed');
+            expect(result.llmContent).not.toContain('pgrep');
+        });
+        it('should return a SHELL_EXECUTE_ERROR for a command failure', async () => {
+            const error = new Error('command failed');
+            const invocation = shellTool.build({
+                command: 'user-command',
+                is_background: false,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveShellExecution({
+                error,
+                exitCode: 1,
+            });
+            const result = await promise;
+            expect(result.error).toBeDefined();
+            expect(result.error?.type).toBe(ToolErrorType.SHELL_EXECUTE_ERROR);
+            expect(result.error?.message).toBe('command failed');
+        });
+        it('should throw an error for invalid parameters', () => {
+            expect(() => shellTool.build({ command: '', is_background: false })).toThrow('Command cannot be empty.');
+        });
+        it('should throw an error for invalid directory', () => {
+            vi.mocked(fs.existsSync).mockReturnValue(false);
+            expect(() => shellTool.build({
+                command: 'ls',
+                directory: 'nonexistent',
+                is_background: false,
+            })).toThrow(`Directory 'nonexistent' is not a registered workspace directory.`);
+        });
+        it('should summarize output when configured', async () => {
+            mockConfig.getSummarizeToolOutputConfig.mockReturnValue({
+                [shellTool.name]: { tokenBudget: 1000 },
+            });
+            vi.mocked(summarizer.summarizeToolOutput).mockResolvedValue('summarized output');
+            const invocation = shellTool.build({
+                command: 'ls',
+                is_background: false,
+            });
+            const promise = invocation.execute(mockAbortSignal);
+            resolveExecutionPromise({
+                output: 'long output',
+                rawOutput: Buffer.from('long output'),
+                exitCode: 0,
+                signal: null,
+                error: null,
+                aborted: false,
+                pid: 12345,
+                executionMethod: 'child_process',
+            });
+            const result = await promise;
+            expect(summarizer.summarizeToolOutput).toHaveBeenCalledWith(expect.any(String), mockConfig.getGeminiClient(), mockAbortSignal, 1000);
+            expect(result.llmContent).toBe('summarized output');
+            expect(result.returnDisplay).toBe('long output');
+        });
+        it('should clean up the temp file on synchronous execution error', async () => {
+            const error = new Error('sync spawn error');
+            mockShellExecutionService.mockImplementation(() => {
+                throw error;
+            });
+            vi.mocked(fs.existsSync).mockReturnValue(true); // Pretend the file exists
+            const invocation = shellTool.build({
+                command: 'a-command',
+                is_background: false,
+            });
+            await expect(invocation.execute(mockAbortSignal)).rejects.toThrow(error);
+            const tmpFile = path.join(os.tmpdir(), 'shell_pgrep_abcdef.tmp');
+            expect(vi.mocked(fs.unlinkSync)).toHaveBeenCalledWith(tmpFile);
+        });
+        describe('Streaming to `updateOutput`', () => {
+            let updateOutputMock;
+            beforeEach(() => {
+                vi.useFakeTimers({ toFake: ['Date'] });
+                updateOutputMock = vi.fn();
+            });
+            afterEach(() => {
+                vi.useRealTimers();
+            });
+            it('should throttle text output updates', async () => {
+                const invocation = shellTool.build({
+                    command: 'stream',
+                    is_background: false,
+                });
+                const promise = invocation.execute(mockAbortSignal, updateOutputMock);
+                // First chunk, should be throttled.
+                mockShellOutputCallback({
+                    type: 'data',
+                    chunk: 'hello ',
+                });
+                expect(updateOutputMock).not.toHaveBeenCalled();
+                // Advance time past the throttle interval.
+                await vi.advanceTimersByTimeAsync(OUTPUT_UPDATE_INTERVAL_MS + 1);
+                // Send a second chunk. THIS event triggers the update with the CUMULATIVE content.
+                mockShellOutputCallback({
+                    type: 'data',
+                    chunk: 'hello world',
+                });
+                // It should have been called once now with the combined output.
+                expect(updateOutputMock).toHaveBeenCalledOnce();
+                expect(updateOutputMock).toHaveBeenCalledWith('hello world');
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+            });
+            it('should immediately show binary detection message and throttle progress', async () => {
+                const invocation = shellTool.build({
+                    command: 'cat img',
+                    is_background: false,
+                });
+                const promise = invocation.execute(mockAbortSignal, updateOutputMock);
+                mockShellOutputCallback({ type: 'binary_detected' });
+                expect(updateOutputMock).toHaveBeenCalledOnce();
+                expect(updateOutputMock).toHaveBeenCalledWith('[Binary output detected. Halting stream...]');
+                mockShellOutputCallback({
+                    type: 'binary_progress',
+                    bytesReceived: 1024,
+                });
+                expect(updateOutputMock).toHaveBeenCalledOnce();
+                // Advance time past the throttle interval.
+                await vi.advanceTimersByTimeAsync(OUTPUT_UPDATE_INTERVAL_MS + 1);
+                // Send a SECOND progress event. This one will trigger the flush.
+                mockShellOutputCallback({
+                    type: 'binary_progress',
+                    bytesReceived: 2048,
+                });
+                // Now it should be called a second time with the latest progress.
+                expect(updateOutputMock).toHaveBeenCalledTimes(2);
+                expect(updateOutputMock).toHaveBeenLastCalledWith('[Receiving binary output... 2.0 KB received]');
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+            });
+        });
+        describe('addCoAuthorToGitCommit', () => {
+            it('should add co-author to git commit with double quotes', async () => {
+                const command = 'git commit -m "Initial commit"';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                // Mock the shell execution to return success
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                // Verify that the command was executed with co-author added
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('Co-authored-by: BlackboxAI <code@blackbox.ai>'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should add co-author to git commit with single quotes', async () => {
+                const command = "git commit -m 'Fix bug'";
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('Co-authored-by: BlackboxAI <code@blackbox.ai>'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should handle git commit with additional flags', async () => {
+                const command = 'git commit -a -m "Add feature"';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('Co-authored-by: BlackboxAI <code@blackbox.ai>'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should not modify non-git commands', async () => {
+                const command = 'npm install';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                // On Linux, commands are wrapped with pgrep functionality
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('npm install'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should not modify git commands without -m flag', async () => {
+                const command = 'git commit';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                // On Linux, commands are wrapped with pgrep functionality
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('git commit'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should handle git commit with escaped quotes in message', async () => {
+                const command = 'git commit -m "Fix \\"quoted\\" text"';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('Co-authored-by: BlackboxAI <code@blackbox.ai>'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should not add co-author when disabled in config', async () => {
+                // Mock config with disabled co-author
+                mockConfig.getGitCoAuthor.mockReturnValue({
+                    enabled: false,
+                    name: 'BlackboxAI',
+                    email: 'code@blackbox.ai',
+                });
+                const command = 'git commit -m "Initial commit"';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                // On Linux, commands are wrapped with pgrep functionality
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('git commit -m "Initial commit"'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+            it('should use custom name and email from config', async () => {
+                // Mock config with custom co-author details
+                mockConfig.getGitCoAuthor.mockReturnValue({
+                    enabled: true,
+                    name: 'Custom Bot',
+                    email: 'custom@example.com',
+                });
+                const command = 'git commit -m "Test commit"';
+                const invocation = shellTool.build({ command, is_background: false });
+                const promise = invocation.execute(mockAbortSignal);
+                resolveExecutionPromise({
+                    rawOutput: Buffer.from(''),
+                    output: '',
+                    exitCode: 0,
+                    signal: null,
+                    error: null,
+                    aborted: false,
+                    pid: 12345,
+                    executionMethod: 'child_process',
+                });
+                await promise;
+                expect(mockShellExecutionService).toHaveBeenCalledWith(expect.stringContaining('Co-authored-by: Custom Bot <custom@example.com>'), expect.any(String), expect.any(Function), mockAbortSignal, false, undefined, undefined);
+            });
+        });
+    });
+    describe('shouldConfirmExecute', () => {
+        it('should not request confirmation for read-only commands', async () => {
+            const invocation = shellTool.build({
+                command: 'ls -la',
+                is_background: false,
+            });
+            const confirmation = await invocation.shouldConfirmExecute(new AbortController().signal);
+            expect(confirmation).toBe(false);
+        });
+        it('should request confirmation for a new command and whitelist it on "Always"', async () => {
+            const params = { command: 'npm install', is_background: false };
+            const invocation = shellTool.build(params);
+            const confirmation = await invocation.shouldConfirmExecute(new AbortController().signal);
+            expect(confirmation).not.toBe(false);
+            expect(confirmation && confirmation.type).toBe('exec');
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            await confirmation.onConfirm(ToolConfirmationOutcome.ProceedAlways);
+            // Should now be whitelisted
+            const secondInvocation = shellTool.build({
+                command: 'npm test',
+                is_background: false,
+            });
+            const secondConfirmation = await secondInvocation.shouldConfirmExecute(new AbortController().signal);
+            expect(secondConfirmation).toBe(false);
+        });
+        it('should throw an error if validation fails', () => {
+            expect(() => shellTool.build({ command: '', is_background: false })).toThrow();
+        });
+    });
+    describe('getDescription', () => {
+        it('should return the windows description when on windows', () => {
+            vi.mocked(os.platform).mockReturnValue('win32');
+            const shellTool = new ShellTool(mockConfig);
+            expect(shellTool.description).toMatchSnapshot();
+        });
+        it('should return the non-windows description when not on windows', () => {
+            vi.mocked(os.platform).mockReturnValue('linux');
+            const shellTool = new ShellTool(mockConfig);
+            expect(shellTool.description).toMatchSnapshot();
+        });
+    });
+});
+describe('validateToolParams', () => {
+    it('should return null for valid directory', () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => undefined,
+            getTargetDir: () => '/root',
+            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.validateToolParams({
+            command: 'ls',
+            directory: 'test',
+            is_background: false,
+        });
+        expect(result).toBeNull();
+    });
+    it('should return error for directory outside workspace', () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => undefined,
+            getTargetDir: () => '/root',
+            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
+        };
+        const shellTool = new ShellTool(config);
+        const result = shellTool.validateToolParams({
+            command: 'ls',
+            directory: 'test2',
+            is_background: false,
+        });
+        expect(result).toContain('is not a registered workspace directory');
+    });
+});
+describe('build', () => {
+    it('should return an invocation for valid directory', () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => undefined,
+            getTargetDir: () => '/root',
+            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
+        };
+        const shellTool = new ShellTool(config);
+        const invocation = shellTool.build({
+            command: 'ls',
+            directory: 'test',
+            is_background: false,
+        });
+        expect(invocation).toBeDefined();
+    });
+    it('should throw an error for directory outside workspace', () => {
+        const config = {
+            getCoreTools: () => undefined,
+            getExcludeTools: () => undefined,
+            getTargetDir: () => '/root',
+            getWorkspaceContext: () => createMockWorkspaceContext('/root', ['/users/test']),
+        };
+        const shellTool = new ShellTool(config);
+        expect(() => shellTool.build({
+            command: 'ls',
+            directory: 'test2',
+            is_background: false,
+        })).toThrow('is not a registered workspace directory');
+    });
+});
+//# sourceMappingURL=shell.test.js.map