test: fix chain names

morpho-org · Oct 31, 2022 · 3fda517 · 3fda517
1 parent 595fc16
commit 3fda517
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/test-foundry/aave-v3/TestRepay.t.sol b/test-foundry/aave-v3/TestRepay.t.sol
@@ -348,7 +348,7 @@ contract TestRepay is TestSetup {
 
     function testDeltaRepay() public {
         // Allows only 10 unmatch borrowers
-        if (block.chainid == Chains.POLYGON_MAINNET || block.chainid == Chains.ETH_MAINNET)
+        if (block.chainid == Chains.Polygon || block.chainid == Chains.Mainnet)
             setDefaultMaxGasForMatchingHelper(3e6, 3e6, 3e6, 1.1e6);
         else setDefaultMaxGasForMatchingHelper(3e6, 3e6, 3e6, 1.2e6);
 

diff --git a/test-foundry/aave-v3/TestWithdraw.t.sol b/test-foundry/aave-v3/TestWithdraw.t.sol
@@ -367,7 +367,7 @@ contract TestWithdraw is TestSetup {
 
     function testDeltaWithdraw() public {
         // Allows only 10 unmatch borrowers
-        if (block.chainid == Chains.POLYGON_MAINNET)
+        if (block.chainid == Chains.Polygon)
             setDefaultMaxGasForMatchingHelper(3e6, 3e6, 1.2e6, 3e6);
         else setDefaultMaxGasForMatchingHelper(3e6, 3e6, 1.2e6, 3e6);
 

diff --git a/test-foundry/fuzzing/aave-v2/setup/TestSetupFuzzing.sol b/test-foundry/fuzzing/aave-v2/setup/TestSetupFuzzing.sol
@@ -123,9 +123,9 @@ contract TestSetupFuzzing is Config, Utils, stdCheats {
         treasuryVault = new User(morpho);
         morpho.setTreasuryVault(address(treasuryVault));
 
-        if (block.chainid == Chains.ETH_MAINNET || block.chainid == Chains.AVALANCHE_MAINNET) {
+        if (block.chainid == Chains.Mainnet || block.chainid == Chains.Avalanche) {
             rewardsManagerImplV1 = new RewardsManagerOnMainnetAndAvalanche();
-        } else if (block.chainid == Chains.POLYGON_MAINNET) {
+        } else if (block.chainid == Chains.Polygon) {
             rewardsManagerImplV1 = new RewardsManagerOnPolygon();
         }