test: fix flaky test (#1326)

FuelLabs · Oct 10, 2023 · 3b8648b · 3b8648b
1 parent fec3f74
commit 3b8648b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/.changeset/nervous-bottles-remember.md b/.changeset/nervous-bottles-remember.md
@@ -0,0 +1,2 @@
+---
+---
diff --git a/packages/fuel-gauge/src/multi-token-contract.test.ts b/packages/fuel-gauge/src/multi-token-contract.test.ts
@@ -77,12 +77,14 @@ describe('MultiTokenContract', () => {
     };
 
     // validates contract has expected balance after mint
-    subIds.forEach(async (subId) => {
+    const validateMintPromises = subIds.map(async (subId) => {
       expect(bn(await getBalance(contractId, helperDict[subId].assetId)).toNumber()).toBe(
         helperDict[subId].amount
       );
     });
 
+    await Promise.all(validateMintPromises);
+
     // transfer coins to user wallet
     await multiTokenContract
       .multiCall(
@@ -96,15 +98,16 @@ describe('MultiTokenContract', () => {
       )
       .call();
 
-    subIds.forEach(async (subId) => {
+    const validateTransferPromises = subIds.map(async (subId) => {
       // validates that user wallet has expected balance after transfer
       expect(bn(await userWallet.getBalance(helperDict[subId].assetId)).toNumber()).toBe(
         helperDict[subId].amount
       );
-
       // validates contract has not balance after transfer
       expect(bn(await getBalance(contractId, helperDict[subId].assetId)).toNumber()).toBe(0);
     });
+
+    await Promise.all(validateTransferPromises);
   });
 
   it('can burn coins', async () => {
@@ -158,12 +161,14 @@ describe('MultiTokenContract', () => {
     };
 
     // validates contract has expected balance after mint
-    subIds.forEach(async (subId) => {
+    const validateMintPromises = subIds.map(async (subId) => {
       expect(bn(await getBalance(contractId, helperDict[subId].assetId)).toNumber()).toBe(
         helperDict[subId].amount
       );
     });
 
+    await Promise.all(validateMintPromises);
+
     // burning coins
     await multiTokenContract
       .multiCall(
@@ -173,11 +178,13 @@ describe('MultiTokenContract', () => {
       )
       .call();
 
-    subIds.forEach(async (subId) => {
+    const validateBurnPromises = subIds.map(async (subId) => {
       // validates contract has expected balance for each coin after burn
       expect(bn(await getBalance(contractId, helperDict[subId].assetId)).toNumber()).toBe(
         helperDict[subId].amount - helperDict[subId].amountToBurn
       );
     });
+
+    await Promise.all(validateBurnPromises);
   });
 });