refresh multiple contracts ids

Author: andreespirela

package.json

@@ -1,6 +1,6 @@
 {
   "name": "verto-cache-interface",
-  "version": "1.0.6",
+  "version": "1.0.7",
   "description": "A communication package with Verto Cache System",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

src/hooks/cache-contract-hook.ts

@@ -9,11 +9,21 @@ import {CacheInterfaceConstants} from "../constants";
  * @return the result of the call for {@param action}
  */
 export const cacheContractHook = async (action: () => Promise<any> | any,
-                                        contractId?: string,
+                                        contractId?: string | string[],
                                         refreshCommunityContract?: boolean) => {
     let response = await action();
     if(contractId) {
-        await cacheContract(contractId);
+        let ids: Array<string> = [];
+
+        if(!Array.isArray(contractId)) {
+            ids = [contractId];
+        } else {
+            ids = [...contractId];
+        }
+
+        for (let id of ids) {
+            await cacheContract(id);
+        }
     }
     if(refreshCommunityContract) {
         await cacheContract(CacheInterfaceConstants.COMMUNITY_CONTRACT);

src/tests/hooks.test.ts

@@ -28,4 +28,16 @@ describe('Test hooks', () => {
         expect(spy).toHaveBeenCalledWith(CacheInterfaceConstants.COMMUNITY_CONTRACT);
         expect(spy).toHaveBeenCalledTimes(2);
     });
+
+    test('Cache contract hook with multiple contracts to refresh', async () => {
+        const spy = jest.spyOn(cacheContractsModule, 'cacheContract');
+        spy.mockImplementation();
+        const fn = jest.fn();
+        await cacheContractHook(fn, ['ABC', 'DEF'], true);
+        expect(fn).toHaveBeenCalled();
+        expect(spy).toHaveBeenCalledWith('ABC');
+        expect(spy).toHaveBeenCalledWith('DEF');
+        expect(spy).toHaveBeenCalledWith(CacheInterfaceConstants.COMMUNITY_CONTRACT);
+        expect(spy).toHaveBeenCalledTimes(3);
+    });
 })