test: cleanup snapshot suite

bgd-labs · Jan 10, 2025 · 55a5c5a · 55a5c5a
1 parent 2967f6a
commit 55a5c5a
Showing 1 changed file with 18 additions and 62 deletions.
diff --git a/tests/utils/ProtocolV3TestBase.sol b/tests/utils/ProtocolV3TestBase.sol
@@ -61,7 +61,7 @@ struct ReserveConfig {
 }
 
 struct LocalVars {
-  IPoolDataProvider.TokenData[] reserves;
+  address[] reserves;
   ReserveConfig[] configs;
 }
 
@@ -93,10 +93,10 @@ contract ProtocolV3TestBase is DiffUtils {
     bool poolConfigs
   ) public virtual returns (ReserveConfig[] memory) {
     string memory path = string(abi.encodePacked('./reports/', reportName, '.json'));
-    // overwrite with empty json to later be extended
+    // overwrite with empty json to later be extended as foundry does not currently support adding new keys
     vm.writeFile(
       path,
-      '{ "eModes": {}, "reserves": {}, "strategies": {}, "poolConfiguration": {} }'
+      '{ "eModes": {}, "reserves": {}, "strategies": {}, "poolConfiguration": {}, "raw": {} }'
     );
     vm.serializeUint('root', 'chainId', block.chainid);
     ReserveConfig[] memory configs = _getReservesConfigs(pool);
@@ -236,21 +236,8 @@ contract ProtocolV3TestBase is DiffUtils {
       vm.serializeAddress(key, 'underlying', config.underlying);
       vm.serializeAddress(key, 'aToken', config.aToken);
       vm.serializeAddress(key, 'variableDebtToken', config.variableDebtToken);
-      vm.serializeAddress(
-        key,
-        'aTokenImpl',
-        ProxyHelpers.getInitializableAdminUpgradeabilityProxyImplementation(vm, config.aToken)
-      );
       vm.serializeString(key, 'aTokenSymbol', IERC20Detailed(config.aToken).symbol());
       vm.serializeString(key, 'aTokenName', IERC20Detailed(config.aToken).name());
-      vm.serializeAddress(
-        key,
-        'variableDebtTokenImpl',
-        ProxyHelpers.getInitializableAdminUpgradeabilityProxyImplementation(
-          vm,
-          config.variableDebtToken
-        )
-      );
       vm.serializeString(
         key,
         'variableDebtTokenSymbol',
@@ -317,76 +304,43 @@ contract ProtocolV3TestBase is DiffUtils {
     // pool configurator
     IPoolConfigurator configurator = IPoolConfigurator(addressesProvider.getPoolConfigurator());
     vm.serializeAddress(poolConfigKey, 'poolConfigurator', address(configurator));
-    vm.serializeAddress(
-      poolConfigKey,
-      'poolConfiguratorImpl',
-      ProxyHelpers.getInitializableAdminUpgradeabilityProxyImplementation(vm, address(configurator))
-    );
 
     // PoolDataProvider
     IPoolDataProvider pdp = IPoolDataProvider(addressesProvider.getPoolDataProvider());
     vm.serializeAddress(poolConfigKey, 'protocolDataProvider', address(pdp));
 
     // pool
-    vm.serializeAddress(
-      poolConfigKey,
-      'poolImpl',
-      ProxyHelpers.getInitializableAdminUpgradeabilityProxyImplementation(vm, address(pool))
-    );
     string memory content = vm.serializeAddress(poolConfigKey, 'pool', address(pool));
 
     string memory output = vm.serializeString('root', 'poolConfig', content);
     vm.writeJson(output, path);
   }
 
   function _getReservesConfigs(IPool pool) internal view virtual returns (ReserveConfig[] memory) {
-    IPoolAddressesProvider addressesProvider = IPoolAddressesProvider(pool.ADDRESSES_PROVIDER());
-    IPoolDataProvider poolDataProvider = IPoolDataProvider(addressesProvider.getPoolDataProvider());
     LocalVars memory vars;
 
-    vars.reserves = poolDataProvider.getAllReservesTokens();
+    vars.reserves = pool.getReservesList();
 
     vars.configs = new ReserveConfig[](vars.reserves.length);
 
     for (uint256 i = 0; i < vars.reserves.length; i++) {
-      vars.configs[i] = _getStructReserveConfig(pool, poolDataProvider, vars.reserves[i]);
+      vars.configs[i] = _getStructReserveConfig(pool, vars.reserves[i]);
     }
 
     return vars.configs;
   }
 
-  function _getStructReserveTokens(
-    IPoolDataProvider pdp,
-    address underlyingAddress
-  ) internal view virtual returns (ReserveTokens memory) {
-    ReserveTokens memory reserveTokens;
-    (reserveTokens.aToken, , reserveTokens.variableDebtToken) = pdp.getReserveTokensAddresses(
-      underlyingAddress
-    );
-
-    return reserveTokens;
-  }
-
   function _getStructReserveConfig(
     IPool pool,
-    IPoolDataProvider poolDataProvider,
-    IPoolDataProvider.TokenData memory reserve
+    address reserve
   ) internal view virtual returns (ReserveConfig memory) {
     ReserveConfig memory localConfig;
-    DataTypes.ReserveConfigurationMap memory configuration = pool.getConfiguration(
-      reserve.tokenAddress
-    );
+    DataTypes.ReserveConfigurationMap memory configuration = pool.getConfiguration(reserve);
 
-    localConfig.underlying = reserve.tokenAddress;
-    ReserveTokens memory reserveTokens = _getStructReserveTokens(
-      poolDataProvider,
-      reserve.tokenAddress
-    );
-    localConfig.aToken = reserveTokens.aToken;
-    localConfig.variableDebtToken = reserveTokens.variableDebtToken;
-    localConfig.interestRateStrategy = pool
-      .getReserveData(reserve.tokenAddress)
-      .interestRateStrategyAddress;
+    localConfig.underlying = reserve;
+    localConfig.aToken = pool.getReserveAToken(reserve);
+    localConfig.variableDebtToken = pool.getReserveVariableDebtToken(reserve);
+    localConfig.interestRateStrategy = pool.getReserveData(reserve).interestRateStrategyAddress;
     (
       localConfig.ltv,
       localConfig.liquidationThreshold,
@@ -400,7 +354,11 @@ contract ProtocolV3TestBase is DiffUtils {
       localConfig.borrowingEnabled,
       localConfig.isPaused
     ) = configuration.getFlags();
-    localConfig.symbol = reserve.symbol;
+    if (reserve == 0x9f8F72aA9304c8B593d555F12eF6589cC3A579A2) {
+      localConfig.symbol = 'MKR';
+    } else {
+      localConfig.symbol = IERC20Detailed(reserve).symbol();
+    }
     localConfig.usageAsCollateralEnabled = localConfig.liquidationThreshold != 0;
     localConfig.isSiloed = configuration.getSiloedBorrowing();
     (localConfig.borrowCap, localConfig.supplyCap) = configuration.getCaps();
@@ -414,11 +372,9 @@ contract ProtocolV3TestBase is DiffUtils {
     localConfig.virtualAccActive = configuration.getIsVirtualAccActive();
 
     if (localConfig.virtualAccActive) {
-      localConfig.virtualBalance = pool.getVirtualUnderlyingBalance(reserve.tokenAddress);
+      localConfig.virtualBalance = pool.getVirtualUnderlyingBalance(reserve);
     }
-    localConfig.aTokenUnderlyingBalance = IERC20Detailed(reserve.tokenAddress).balanceOf(
-      localConfig.aToken
-    );
+    localConfig.aTokenUnderlyingBalance = IERC20Detailed(reserve).balanceOf(localConfig.aToken);
 
     return localConfig;
   }