test: upgrade lens

Author: MerlinEgalite

Makefile

@@ -37,7 +37,7 @@ else
       FOUNDRY_FORK_BLOCK_NUMBER?=29116728
       FOUNDRY_CONTRACT_PATTERN_INVERSE=(Fees|IncentivesVault|Rewards)
     endif
-  
+
     FOUNDRY_CHAIN_ID=137
     FOUNDRY_FORK_BLOCK_NUMBER?=22116728
   endif
@@ -121,7 +121,7 @@ single-% s-%:
 storage-layout-generate:
 	@./scripts/storage-layout.sh generate snapshots/.storage-layout-${PROTOCOL} Morpho RewardsManager Lens
 
-storage-layout-check: 
+storage-layout-check:
 	@./scripts/storage-layout.sh check snapshots/.storage-layout-${PROTOCOL} Morpho RewardsManager Lens
 
 config:

test-foundry/prod/aave-v2/TestLens.t.sol

@@ -0,0 +1,36 @@
+// SPDX-License-Identifier: GNU AGPLv3
+pragma solidity 0.8.13;
+
+import "./setup/TestSetup.sol";
+
+contract TestBorrow is TestSetup {
+    using ReserveConfiguration for DataTypes.ReserveConfigurationMap;
+    using WadRayMath for uint256;
+    using Math for uint256;
+
+    function testUpgradeLens() public {
+        Lens lensImplV2 = new Lens(address(morpho));
+
+        (
+            uint256 p2pSupplyIndexBefore,
+            uint256 p2pBorrowIndexBefore,
+            uint256 poolSupplyIndexBefore,
+            uint256 poolBorrowIndexBefore
+        ) = lens.getIndexes(aStEth);
+
+        vm.prank(proxyAdmin.owner());
+        proxyAdmin.upgrade(lensProxy, address(lensImplV2));
+
+        (
+            uint256 p2pSupplyIndexAfter,
+            uint256 p2pBorrowIndexAfter,
+            uint256 poolSupplyIndexAfter,
+            uint256 poolBorrowIndexAfter
+        ) = lens.getIndexes(aStEth);
+
+        assertEq(p2pSupplyIndexAfter, p2pSupplyIndexBefore);
+        assertEq(p2pBorrowIndexAfter, p2pBorrowIndexBefore);
+        assertEq(poolSupplyIndexAfter, poolSupplyIndexBefore);
+        assertEq(poolBorrowIndexAfter, poolBorrowIndexBefore);
+    }
+}