test addition for get transient

Author: benzekrimaha

tests/mocks/mongoClient.js

@@ -5,6 +5,8 @@ module.exports = {
     setup: jest.fn(),
     close: jest.fn(),
     getObjectMDStats: jest.fn(),
+    isLocationTransient: jest.fn(),
+    getObject: jest.fn(),
     getBucketInfos: jest.fn(),
     updateStorageConsumptionMetrics: jest.fn(),
     getUsersBucketCreationDate: jest.fn(),

tests/unit/CountItems/CountWorker.js

@@ -216,4 +216,78 @@ describe('CountItems::CountWorker', () => {
             done();
         });
     });
+
+    test('should use real getIsTransient and not crash', done => {
+        const testSendFn = jest.fn();
+        let getObjectCall = 0;
+        const w = new CountWorker({
+            log: new DummyLogger(),
+            sendFn: testSendFn,
+            client: mongoMock,
+        });
+        mongoMock.setup.mockImplementationOnce(cb => cb());
+        mongoMock.close.mockImplementationOnce(cb => cb());
+        mongoMock.client.isConnected.mockImplementationOnce(() => false);
+        mongoMock.getObjectMDStats.mockImplementationOnce((_a, _b, _c, _d, cb) => cb(null, { value: 42 }));
+        mongoMock.isLocationTransient.mockImplementationOnce((_a, _b, cb) => {
+            cb(null, false);
+        });
+        mongoMock.getObject = jest.fn((_a, _b, _c, _d, cb) => {
+            getObjectCall += 1;
+            if (getObjectCall === 1) {
+                cb(null, 'v1');
+            } else {
+                cb(null, {
+                    locations: {
+                        undefined: { isTransient: true },
+                    },
+                });
+            }
+        });
+        const bucketInfo = {
+            _name: 'test-bucket',
+            _owner: 'any',
+            _ownerDisplayName: 'any',
+            _creationDate: Date.now().toString(),
+            getLocationConstraint: jest.fn(() => 'test-location'),
+        };
+        w.countItems(bucketInfo, (err, results) => {
+            expect(err).toBeNull();
+            expect(results).toEqual({ value: 42 });
+            done();
+        });
+    });
+    
+    test('should call isLocationTransient if defined', done => {
+        const testSendFn = jest.fn();
+        const w = new CountWorker({
+            log: new DummyLogger(),
+            sendFn: testSendFn,
+            client: mongoMock,
+        });
+    
+        mongoMock.setup.mockImplementationOnce(cb => cb());
+        mongoMock.close.mockImplementationOnce(cb => cb());
+        mongoMock.client.isConnected.mockImplementationOnce(() => false);
+        mongoMock.isLocationTransient.mockImplementationOnce((_a, _b, cb) => {
+            cb(null, true);
+        });
+        mongoMock.getObjectMDStats.mockImplementationOnce((_a, _b, _c, _d, cb) => cb(null, { value: 42 }));
+    
+        const bucketInfo = {
+            getLocationConstraint: () => 'some-location',
+            _name: 'test-bucket',
+            _owner: 'any',
+            _ownerDisplayName: 'any',
+            _creationDate: Date.now().toString(),
+        };
+    
+        w.countItems(bucketInfo, (err, results) => {
+            expect(err).toBeNull();
+            expect(results).toEqual({ value: 42 });
+            expect(mongoMock.isLocationTransient).toHaveBeenCalledTimes(2);
+            done();
+        });
+    });
+    
 });