feat: add initial gas snapshots

.prettierignore

@@ -3,3 +3,4 @@ lib
 cache
 node_modules
 report
+snapshots

Makefile

@@ -33,6 +33,9 @@ coverage :
 	make coverage-report
 	make coverage-badge
 
+# Gas reports
+forge test --mp 'tests/gas/*.t.sol' --isolate
+
 # Utilities
 download :; cast etherscan-source --chain ${chain} -d src/etherscan/${chain}_${address} ${address}
 git-diff :

snapshots/AToken.transfer.json

@@ -0,0 +1,11 @@
+{
+  "full amount; receiver: ->enableCollateral": "144837",
+  "full amount; sender: ->disableCollateral;": "103274",
+  "full amount; sender: ->disableCollateral; receiver: ->enableCollateral": "145016",
+  "full amount; sender: ->disableCollateral; receiver: dirty, ->enableCollateral": "133059",
+  "full amount; sender: collateralDisabled": "103095",
+  "partial amount; sender: collateralDisabled;": "103095",
+  "partial amount; sender: collateralDisabled; receiver: ->enableCollateral": "144837",
+  "partial amount; sender: collateralEnabled;": "103303",
+  "partial amount; sender: collateralEnabled; receiver: ->enableCollateral": "145045"
+}

snapshots/Pool.Borrow.json

@@ -0,0 +1,6 @@
+{
+  "first borrow": "256189",
+  "repay full": "176547",
+  "repay partial": "189971",
+  "second borrow": "248474"
+}

snapshots/Pool.Getters.json

@@ -0,0 +1,10 @@
+{
+  "getEModeCategoryCollateralConfig": "8000",
+  "getEModeCategoryData": "13430",
+  "getLiquidationGracePeriod": "29197",
+  "getReserveData": "34935",
+  "getUserAccountData: supplies: 0, borrows: 0": "22641",
+  "getUserAccountData: supplies: 1, borrows: 0": "34774",
+  "getUserAccountData: supplies: 2, borrows: 0": "47296",
+  "getUserAccountData: supplies: 2, borrows: 1": "28968"
+}

snapshots/Pool.Setters.json

@@ -0,0 +1,6 @@
+{
+  "setUserEMode: enter eMode, 1 borrow, 1 supply": "140164",
+  "setUserEMode: leave eMode, 1 borrow, 1 supply": "112199",
+  "setUserUseReserveAsCollateral: disableCollateral, 1 supply": "93434",
+  "setUserUseReserveAsCollateral: enableCollateral, 1 supply": "105145"
+}

snapshots/Pool.Supply.json

@@ -0,0 +1,5 @@
+{
+  "supply: collateralDisabled": "146829",
+  "supply: collateralEnabled": "146829",
+  "supply: first supply->collateralEnabled": "188344"
+}

snapshots/StataTokenV2.json

@@ -0,0 +1,6 @@
+{
+  "deposit": "284677",
+  "depositATokens": "220136",
+  "redeem": "205767",
+  "redeemAToken": "153413"
+}

tests/gas/AToken.Transfer.gas.t.sol

