summaryrefslogtreecommitdiff
path: root/packages/core/src/tools/read-many-files.test.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/core/src/tools/read-many-files.test.ts')
-rw-r--r--packages/core/src/tools/read-many-files.test.ts27
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;