@office-ai/aioncli-core 0.2.3 → 0.18.4

package/dist/src/utils/fileUtils.test.js

@@ -5,17 +5,20 @@
  */
 import { describe, it, expect, vi, beforeEach, afterEach, } from 'vitest';
 import * as actualNodeFs from 'node:fs'; // For setup/teardown
+import fs from 'node:fs';
 import fsPromises from 'node:fs/promises';
 import path from 'node:path';
 import os from 'node:os';
-import mime from 'mime-types';
-import { isWithinRoot, isBinaryFile, detectFileType, processSingleFileContent, } from './fileUtils.js';
+import { fileURLToPath } from 'node:url';
+// eslint-disable-next-line import/no-internal-modules
+import mime from 'mime/lite';
+import { isWithinRoot, isBinaryFile, detectFileType, processSingleFileContent, detectBOM, readFileWithEncoding, fileExists, readWasmBinaryFromDisk, } from './fileUtils.js';
 import { StandardFileSystemService } from '../services/fileSystemService.js';
-vi.mock('mime-types', () => ({
-    default: { lookup: vi.fn() },
-    lookup: vi.fn(),
+vi.mock('mime/lite', () => ({
+    default: { getType: vi.fn() },
+    getType: vi.fn(),
 }));
-const mockMimeLookup = mime.lookup;
+const mockMimeGetType = mime.getType;
 describe('fileUtils', () => {
     let tempRootDir;
     const originalProcessCwd = process.cwd;
@@ -26,7 +29,7 @@ describe('fileUtils', () => {
     let nonexistentFilePath;
     let directoryPath;
     beforeEach(() => {
-        vi.resetAllMocks(); // Reset all mocks, including mime.lookup
+        vi.resetAllMocks(); // Reset all mocks, including mime.getType
         tempRootDir = actualNodeFs.mkdtempSync(path.join(os.tmpdir(), 'fileUtils-test-'));
         process.cwd = vi.fn(() => tempRootDir); // Mock cwd if necessary for relative path logic within tests
         testTextFilePath = path.join(tempRootDir, 'test.txt');
@@ -44,40 +47,97 @@ describe('fileUtils', () => {
         process.cwd = originalProcessCwd;
         vi.restoreAllMocks(); // Restore any spies
     });
+    describe('readWasmBinaryFromDisk', () => {
+        it('loads a WASM binary from disk as a Uint8Array', async () => {
+            const wasmFixtureUrl = new URL('./__fixtures__/dummy.wasm', import.meta.url);
+            const wasmFixturePath = fileURLToPath(wasmFixtureUrl);
+            const result = await readWasmBinaryFromDisk(wasmFixturePath);
+            const expectedBytes = new Uint8Array(await fsPromises.readFile(wasmFixturePath));
+            expect(result).toBeInstanceOf(Uint8Array);
+            expect(result).toStrictEqual(expectedBytes);
+        });
+    });
     describe('isWithinRoot', () => {
-        const root = path.resolve('/project/root');
-        it('should return true for paths directly within the root', () => {
-            expect(isWithinRoot(path.join(root, 'file.txt'), root)).toBe(true);
-            expect(isWithinRoot(path.join(root, 'subdir', 'file.txt'), root)).toBe(true);
-        });
-        it('should return true for the root path itself', () => {
-            expect(isWithinRoot(root, root)).toBe(true);
-        });
-        it('should return false for paths outside the root', () => {
-            expect(isWithinRoot(path.resolve('/project/other', 'file.txt'), root)).toBe(false);
-            expect(isWithinRoot(path.resolve('/unrelated', 'file.txt'), root)).toBe(false);
-        });
-        it('should return false for paths that only partially match the root prefix', () => {
-            expect(isWithinRoot(path.resolve('/project/root-but-actually-different'), root)).toBe(false);
-        });
-        it('should handle paths with trailing slashes correctly', () => {
-            expect(isWithinRoot(path.join(root, 'file.txt') + path.sep, root)).toBe(true);
-            expect(isWithinRoot(root + path.sep, root)).toBe(true);
-        });
-        it('should handle different path separators (POSIX vs Windows)', () => {
-            const posixRoot = '/project/root';
-            const posixPathInside = '/project/root/file.txt';
-            const posixPathOutside = '/project/other/file.txt';
-            expect(isWithinRoot(posixPathInside, posixRoot)).toBe(true);
-            expect(isWithinRoot(posixPathOutside, posixRoot)).toBe(false);
-        });
-        it('should return false for a root path that is a sub-path of the path to check', () => {
-            const pathToCheck = path.resolve('/project/root/sub');
-            const rootSub = path.resolve('/project/root');
-            expect(isWithinRoot(pathToCheck, rootSub)).toBe(true);
-            const pathToCheckSuper = path.resolve('/project/root');
-            const rootSuper = path.resolve('/project/root/sub');
-            expect(isWithinRoot(pathToCheckSuper, rootSuper)).toBe(false);
+        const defaultRoot = path.resolve('/project/root');
+        it.each([
+            {
+                name: 'a path directly within the root',
+                path: path.join(defaultRoot, 'file.txt'),
+                expected: true,
+            },
+            {
+                name: 'a path in a subdirectory within the root',
+                path: path.join(defaultRoot, 'subdir', 'file.txt'),
+                expected: true,
+            },
+            { name: 'the root path itself', path: defaultRoot, expected: true },
+            {
+                name: 'a path with a trailing slash',
+                path: path.join(defaultRoot, 'file.txt') + path.sep,
+                expected: true,
+            },
+            {
+                name: 'the root path with a trailing slash',
+                path: defaultRoot + path.sep,
+                expected: true,
+            },
+            {
+                name: 'a sub-path of the path to check',
+                path: path.resolve('/project/root/sub'),
+                root: path.resolve('/project/root'),
+                expected: true,
+            },
+            {
+                name: 'a path outside the root',
+                path: path.resolve('/project/other', 'file.txt'),
+                expected: false,
+            },
+            {
+                name: 'an unrelated path',
+                path: path.resolve('/unrelated', 'file.txt'),
+                expected: false,
+            },
+            {
+                name: 'a path that only partially matches the root prefix',
+                path: path.resolve('/project/root-but-actually-different'),
+                expected: false,
+            },
+            {
+                name: 'a root path that is a sub-path of the path to check',
+                path: path.resolve('/project/root'),
+                root: path.resolve('/project/root/sub'),
+                expected: false,
+            },
+            {
+                name: 'a POSIX path inside',
+                path: '/project/root/file.txt',
+                root: '/project/root',
+                expected: true,
+            },
+            {
+                name: 'a POSIX path outside',
+                path: '/project/other/file.txt',
+                root: '/project/root',
+                expected: false,
+            },
+        ])('should return $expected for $name', ({ path: testPath, root, expected }) => {
+            expect(isWithinRoot(testPath, root || defaultRoot)).toBe(expected);
+        });
+    });
+    describe('fileExists', () => {
+        it('should return true if the file exists', async () => {
+            const testFile = path.join(tempRootDir, 'exists.txt');
+            actualNodeFs.writeFileSync(testFile, 'content');
+            await expect(fileExists(testFile)).resolves.toBe(true);
+        });
+        it('should return false if the file does not exist', async () => {
+            const testFile = path.join(tempRootDir, 'does-not-exist.txt');
+            await expect(fileExists(testFile)).resolves.toBe(false);
+        });
+        it('should return true for a directory that exists', async () => {
+            const testDir = path.join(tempRootDir, 'exists-dir');
+            actualNodeFs.mkdirSync(testDir);
+            await expect(fileExists(testDir)).resolves.toBe(true);
         });
     });
     describe('isBinaryFile', () => {
@@ -120,6 +180,298 @@ describe('fileUtils', () => {
             expect(await isBinaryFile(filePathForBinaryTest)).toBe(false);
         });
     });
+    describe('BOM detection and encoding', () => {
+        let testDir;
+        beforeEach(async () => {
+            testDir = await fsPromises.mkdtemp(path.join(await fsPromises.realpath(os.tmpdir()), 'fileUtils-bom-test-'));
+        });
+        afterEach(async () => {
+            if (testDir) {
+                await fsPromises.rm(testDir, { recursive: true, force: true });
+            }
+        });
+        describe('detectBOM', () => {
+            it('should detect UTF-8 BOM', () => {
+                const buf = Buffer.from([
+                    0xef, 0xbb, 0xbf, 0x48, 0x65, 0x6c, 0x6c, 0x6f,
+                ]);
+                const result = detectBOM(buf);
+                expect(result).toEqual({ encoding: 'utf8', bomLength: 3 });
+            });
+            it('should detect UTF-16 LE BOM', () => {
+                const buf = Buffer.from([0xff, 0xfe, 0x48, 0x00, 0x65, 0x00]);
+                const result = detectBOM(buf);
+                expect(result).toEqual({ encoding: 'utf16le', bomLength: 2 });
+            });
+            it('should detect UTF-16 BE BOM', () => {
+                const buf = Buffer.from([0xfe, 0xff, 0x00, 0x48, 0x00, 0x65]);
+                const result = detectBOM(buf);
+                expect(result).toEqual({ encoding: 'utf16be', bomLength: 2 });
+            });
+            it('should detect UTF-32 LE BOM', () => {
+                const buf = Buffer.from([
+                    0xff, 0xfe, 0x00, 0x00, 0x48, 0x00, 0x00, 0x00,
+                ]);
+                const result = detectBOM(buf);
+                expect(result).toEqual({ encoding: 'utf32le', bomLength: 4 });
+            });
+            it('should detect UTF-32 BE BOM', () => {
+                const buf = Buffer.from([
+                    0x00, 0x00, 0xfe, 0xff, 0x00, 0x00, 0x00, 0x48,
+                ]);
+                const result = detectBOM(buf);
+                expect(result).toEqual({ encoding: 'utf32be', bomLength: 4 });
+            });
+            it('should return null for no BOM', () => {
+                const buf = Buffer.from([0x48, 0x65, 0x6c, 0x6c, 0x6f]);
+                const result = detectBOM(buf);
+                expect(result).toBeNull();
+            });
+            it('should return null for empty buffer', () => {
+                const buf = Buffer.alloc(0);
+                const result = detectBOM(buf);
+                expect(result).toBeNull();
+            });
+            it('should return null for partial BOM', () => {
+                const buf = Buffer.from([0xef, 0xbb]); // Incomplete UTF-8 BOM
+                const result = detectBOM(buf);
+                expect(result).toBeNull();
+            });
+        });
+        describe('readFileWithEncoding', () => {
+            it('should read UTF-8 BOM file correctly', async () => {
+                const content = 'Hello, 世界! 🌍';
+                const utf8Bom = Buffer.from([0xef, 0xbb, 0xbf]);
+                const utf8Content = Buffer.from(content, 'utf8');
+                const fullBuffer = Buffer.concat([utf8Bom, utf8Content]);
+                const filePath = path.join(testDir, 'utf8-bom.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe(content);
+            });
+            it('should read UTF-16 LE BOM file correctly', async () => {
+                const content = 'Hello, 世界! 🌍';
+                const utf16leBom = Buffer.from([0xff, 0xfe]);
+                const utf16leContent = Buffer.from(content, 'utf16le');
+                const fullBuffer = Buffer.concat([utf16leBom, utf16leContent]);
+                const filePath = path.join(testDir, 'utf16le-bom.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe(content);
+            });
+            it('should read UTF-16 BE BOM file correctly', async () => {
+                const content = 'Hello, 世界! 🌍';
+                // Manually encode UTF-16 BE: each char as big-endian 16-bit
+                const utf16beBom = Buffer.from([0xfe, 0xff]);
+                const chars = Array.from(content);
+                const utf16beBytes = [];
+                for (const char of chars) {
+                    const code = char.codePointAt(0);
+                    if (code > 0xffff) {
+                        // Surrogate pair for emoji
+                        const surrogate1 = 0xd800 + ((code - 0x10000) >> 10);
+                        const surrogate2 = 0xdc00 + ((code - 0x10000) & 0x3ff);
+                        utf16beBytes.push((surrogate1 >> 8) & 0xff, surrogate1 & 0xff);
+                        utf16beBytes.push((surrogate2 >> 8) & 0xff, surrogate2 & 0xff);
+                    }
+                    else {
+                        utf16beBytes.push((code >> 8) & 0xff, code & 0xff);
+                    }
+                }
+                const utf16beContent = Buffer.from(utf16beBytes);
+                const fullBuffer = Buffer.concat([utf16beBom, utf16beContent]);
+                const filePath = path.join(testDir, 'utf16be-bom.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe(content);
+            });
+            it('should read UTF-32 LE BOM file correctly', async () => {
+                const content = 'Hello, 世界! 🌍';
+                const utf32leBom = Buffer.from([0xff, 0xfe, 0x00, 0x00]);
+                const utf32leBytes = [];
+                for (const char of Array.from(content)) {
+                    const code = char.codePointAt(0);
+                    utf32leBytes.push(code & 0xff, (code >> 8) & 0xff, (code >> 16) & 0xff, (code >> 24) & 0xff);
+                }
+                const utf32leContent = Buffer.from(utf32leBytes);
+                const fullBuffer = Buffer.concat([utf32leBom, utf32leContent]);
+                const filePath = path.join(testDir, 'utf32le-bom.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe(content);
+            });
+            it('should read UTF-32 BE BOM file correctly', async () => {
+                const content = 'Hello, 世界! 🌍';
+                const utf32beBom = Buffer.from([0x00, 0x00, 0xfe, 0xff]);
+                const utf32beBytes = [];
+                for (const char of Array.from(content)) {
+                    const code = char.codePointAt(0);
+                    utf32beBytes.push((code >> 24) & 0xff, (code >> 16) & 0xff, (code >> 8) & 0xff, code & 0xff);
+                }
+                const utf32beContent = Buffer.from(utf32beBytes);
+                const fullBuffer = Buffer.concat([utf32beBom, utf32beContent]);
+                const filePath = path.join(testDir, 'utf32be-bom.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe(content);
+            });
+            it('should read file without BOM as UTF-8', async () => {
+                const content = 'Hello, 世界!';
+                const filePath = path.join(testDir, 'no-bom.txt');
+                await fsPromises.writeFile(filePath, content, 'utf8');
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe(content);
+            });
+            it('should handle empty file', async () => {
+                const filePath = path.join(testDir, 'empty.txt');
+                await fsPromises.writeFile(filePath, '');
+                const result = await readFileWithEncoding(filePath);
+                expect(result).toBe('');
+            });
+        });
+        describe('isBinaryFile with BOM awareness', () => {
+            it('should not treat UTF-8 BOM file as binary', async () => {
+                const content = 'Hello, world!';
+                const utf8Bom = Buffer.from([0xef, 0xbb, 0xbf]);
+                const utf8Content = Buffer.from(content, 'utf8');
+                const fullBuffer = Buffer.concat([utf8Bom, utf8Content]);
+                const filePath = path.join(testDir, 'utf8-bom-test.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(false);
+            });
+            it('should not treat UTF-16 LE BOM file as binary', async () => {
+                const content = 'Hello, world!';
+                const utf16leBom = Buffer.from([0xff, 0xfe]);
+                const utf16leContent = Buffer.from(content, 'utf16le');
+                const fullBuffer = Buffer.concat([utf16leBom, utf16leContent]);
+                const filePath = path.join(testDir, 'utf16le-bom-test.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(false);
+            });
+            it('should not treat UTF-16 BE BOM file as binary', async () => {
+                const utf16beBom = Buffer.from([0xfe, 0xff]);
+                // Simple ASCII in UTF-16 BE
+                const utf16beContent = Buffer.from([
+                    0x00,
+                    0x48, // H
+                    0x00,
+                    0x65, // e
+                    0x00,
+                    0x6c, // l
+                    0x00,
+                    0x6c, // l
+                    0x00,
+                    0x6f, // o
+                    0x00,
+                    0x2c, // ,
+                    0x00,
+                    0x20, // space
+                    0x00,
+                    0x77, // w
+                    0x00,
+                    0x6f, // o
+                    0x00,
+                    0x72, // r
+                    0x00,
+                    0x6c, // l
+                    0x00,
+                    0x64, // d
+                    0x00,
+                    0x21, // !
+                ]);
+                const fullBuffer = Buffer.concat([utf16beBom, utf16beContent]);
+                const filePath = path.join(testDir, 'utf16be-bom-test.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(false);
+            });
+            it('should not treat UTF-32 LE BOM file as binary', async () => {
+                const utf32leBom = Buffer.from([0xff, 0xfe, 0x00, 0x00]);
+                const utf32leContent = Buffer.from([
+                    0x48,
+                    0x00,
+                    0x00,
+                    0x00, // H
+                    0x65,
+                    0x00,
+                    0x00,
+                    0x00, // e
+                    0x6c,
+                    0x00,
+                    0x00,
+                    0x00, // l
+                    0x6c,
+                    0x00,
+                    0x00,
+                    0x00, // l
+                    0x6f,
+                    0x00,
+                    0x00,
+                    0x00, // o
+                ]);
+                const fullBuffer = Buffer.concat([utf32leBom, utf32leContent]);
+                const filePath = path.join(testDir, 'utf32le-bom-test.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(false);
+            });
+            it('should not treat UTF-32 BE BOM file as binary', async () => {
+                const utf32beBom = Buffer.from([0x00, 0x00, 0xfe, 0xff]);
+                const utf32beContent = Buffer.from([
+                    0x00,
+                    0x00,
+                    0x00,
+                    0x48, // H
+                    0x00,
+                    0x00,
+                    0x00,
+                    0x65, // e
+                    0x00,
+                    0x00,
+                    0x00,
+                    0x6c, // l
+                    0x00,
+                    0x00,
+                    0x00,
+                    0x6c, // l
+                    0x00,
+                    0x00,
+                    0x00,
+                    0x6f, // o
+                ]);
+                const fullBuffer = Buffer.concat([utf32beBom, utf32beContent]);
+                const filePath = path.join(testDir, 'utf32be-bom-test.txt');
+                await fsPromises.writeFile(filePath, fullBuffer);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(false);
+            });
+            it('should still treat actual binary file as binary', async () => {
+                // PNG header + some binary data with null bytes
+                const pngHeader = Buffer.from([
+                    0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a,
+                ]);
+                const binaryData = Buffer.from([
+                    0x00, 0x00, 0x00, 0x0d, 0x49, 0x48, 0x44, 0x52,
+                ]); // IHDR chunk with nulls
+                const fullContent = Buffer.concat([pngHeader, binaryData]);
+                const filePath = path.join(testDir, 'test.png');
+                await fsPromises.writeFile(filePath, fullContent);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(true);
+            });
+            it('should treat file with null bytes (no BOM) as binary', async () => {
+                const content = Buffer.from([
+                    0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x00, 0x77, 0x6f, 0x72, 0x6c, 0x64,
+                ]);
+                const filePath = path.join(testDir, 'null-bytes.bin');
+                await fsPromises.writeFile(filePath, content);
+                const result = await isBinaryFile(filePath);
+                expect(result).toBe(true);
+            });
+        });
+    });
     describe('detectFileType', () => {
         let filePathForDetectTest;
         beforeEach(() => {
@@ -141,40 +493,24 @@ describe('fileUtils', () => {
             expect(await detectFileType('file.cts')).toBe('text');
             expect(await detectFileType('component.tsx')).toBe('text');
         });
-        it('should detect image type by extension (png)', async () => {
-            mockMimeLookup.mockReturnValueOnce('image/png');
-            expect(await detectFileType('file.png')).toBe('image');
-        });
-        it('should detect image type by extension (jpeg)', async () => {
-            mockMimeLookup.mockReturnValueOnce('image/jpeg');
-            expect(await detectFileType('file.jpg')).toBe('image');
+        it.each([
+            { type: 'image', file: 'file.png', mime: 'image/png' },
+            { type: 'image', file: 'file.jpg', mime: 'image/jpeg' },
+            { type: 'pdf', file: 'file.pdf', mime: 'application/pdf' },
+            { type: 'audio', file: 'song.mp3', mime: 'audio/mpeg' },
+            { type: 'video', file: 'movie.mp4', mime: 'video/mp4' },
+            { type: 'binary', file: 'archive.zip', mime: 'application/zip' },
+            { type: 'binary', file: 'app.exe', mime: 'application/octet-stream' },
+        ])('should detect $type type for $file by extension', async ({ file, mime, type }) => {
+            mockMimeGetType.mockReturnValueOnce(mime);
+            expect(await detectFileType(file)).toBe(type);
         });
         it('should detect svg type by extension', async () => {
             expect(await detectFileType('image.svg')).toBe('svg');
             expect(await detectFileType('image.icon.svg')).toBe('svg');
         });
-        it('should detect pdf type by extension', async () => {
-            mockMimeLookup.mockReturnValueOnce('application/pdf');
-            expect(await detectFileType('file.pdf')).toBe('pdf');
-        });
-        it('should detect audio type by extension', async () => {
-            mockMimeLookup.mockReturnValueOnce('audio/mpeg');
-            expect(await detectFileType('song.mp3')).toBe('audio');
-        });
-        it('should detect video type by extension', async () => {
-            mockMimeLookup.mockReturnValueOnce('video/mp4');
-            expect(await detectFileType('movie.mp4')).toBe('video');
-        });
-        it('should detect known binary extensions as binary (e.g. .zip)', async () => {
-            mockMimeLookup.mockReturnValueOnce('application/zip');
-            expect(await detectFileType('archive.zip')).toBe('binary');
-        });
-        it('should detect known binary extensions as binary (e.g. .exe)', async () => {
-            mockMimeLookup.mockReturnValueOnce('application/octet-stream'); // Common for .exe
-            expect(await detectFileType('app.exe')).toBe('binary');
-        });
         it('should use isBinaryFile for unknown extensions and detect as binary', async () => {
-            mockMimeLookup.mockReturnValueOnce(false); // Unknown mime type
+            mockMimeGetType.mockReturnValueOnce(false); // Unknown mime type
             // Create a file that isBinaryFile will identify as binary
             const binaryContent = Buffer.from([
                 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a,
@@ -183,7 +519,7 @@ describe('fileUtils', () => {
             expect(await detectFileType(filePathForDetectTest)).toBe('binary');
         });
         it('should default to text if mime type is unknown and content is not binary', async () => {
-            mockMimeLookup.mockReturnValueOnce(false); // Unknown mime type
+            mockMimeGetType.mockReturnValueOnce(false); // Unknown mime type
             // filePathForDetectTest is already a text file by default from beforeEach
             expect(await detectFileType(filePathForDetectTest)).toBe('text');
         });
@@ -223,7 +559,7 @@ describe('fileUtils', () => {
         });
         it('should handle read errors for image/pdf files', async () => {
             actualNodeFs.writeFileSync(testImageFilePath, 'content'); // File must exist
-            mockMimeLookup.mockReturnValue('image/png');
+            mockMimeGetType.mockReturnValue('image/png');
             const readError = new Error('Simulated image read error');
             vi.spyOn(fsPromises, 'readFile').mockRejectedValueOnce(readError);
             const result = await processSingleFileContent(testImageFilePath, tempRootDir, new StandardFileSystemService());
@@ -233,7 +569,7 @@ describe('fileUtils', () => {
         it('should process an image file', async () => {
             const fakePngData = Buffer.from('fake png data');
             actualNodeFs.writeFileSync(testImageFilePath, fakePngData);
-            mockMimeLookup.mockReturnValue('image/png');
+            mockMimeGetType.mockReturnValue('image/png');
             const result = await processSingleFileContent(testImageFilePath, tempRootDir, new StandardFileSystemService());
             expect(result.llmContent.inlineData).toBeDefined();
             expect(result.llmContent.inlineData
@@ -244,7 +580,7 @@ describe('fileUtils', () => {
         it('should process a PDF file', async () => {
             const fakePdfData = Buffer.from('fake pdf data');
             actualNodeFs.writeFileSync(testPdfFilePath, fakePdfData);
-            mockMimeLookup.mockReturnValue('application/pdf');
+            mockMimeGetType.mockReturnValue('application/pdf');
             const result = await processSingleFileContent(testPdfFilePath, tempRootDir, new StandardFileSystemService());
             expect(result.llmContent.inlineData).toBeDefined();
             expect(result.llmContent.inlineData
@@ -260,14 +596,14 @@ describe('fileUtils', () => {
   `;
             const testSvgFilePath = path.join(tempRootDir, 'test.svg');
             actualNodeFs.writeFileSync(testSvgFilePath, svgContent, 'utf-8');
-            mockMimeLookup.mockReturnValue('image/svg+xml');
+            mockMimeGetType.mockReturnValue('image/svg+xml');
             const result = await processSingleFileContent(testSvgFilePath, tempRootDir, new StandardFileSystemService());
             expect(result.llmContent).toBe(svgContent);
             expect(result.returnDisplay).toContain('Read SVG as text');
         });
         it('should skip binary files', async () => {
             actualNodeFs.writeFileSync(testBinaryFilePath, Buffer.from([0x00, 0x01, 0x02]));
-            mockMimeLookup.mockReturnValueOnce('application/octet-stream');
+            mockMimeGetType.mockReturnValueOnce('application/octet-stream');
             // isBinaryFile will operate on the real file.
             const result = await processSingleFileContent(testBinaryFilePath, tempRootDir, new StandardFileSystemService());
             expect(result.llmContent).toContain('Cannot display content of binary file');
@@ -350,14 +686,22 @@ describe('fileUtils', () => {
             expect(result.returnDisplay).toBe('Read lines 1-10 of 20 from test.txt (some lines were shortened)');
         });
         it('should return an error if the file size exceeds 20MB', async () => {
-            // Create a file just over 20MB
-            const twentyOneMB = 21 * 1024 * 1024;
-            const buffer = Buffer.alloc(twentyOneMB, 0x61); // Fill with 'a'
-            actualNodeFs.writeFileSync(testTextFilePath, buffer);
-            const result = await processSingleFileContent(testTextFilePath, tempRootDir, new StandardFileSystemService());
-            expect(result.error).toContain('File size exceeds the 20MB limit');
-            expect(result.returnDisplay).toContain('File size exceeds the 20MB limit');
-            expect(result.llmContent).toContain('File size exceeds the 20MB limit');
+            // Create a small test file
+            actualNodeFs.writeFileSync(testTextFilePath, 'test content');
+            // Spy on fs.promises.stat to return a large file size
+            const statSpy = vi.spyOn(fs.promises, 'stat').mockResolvedValueOnce({
+                size: 21 * 1024 * 1024,
+                isDirectory: () => false,
+            });
+            try {
+                const result = await processSingleFileContent(testTextFilePath, tempRootDir, new StandardFileSystemService());
+                expect(result.error).toContain('File size exceeds the 20MB limit');
+                expect(result.returnDisplay).toContain('File size exceeds the 20MB limit');
+                expect(result.llmContent).toContain('File size exceeds the 20MB limit');
+            }
+            finally {
+                statSpy.mockRestore();
+            }
         });
     });
 });