@@ -0,0 +1,145 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.0;
+
+import 'forge-std/Test.sol';
+
+import {IAToken} from '../../src/contracts/interfaces/IAToken.sol';
+import {Errors} from '../../src/contracts/protocol/libraries/helpers/Errors.sol';
+import {Testhelpers, IERC20} from './Testhelpers.sol';
+
+/**
+ * Scenario suite for transfer operations.
+ */
+contract ATokenTransfer_gas_Tests is Testhelpers {
+  address token;
+  IAToken aToken;
+  address variableDebtToken;
+
+  address sender = makeAddr('sender');
+  address receiver = makeAddr('receiver');
+
+  function setUp() public override {
+    super.setUp();
+    token = tokenList.usdx;
+    (address aTokenAddress, , address variableDebtTokenAddress) = contracts
+      .protocolDataProvider
+      .getReserveTokensAddresses(tokenList.usdx);
+    aToken = IAToken(aTokenAddress);
+    variableDebtToken = variableDebtTokenAddress;
+  }
+
+  function test_transfer_fullAmount() external {
+    _supplyOnReserve(sender, 1 ether);
+    vm.prank(sender);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 1 ether);
+    vm.snapshotGasLastCall(
+      'AToken.transfer',
+      'full amount; sender: ->disableCollateral; receiver: ->enableCollateral'
+    );
+  }
+
+  function test_transfer_fullAmount_dirtyReceiver() external {
+    _supplyOnReserve(receiver, 1 ether, tokenList.weth);
+    _supplyOnReserve(sender, 1 ether);
+    vm.prank(sender);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 1 ether);
+    vm.snapshotGasLastCall(
+      'AToken.transfer',
+      'full amount; sender: ->disableCollateral; receiver: dirty, ->enableCollateral'
+    );
+  }
+
+  function test_transfer_fullAmount_senderCollateralDisabled() external {
+    _supplyOnReserve(sender, 1 ether);
+    vm.startPrank(sender);
+    contracts.poolProxy.setUserUseReserveAsCollateral(token, false);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 1 ether);
+    vm.snapshotGasLastCall('AToken.transfer', 'full amount; receiver: ->enableCollateral');
+  }
+
+  function test_transfer_fullAmount_senderCollateralDisabled_receiverNonZeroFunds2() external {
+    _supplyOnReserve(sender, 1 ether);
+    _supplyOnReserve(receiver, 1 ether);
+    vm.startPrank(sender);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 1 ether);
+    vm.snapshotGasLastCall('AToken.transfer', 'full amount; sender: ->disableCollateral;');
+  }
+
+  function test_transfer_fullAmount_senderCollateralDisabled_receiverNonZeroFunds() external {
+    _supplyOnReserve(sender, 1 ether);
+    _supplyOnReserve(receiver, 1 ether);
+    vm.startPrank(sender);
+    contracts.poolProxy.setUserUseReserveAsCollateral(token, false);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 1 ether);
+    vm.snapshotGasLastCall('AToken.transfer', 'full amount; sender: collateralDisabled');
+  }
+
+  function test_transfer_partialAmount_senderCollateralEnabled() external {
+    _supplyOnReserve(sender, 1 ether);
+    vm.startPrank(sender);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 0.5 ether);
+    vm.snapshotGasLastCall(
+      'AToken.transfer',
+      'partial amount; sender: collateralEnabled; receiver: ->enableCollateral'
+    );
+  }
+
+  function test_transfer_partialAmount_senderCollateralEnabled_receiverNonZeroFunds() external {
+    _supplyOnReserve(sender, 1 ether);
+    _supplyOnReserve(receiver, 1 ether);
+    vm.startPrank(sender);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 0.5 ether);
+    vm.snapshotGasLastCall('AToken.transfer', 'partial amount; sender: collateralEnabled;');
+  }
+
+  function test_transfer_partialAmount_receiverNonZeroFunds() external {
+    _supplyOnReserve(sender, 1 ether);
+    _supplyOnReserve(receiver, 1 ether);
+    vm.startPrank(sender);
+    contracts.poolProxy.setUserUseReserveAsCollateral(token, false);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 0.5 ether);
+    vm.snapshotGasLastCall('AToken.transfer', 'partial amount; sender: collateralDisabled;');
+  }
+
+  function test_transfer_partialAmount() external {
+    _supplyOnReserve(sender, 1 ether);
+    vm.startPrank(sender);
+    contracts.poolProxy.setUserUseReserveAsCollateral(token, false);
+
+    _skip(100);
+
+    aToken.transfer(receiver, 0.5 ether);
+    vm.snapshotGasLastCall(
+      'AToken.transfer',
+      'partial amount; sender: collateralDisabled; receiver: ->enableCollateral'
+    );
+  }
+
+  function _supplyOnReserve(address user, uint256 amount) internal {
+    _supplyOnReserve(user, amount, token);
+  }
+}

tests/gas/Pool.Borrow.gas.t.sol

