fixup on tests

Author: benzekrimaha

tests/mocks/mongoClient.js

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

tests/unit/CountItems/CountWorker.js

@@ -229,6 +229,9 @@ describe('CountItems::CountWorker', () => {
         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) {
@@ -246,11 +249,45 @@ describe('CountItems::CountWorker', () => {
             _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();
+        });
+    });
+    
 });