test: fix tests on base

morpho-org · May 13, 2024 · de336f3 · de336f3
1 parent 2c982d4
commit de336f3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/forge/fork/migration/AaveV3MigrationBundlerForkTest.sol b/test/forge/fork/migration/AaveV3MigrationBundlerForkTest.sol
@@ -17,13 +17,17 @@ contract AaveV3MigrationBundlerForkTest is MigrationForkTest {
 
     uint256 public constant RATE_MODE = 2;
 
-    uint256 collateralSupplied = 10_000 ether;
+    uint256 collateralSupplied = 1_000 ether;
     uint256 borrowed = 1 ether;
 
     function setUp() public override {
         super.setUp();
 
-        _initMarket(WST_ETH, WETH);
+        if (block.chainid == 1) {
+            _initMarket(DAI, WETH);
+        } else if (block.chainid == 8453) {
+            _initMarket(CB_ETH, WETH);
+        }
 
         vm.label(AAVE_V3_POOL, "Aave V3 Pool");