@quatrain/worker 1.2.8 → 1.2.10

package/dist/FileSystem.test.js

@@ -44,7 +44,14 @@ const node_fetch_native_1 = __importDefault(require("node-fetch-native"));
 const ffmpeg = __importStar(require("fluent-ffmpeg"));
 const node_stream_1 = require("node:stream");
 jest.mock('axios');
-jest.mock('node-fetch-native');
+jest.mock('node-fetch-native', () => {
+    const fn = jest.fn();
+    return {
+        __esModule: true,
+        default: fn,
+        fetch: fn,
+    };
+});
 jest.mock('fluent-ffmpeg', () => ({
     ffprobe: jest.fn()
 }));
@@ -88,32 +95,38 @@ describe('FileSystem Utilities (Worker)', () => {
             return node_stream_1.Readable.from(Buffer.alloc(35 * 1024));
         });
         jest.spyOn(node_fs_1.default, 'existsSync').mockImplementation((p) => {
-            if (typeof p === 'string' && (p.endsWith('test.txt') || p.endsWith('small.mp4') || p.endsWith('large.mp4') || p.endsWith('large-fail.mp4'))) {
+            if (typeof p === 'string' && (p.endsWith('test.txt') || p.endsWith('small.mp4') || p.endsWith('large.mp4') || p.endsWith('large-fail.mp4') || p.endsWith('small-fail.mp4') || p.endsWith('large-sync-fail.mp4'))) {
                 return true;
             }
             return originalExistsSync(p);
         });
         jest.spyOn(node_fs_1.default, 'statSync').mockImplementation((p) => {
             if (typeof p === 'string') {
-                if (p.endsWith('small.mp4')) {
+                if (p.endsWith('small.mp4') || p.endsWith('small-fail.mp4')) {
                     return { size: 13 };
                 }
-                if (p.endsWith('large.mp4') || p.endsWith('large-fail.mp4')) {
+                if (p.endsWith('large.mp4') || p.endsWith('large-fail.mp4') || p.endsWith('large-sync-fail.mp4')) {
                     return { size: 35 * 1024 };
                 }
             }
             return (jest.requireActual('node:fs').statSync)(p);
         });
         // Setup base temp folder if needed
-        if (originalExistsSync(testBaseDir)) {
-            node_fs_1.default.rmSync(testBaseDir, { recursive: true, force: true });
+        try {
+            if (originalExistsSync(testBaseDir)) {
+                node_fs_1.default.rmSync(testBaseDir, { recursive: true, force: true });
+            }
         }
+        catch { }
     });
     afterEach(() => {
         // Clean up base temp folder
-        if (originalExistsSync(testBaseDir)) {
-            node_fs_1.default.rmSync(testBaseDir, { recursive: true, force: true });
+        try {
+            if (originalExistsSync(testBaseDir)) {
+                node_fs_1.default.rmSync(testBaseDir, { recursive: true, force: true });
+            }
         }
+        catch { }
         jest.clearAllMocks();
         jest.restoreAllMocks();
     });
@@ -252,8 +265,18 @@ describe('FileSystem Utilities (Worker)', () => {
             ffmpeg.ffprobe.mockImplementation((file, callback) => {
                 callback(null, mockMetadata);
             });
-            node_fetch_native_1.default.mockResolvedValue({
-                ok: true
+            node_fetch_native_1.default.mockImplementation((url, options) => {
+                return new Promise((resolve) => {
+                    if (options.body && typeof options.body.on === 'function') {
+                        options.body.on('data', () => { });
+                        options.body.on('end', () => {
+                            resolve({ ok: true });
+                        });
+                    }
+                    else {
+                        resolve({ ok: true });
+                    }
+                });
             });
             const meta = {
                 bucket: 'mock-bucket',
@@ -281,5 +304,42 @@ describe('FileSystem Utilities (Worker)', () => {
             };
             await expect(FileSystem_1.FileSystem.uploadFile(testFile, meta)).rejects.toThrow('Connection abort');
         });
+        it('should reject small uploads if HTTP request fails', async () => {
+            const testFile = node_path_1.default.join(testBaseDir, 'small-fail.mp4');
+            const mockMetadata = {
+                streams: [{ width: 320, height: 240, duration: '1', bit_rate: '100', nb_frames: '10' }]
+            };
+            ffmpeg.ffprobe.mockImplementation((file, callback) => {
+                callback(null, mockMetadata);
+            });
+            node_fetch_native_1.default.mockRejectedValue(new Error('Upload failed'));
+            // Stub readFileSync for small content read
+            jest.spyOn(node_fs_1.default, 'readFileSync').mockReturnValue(Buffer.from('small content'));
+            const meta = {
+                bucket: 'mock-bucket',
+                ref: 'remote/small-fail.mp4',
+                uploadUrl: 'https://upload.com/small-fail.mp4'
+            };
+            await expect(FileSystem_1.FileSystem.uploadFile(testFile, meta)).rejects.toThrow('Upload failed');
+        });
+        it('should handle synchronous errors inside stream initialization', async () => {
+            const testFile = node_path_1.default.join(testBaseDir, 'large-sync-fail.mp4');
+            const mockMetadata = {
+                streams: [{ width: 1280, height: 720, duration: '5', bit_rate: '2000', nb_frames: '150' }]
+            };
+            ffmpeg.ffprobe.mockImplementation((file, callback) => {
+                callback(null, mockMetadata);
+            });
+            // Make createReadStream throw a synchronous error
+            jest.spyOn(node_fs_1.default, 'createReadStream').mockImplementationOnce(() => {
+                throw new Error('Sync stream error');
+            });
+            const meta = {
+                bucket: 'mock-bucket',
+                ref: 'remote/large-sync-fail.mp4',
+                uploadUrl: 'https://upload.com/large-sync-fail.mp4'
+            };
+            await expect(FileSystem_1.FileSystem.uploadFile(testFile, meta)).rejects.toThrow('Sync stream error');
+        });
     });
 });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@quatrain/worker",
-  "version": "1.2.8",
+  "version": "1.2.10",
   "description": "Container Worker helpers",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -34,9 +34,9 @@
     "typescript": "^5.1.5"
   },
   "dependencies": {
-    "@quatrain/core": "^1.2.14",
+    "@quatrain/core": "^1.2.17",
     "@quatrain/queue": "^1.2.3",
-    "@quatrain/storage": "^1.2.9",
+    "@quatrain/storage": "^1.2.13",
     "axios": "^1.7.7",
     "fluent-ffmpeg": "^2.1.2",
     "fs-extra": "^11.2.0",

package/src/FileSystem.test.ts

@@ -7,7 +7,14 @@ import * as ffmpeg from 'fluent-ffmpeg'
 import { PassThrough, Readable } from 'node:stream'
 
 jest.mock('axios')
-jest.mock('node-fetch-native')
+jest.mock('node-fetch-native', () => {
+   const fn = jest.fn()
+   return {
+      __esModule: true,
+      default: fn,
+      fetch: fn,
+   }
+})
 jest.mock('fluent-ffmpeg', () => ({
    ffprobe: jest.fn()
 }))
@@ -57,7 +64,7 @@ describe('FileSystem Utilities (Worker)', () => {
       })
 
       jest.spyOn(fs, 'existsSync').mockImplementation((p) => {
-         if (typeof p === 'string' && (p.endsWith('test.txt') || p.endsWith('small.mp4') || p.endsWith('large.mp4') || p.endsWith('large-fail.mp4'))) {
+         if (typeof p === 'string' && (p.endsWith('test.txt') || p.endsWith('small.mp4') || p.endsWith('large.mp4') || p.endsWith('large-fail.mp4') || p.endsWith('small-fail.mp4') || p.endsWith('large-sync-fail.mp4'))) {
             return true
          }
          return originalExistsSync(p)
@@ -65,10 +72,10 @@ describe('FileSystem Utilities (Worker)', () => {
 
       jest.spyOn(fs, 'statSync').mockImplementation((p) => {
          if (typeof p === 'string') {
-            if (p.endsWith('small.mp4')) {
+            if (p.endsWith('small.mp4') || p.endsWith('small-fail.mp4')) {
                return { size: 13 } as any
             }
-            if (p.endsWith('large.mp4') || p.endsWith('large-fail.mp4')) {
+            if (p.endsWith('large.mp4') || p.endsWith('large-fail.mp4') || p.endsWith('large-sync-fail.mp4')) {
                return { size: 35 * 1024 } as any
             }
          }
@@ -76,16 +83,20 @@ describe('FileSystem Utilities (Worker)', () => {
       })
 
       // Setup base temp folder if needed
-      if (originalExistsSync(testBaseDir)) {
-         fs.rmSync(testBaseDir, { recursive: true, force: true })
-      }
+      try {
+         if (originalExistsSync(testBaseDir)) {
+            fs.rmSync(testBaseDir, { recursive: true, force: true })
+         }
+      } catch {}
    })
 
    afterEach(() => {
       // Clean up base temp folder
-      if (originalExistsSync(testBaseDir)) {
-         fs.rmSync(testBaseDir, { recursive: true, force: true })
-      }
+      try {
+         if (originalExistsSync(testBaseDir)) {
+            fs.rmSync(testBaseDir, { recursive: true, force: true })
+         }
+      } catch {}
       jest.clearAllMocks()
       jest.restoreAllMocks()
    })
@@ -255,8 +266,17 @@ describe('FileSystem Utilities (Worker)', () => {
             callback(null, mockMetadata)
          });
 
-         (fetch as any).mockResolvedValue({
-            ok: true
+         (fetch as any).mockImplementation((url: string, options: any) => {
+            return new Promise((resolve) => {
+               if (options.body && typeof options.body.on === 'function') {
+                  options.body.on('data', () => {})
+                  options.body.on('end', () => {
+                     resolve({ ok: true })
+                  })
+               } else {
+                  resolve({ ok: true })
+               }
+            })
          })
 
          const meta = {
@@ -291,5 +311,53 @@ describe('FileSystem Utilities (Worker)', () => {
 
          await expect(FileSystem.uploadFile(testFile, meta)).rejects.toThrow('Connection abort')
       })
+
+      it('should reject small uploads if HTTP request fails', async () => {
+         const testFile = path.join(testBaseDir, 'small-fail.mp4')
+
+         const mockMetadata = {
+            streams: [{ width: 320, height: 240, duration: '1', bit_rate: '100', nb_frames: '10' }]
+         };
+         (ffmpeg.ffprobe as any).mockImplementation((file: string, callback: any) => {
+            callback(null, mockMetadata)
+         });
+
+         (fetch as any).mockRejectedValue(new Error('Upload failed'))
+
+         // Stub readFileSync for small content read
+         jest.spyOn(fs, 'readFileSync').mockReturnValue(Buffer.from('small content'))
+
+         const meta = {
+            bucket: 'mock-bucket',
+            ref: 'remote/small-fail.mp4',
+            uploadUrl: 'https://upload.com/small-fail.mp4'
+         }
+
+         await expect(FileSystem.uploadFile(testFile, meta)).rejects.toThrow('Upload failed')
+      })
+
+      it('should handle synchronous errors inside stream initialization', async () => {
+         const testFile = path.join(testBaseDir, 'large-sync-fail.mp4')
+
+         const mockMetadata = {
+            streams: [{ width: 1280, height: 720, duration: '5', bit_rate: '2000', nb_frames: '150' }]
+         };
+         (ffmpeg.ffprobe as any).mockImplementation((file: string, callback: any) => {
+            callback(null, mockMetadata)
+         });
+
+         // Make createReadStream throw a synchronous error
+         jest.spyOn(fs, 'createReadStream').mockImplementationOnce(() => {
+            throw new Error('Sync stream error')
+         })
+
+         const meta = {
+            bucket: 'mock-bucket',
+            ref: 'remote/large-sync-fail.mp4',
+            uploadUrl: 'https://upload.com/large-sync-fail.mp4'
+         }
+
+         await expect(FileSystem.uploadFile(testFile, meta)).rejects.toThrow('Sync stream error')
+      })
    })
 })