Update InvariantBridge.t.sol

immutable · Feb 21, 2024 · 32e49ed · 32e49ed
1 parent da6dcf3
commit 32e49ed
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/test/invariant/InvariantBridge.t.sol b/test/invariant/InvariantBridge.t.sol
@@ -215,7 +215,14 @@ contract InvariantBridge is Test {
             vm.selectFork(childId);
             uint256 totalSupply = ChildERC20(childToken).totalSupply();
 
+            uint256 userBalanceSum = 0;
+            for (uint256 j = 0; j < NO_OF_USERS; j++) {
+                address user = users[j];
+                userBalanceSum += ChildERC20(childToken).balanceOf(user);
+            }
+
             assertEq(bridgeBalance, totalSupply);
+            assertEq(bridgeBalance, userBalanceSum);
         }
     }
 }