add retries to tests

Author: YaroShkvorets

packages/cli/src/command-helpers/contracts.test.ts

@@ -113,33 +113,54 @@ const TEST_SOURCIFY_CONTRACT_INFO = {
   },
 };
 
-describe('getStartBlockForContract', { sequential: true }, async () => {
+// Retry helper with configurable number of retries
+async function retry<T>(operation: () => Promise<T>, maxRetries = 3, sleepMs = 5000): Promise<T> {
+  let lastError: Error | undefined;
+  for (let attempt = 0; attempt < maxRetries; attempt++) {
+    try {
+      return await operation();
+    } catch (error) {
+      lastError = error as Error;
+      if (attempt < maxRetries - 1) {
+        await new Promise(resolve => setTimeout(resolve, sleepMs));
+      }
+    }
+  }
+  throw lastError;
+}
+
+describe('getStartBlockForContract', { concurrent: true }, async () => {
   const registry = await loadRegistry();
   const contractService = new ContractService(registry);
   for (const [network, contracts] of Object.entries(TEST_CONTRACT_START_BLOCKS)) {
     for (const [contract, startBlockExp] of Object.entries(contracts)) {
       test(
         `Returns the start block ${network} ${contract} ${startBlockExp}`,
-        async () => {
-          //loop through the TEST_CONTRACT_START_BLOCKS object and test each network
-          const startBlock = await contractService.getStartBlock(network, contract);
+        { timeout: 50_000 },
+        async ({ expect }) => {
+          const startBlock = await retry(
+            () => contractService.getStartBlock(network, contract),
+            10,
+          );
           expect(parseInt(startBlock)).toBe(startBlockExp);
         },
-        { timeout: 10_000 },
       );
     }
   }
 });
 
-describe('getFromSourcifyForContract', { sequential: true }, async () => {
+describe('getFromSourcifyForContract', { concurrent: true }, async () => {
   const registry = await loadRegistry();
   const contractService = new ContractService(registry);
   for (const [networkId, contractInfo] of Object.entries(TEST_SOURCIFY_CONTRACT_INFO)) {
     for (const [contract, t] of Object.entries(contractInfo)) {
       test(
         `Returns contract information ${networkId} ${contract} ${t.name} ${t.startBlock}`,
+        { timeout: 50_000 },
         async () => {
-          const result = await contractService.getFromSourcify(EthereumABI, networkId, contract);
+          const result = await retry(() =>
+            contractService.getFromSourcify(EthereumABI, networkId, contract),
+          );
           if (t.name === null && t.startBlock === null) {
             expect(result).toBeNull();
           } else {
@@ -148,7 +169,6 @@ describe('getFromSourcifyForContract', { sequential: true }, async () => {
             expect(t).toEqual({ name, startBlock: parseInt(startBlock) });
           }
         },
-        { timeout: 10_000 },
       );
     }
   }