diff options
Diffstat (limited to 'packages/core/src/tools/read-many-files.test.ts')
| -rw-r--r-- | packages/core/src/tools/read-many-files.test.ts | 27 |
1 files changed, 27 insertions, 0 deletions
diff --git a/packages/core/src/tools/read-many-files.test.ts b/packages/core/src/tools/read-many-files.test.ts index 666c484c..697b7d1b 100644 --- a/packages/core/src/tools/read-many-files.test.ts +++ b/packages/core/src/tools/read-many-files.test.ts @@ -14,6 +14,33 @@ import fs from 'fs'; // Actual fs for setup import os from 'os'; import { Config } from '../config/config.js'; +vi.mock('mime-types', () => { + const lookup = (filename: string) => { + if (filename.endsWith('.ts') || filename.endsWith('.js')) { + return 'text/plain'; + } + if (filename.endsWith('.png')) { + return 'image/png'; + } + if (filename.endsWith('.pdf')) { + return 'application/pdf'; + } + if (filename.endsWith('.mp3') || filename.endsWith('.wav')) { + return 'audio/mpeg'; + } + if (filename.endsWith('.mp4') || filename.endsWith('.mov')) { + return 'video/mp4'; + } + return false; + }; + return { + default: { + lookup, + }, + lookup, + }; +}); + describe('ReadManyFilesTool', () => { let tool: ReadManyFilesTool; let tempRootDir: string; |