@@ -0,0 +1,49 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.0;
+
+import 'forge-std/Test.sol';
+
+import {Errors} from '../../src/contracts/protocol/libraries/helpers/Errors.sol';
+import {UserConfiguration} from '../../src/contracts/protocol/libraries/configuration/UserConfiguration.sol';
+import {Testhelpers, IERC20} from './Testhelpers.sol';
+
+/**
+ * Scenario suite for borrow/repay operations.
+ */
+contract PoolBorrow_gas_Tests is Testhelpers {
+  // mock users to supply and borrow liquidity
+  address borrower = makeAddr('borrower');
+
+  function setUp() public override {
+    super.setUp();
+    // setup testUser with some collateral
+    _supplyOnReserve(borrower, 100 ether, tokenList.weth);
+  }
+
+  function test_borrow() external {
+    uint256 amountToBorrow = 1000e6;
+    vm.startPrank(borrower);
+    contracts.poolProxy.borrow(tokenList.usdx, amountToBorrow, 2, 0, borrower);
+    vm.snapshotGasLastCall('Pool.Borrow', 'first borrow');
+
+    _skip(100); // skip some blocks to allow interest to accrue & the block to be cold
+    contracts.poolProxy.borrow(tokenList.usdx, amountToBorrow, 2, 0, borrower);
+    vm.snapshotGasLastCall('Pool.Borrow', 'second borrow');
+  }
+
+  function test_repay() external {
+    uint256 amountToBorrow = 1000e6;
+    deal(tokenList.usdx, borrower, amountToBorrow);
+    vm.startPrank(borrower);
+    contracts.poolProxy.borrow(tokenList.usdx, amountToBorrow, 2, 0, borrower);
+    IERC20(tokenList.usdx).approve(report.poolProxy, type(uint256).max);
+
+    _skip(100); // skip some blocks to allow interest to accrue & the block to be cold
+    contracts.poolProxy.repay(tokenList.usdx, amountToBorrow / 2, 2, borrower);
+    vm.snapshotGasLastCall('Pool.Borrow', 'repay partial');
+
+    _skip(100); // skip some blocks to allow interest to accrue & the block to be cold
+    contracts.poolProxy.repay(tokenList.usdx, type(uint256).max, 2, borrower);
+    vm.snapshotGasLastCall('Pool.Borrow', 'repay full');
+  }
+}

tests/gas/Pool.Getters.gas.t.sol

@@ -0,0 +1,67 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.0;
+
+import 'forge-std/Test.sol';
+
+import {Errors} from '../../src/contracts/protocol/libraries/helpers/Errors.sol';
+import {UserConfiguration} from '../../src/contracts/protocol/libraries/configuration/UserConfiguration.sol';
+import {Testhelpers, IERC20} from './Testhelpers.sol';
+
+/**
+ * Scenario suite for pool getter operations.
+ */
+contract PoolGetters_gas_Tests is Testhelpers {
+  // mock users to supply and borrow liquidity
+  address user = makeAddr('user');
+
+  function test_getReserveData() external {
+    contracts.poolProxy.getReserveData(tokenList.usdx);
+    vm.snapshotGasLastCall('Pool.Getters', 'getReserveData');
+  }
+
+  function test_getUserAccountData() external {
+    contracts.poolProxy.getUserAccountData(user);
+    vm.snapshotGasLastCall('Pool.Getters', 'getUserAccountData: supplies: 0, borrows: 0');
+  }
+
+  function test_getUserAccountData_oneSupplies() external {
+    _supplyOnReserve(user, 1 ether, tokenList.usdx);
+    contracts.poolProxy.getUserAccountData(user);
+    vm.snapshotGasLastCall('Pool.Getters', 'getUserAccountData: supplies: 1, borrows: 0');
+  }
+
+  function test_getUserAccountData_twoSupplies() external {
+    _supplyOnReserve(user, 1 ether, tokenList.usdx);
+    _supplyOnReserve(user, 1 ether, tokenList.weth);
+
+    contracts.poolProxy.getUserAccountData(user);
+    vm.snapshotGasLastCall('Pool.Getters', 'getUserAccountData: supplies: 2, borrows: 0');
+  }
+
+  function test_getUserAccountData_twoSupplies_oneBorrows() external {
+    _supplyOnReserve(user, 1 ether, tokenList.usdx);
+    _supplyOnReserve(user, 1 ether, tokenList.weth);
+
+    _supplyOnReserve(address(1), 0.001e8, tokenList.wbtc);
+    vm.prank(user);
+    contracts.poolProxy.borrow(tokenList.wbtc, 0.001e8, 2, 0, user);
+
+    contracts.poolProxy.getUserAccountData(user);
+    vm.snapshotGasLastCall('Pool.Getters', 'getUserAccountData: supplies: 2, borrows: 1');
+  }
+
+  function test_getEModeCategoryData() external {
+    contracts.poolProxy.getEModeCategoryData(1);
+    vm.snapshotGasLastCall('Pool.Getters', 'getEModeCategoryData');
+  }
+
+  function test_getEModeCategoryCollateralConfig() external {
+    contracts.poolProxy.getEModeCategoryCollateralConfig(1);
+    vm.snapshotGasLastCall('Pool.Getters', 'getEModeCategoryCollateralConfig');
+  }
+
+  function test_getLiquidationGracePeriod() external {
+    contracts.poolProxy.getLiquidationGracePeriod(tokenList.usdx);
+    vm.snapshotGasLastCall('Pool.Getters', 'getLiquidationGracePeriod');
+  }
+}

