fix: upgrade

morpho-org · Oct 13, 2022 · ff82027 · ff82027
1 parent 9279c6b
commit ff82027
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test-foundry/prod/aave-v2/TestLens.t.sol b/test-foundry/prod/aave-v2/TestLens.t.sol
@@ -15,15 +15,12 @@ contract TestLens is TestSetup {
         uint256 poolBorrowIndex;
     }
 
-    function testUpgradeLens() public {
+    function testIndexesShouldBeEqualAfterUpgrade() public {
         Lens lensImplV2 = new Lens(address(morpho));
 
         address[] memory marketsCreated = lens.getAllMarkets();
         Vars[] memory expectedValues = new Vars[](marketsCreated.length);
 
-        vm.prank(proxyAdmin.owner());
-        proxyAdmin.upgrade(lensProxy, address(lensImplV2));
-
         for (uint256 i; i < marketsCreated.length; ++i) {
             address market = marketsCreated[i];
             if (market == aStEth) continue;
@@ -41,6 +38,9 @@ contract TestLens is TestSetup {
             expectedValues[i].poolBorrowIndex = poolBorrowIndex;
         }
 
+        vm.prank(proxyAdmin.owner());
+        proxyAdmin.upgrade(lensProxy, address(lensImplV2));
+
         for (uint256 i; i < marketsCreated.length; ++i) {
             address market = marketsCreated[i];
             if (market == aStEth) continue;