tests/gas/Pool.Setters.gas.t.sol

@@ -0,0 +1,56 @@
+// SPDX-License-Identifier: BUSL-1.1
+pragma solidity ^0.8.0;
+
+import 'forge-std/Test.sol';
+
+import {Errors} from '../../src/contracts/protocol/libraries/helpers/Errors.sol';
+import {UserConfiguration} from '../../src/contracts/protocol/libraries/configuration/UserConfiguration.sol';
+import {Testhelpers, IERC20} from './Testhelpers.sol';
+
+/**
+ * Scenario suite for pool setter operations.
+ */
+contract PoolSetters_gas_Tests is Testhelpers {
+  // mock users to supply and borrow liquidity
+  address user = makeAddr('user');
+
+  function test_setUserEMode() external {
+    vm.startPrank(poolAdmin);
+    EModeCategoryInput memory ct1 = _genCategoryOne();
+    contracts.poolConfiguratorProxy.setEModeCategory(ct1.id, ct1.ltv, ct1.lt, ct1.lb, ct1.label);
+    contracts.poolConfiguratorProxy.setAssetCollateralInEMode(tokenList.usdx, ct1.id, true);
+    contracts.poolConfiguratorProxy.setAssetBorrowableInEMode(tokenList.weth, ct1.id, true);
+    vm.stopPrank();
+    _supplyOnReserve(address(this), 0.5 ether, tokenList.weth);
+
+    _supplyOnReserve(user, 5000e6, tokenList.usdx);
+    vm.startPrank(user);
+    contracts.poolProxy.borrow(tokenList.weth, 0.5 ether, 2, 0, user);
+
+    _skip(100);
+
+    contracts.poolProxy.setUserEMode(1);
+    vm.snapshotGasLastCall('Pool.Setters', 'setUserEMode: enter eMode, 1 borrow, 1 supply');
+
+    _skip(100);
+
+    contracts.poolProxy.setUserEMode(0);
+    vm.snapshotGasLastCall('Pool.Setters', 'setUserEMode: leave eMode, 1 borrow, 1 supply');
+  }
+
+  function test_setUserUseReserveAsCollateral() external {
+    _supplyOnReserve(address(this), 5000e6, tokenList.usdx);
+
+    contracts.poolProxy.setUserUseReserveAsCollateral(tokenList.usdx, false);
+    vm.snapshotGasLastCall(
+      'Pool.Setters',
+      'setUserUseReserveAsCollateral: disableCollateral, 1 supply'
+    );
+
+    contracts.poolProxy.setUserUseReserveAsCollateral(tokenList.usdx, true);
+    vm.snapshotGasLastCall(
+      'Pool.Setters',
+      'setUserUseReserveAsCollateral: enableCollateral, 1 supply'
+    );
+  }
+}