Gas measurements using snaps (#38)

* gas measurements using snaps * lint * more lint * Gas costs disclaimer * mooooore lint --------- Co-authored-by: alcueca <[email protected]>
alcueca · Apr 16, 2024 · 1ce8014 · 1ce8014
1 parent ec746d4
commit 1ce8014
Show file tree

Hide file tree

Showing 35 changed files with 207 additions and 29 deletions.
diff --git a/.forge-snapshots/AaveV2.snap b/.forge-snapshots/AaveV2.snap
@@ -0,0 +1 @@
+262190
diff --git a/.forge-snapshots/AaveV3.snap b/.forge-snapshots/AaveV3.snap
@@ -0,0 +1 @@
+212569
diff --git a/.forge-snapshots/AlgebraCamelot.snap b/.forge-snapshots/AlgebraCamelot.snap
@@ -0,0 +1 @@
+80679
diff --git a/.forge-snapshots/AlgebraPendleCamelot.snap b/.forge-snapshots/AlgebraPendleCamelot.snap
@@ -0,0 +1 @@
+506792
diff --git a/.forge-snapshots/Balancer.snap b/.forge-snapshots/Balancer.snap
@@ -0,0 +1 @@
+113032
diff --git a/.forge-snapshots/BalancerPendle.snap b/.forge-snapshots/BalancerPendle.snap
@@ -0,0 +1 @@
+525422
diff --git a/.forge-snapshots/CompoundMoonwell.snap b/.forge-snapshots/CompoundMoonwell.snap
@@ -0,0 +1 @@
+1183309
diff --git a/.forge-snapshots/CompoundSonne.snap b/.forge-snapshots/CompoundSonne.snap
@@ -0,0 +1 @@
+1040313
diff --git a/.forge-snapshots/Dolomite.snap b/.forge-snapshots/Dolomite.snap
@@ -0,0 +1 @@
+459815
diff --git a/.forge-snapshots/MorphoBlue.snap b/.forge-snapshots/MorphoBlue.snap
@@ -0,0 +1 @@
+132114
diff --git a/.forge-snapshots/PermissionedAaveV3.snap b/.forge-snapshots/PermissionedAaveV3.snap
@@ -0,0 +1 @@
+229742
diff --git a/.forge-snapshots/Silo.snap b/.forge-snapshots/Silo.snap
@@ -0,0 +1 @@
+1025824
diff --git a/.forge-snapshots/Solidly.snap b/.forge-snapshots/Solidly.snap
@@ -0,0 +1 @@
+163919
diff --git a/.forge-snapshots/UniswapV3.snap b/.forge-snapshots/UniswapV3.snap
@@ -0,0 +1 @@
+94720
diff --git a/.forge-snapshots/UniswapV3Pendle.snap b/.forge-snapshots/UniswapV3Pendle.snap
@@ -0,0 +1 @@
+497567
diff --git a/.gitmodules b/.gitmodules
@@ -11,3 +11,6 @@
 [submodule "lib/openzeppelin-contracts"]
 	path = lib/openzeppelin-contracts
 	url = https://github.com/openzeppelin/openzeppelin-contracts
+[submodule "lib/forge-gas-snapshot"]
+	path = lib/forge-gas-snapshot
+	url = https://github.com/marktoda/forge-gas-snapshot
diff --git a/README.md b/README.md
@@ -26,33 +26,32 @@ sequenceDiagram
 † For the BalancerWrapper and Uniswap v3 the borrower transfers the repayment to the lender and the wrapper skips the
 repayment approval.
 
-## Addresses
-
-Contracts are deployed at the same address for all supported networks.
-
-| Contract                                                        | Lender              | Address                                    | Networks                                          |
-| --------------------------------------------------------------- | ------------------- | ------------------------------------------ | ------------------------------------------------- |
-| [AaveWrapper](src/aave/AaveWrapper.sol)                         | Aave v3             | 0x9D4D2C08b29A2Db1c614483cd8971734BFDCC9F2 | Arbitrum One, Optimism, Polygon                   |
-| [AaveWrapper](src/aave/AaveWrapper.sol)                         | Aave v3             | 0x0c86c636ed5593705b5675d370c831972C787841 | Ethereum, Gnosis                                  |
-| [AaveWrapper](src/aave/AaveWrapper.sol)                         | Spark               | 0x8cB701df93f2Dae295aE8D7beE5Aa7e4D40CB397 | Ethereum, Gnosis                                  |
-| [BalancerWrapper](src/balancer/BalancerWrapper.sol)             | Balancer v2         | 0x9E092cb431e5F1aa70e47e052773711d2Ba4917E | Ethereum, Arbitrum One, Optimism, Polygon, Gnosis |
-| [BalancerWrapper](src/balancer/BalancerWrapper.sol)             | Balancer v2         | 0xD534400B0555F8441c5a3e0E9e585615B54fB2F4 | Base                                              |
-| [UniswapV3Wrapper](src/uniswapV3/UniswapV3Wrapper.sol)          | Uniswap v3          | 0x319300462C37AD2D4f26B584C2b67De51F51f289 | Arbitrum One, Optimism, Polygon                   |
-| [CompoundWrapper](src/compound/CompoundWrapper.sol)             | Balancer + Moonwell | 0x6207ec38da68902CC60D3760c9fe3EB64B426207 | Base                                              |
-| [CompoundWrapper](src/compound/CompoundWrapper.sol)             | Balancer + Sonne    | 0x6412183C579a276f467ad38468D19CC8f1F2b5cb | Optimism                                          |
-| [SiloWrapper](src/silo/SiloWrapper.sol)                         | Balancer + Silo     | 0x0F9104Fec1a5C91e63632E215e8F5c57C8f32c77 | Arbitrum One                                      |
-| [DolomiteWrapper](src/dolomite/DolomiteWrapper.sol)             | Dolomite            | 0x54F1ce5E6bdf027C9a6016C9F52fC5A445b77ed6 | Arbitrum One                                      |
-| [MorphoBlueWrapper](src/morpho/MorphoBlueWrapper.sol)           | MorphoBlue          | 0xa0Cb4e1222d813D6e4dE79f2A7A0B7759209588F | Ethereum                                          |
-| [AlgebraWrapper](src/algebra/DolomiteWrapper.sol)               | Camelot             | 0x5E8820B2832aD8451f65Fa2CCe2F3Cef29016D0d | Arbitrum One                                      |
-| [AlgebraPendleWrapper](src/pendle/AlgebraPendleWrapper.sol)     | Camelot + Pendle    | 0xC9d66F655b7B35A2B4958bE2FB58E472736Bbc47 | Arbitrum One                                      |
-| [BalancerPendleWrapper](src/pendle/BalancerPendleWrapper.sol)   | Balancer + Pendle   | 0xC1Ea6a6df39D991006b39706db7C51f5A1819da7 | Arbitrum One                                      |
-| [UniswapV3PendleWrapper](src/pendle/UniswapV3PendleWrapper.sol) | Uniswap v3 + Pendle | 0xa353Fd50210786F0E038ddD574A21d0CCefb3163 | Arbitrum One                                      |
-| [SolidlyWrapper](src/solidly/SolidlyWrapper.sol)                | Aerodrome           | 0x69b6E55f00d908018E2D745c524995bc231D762b | Base                                              |
-| [SolidlyWrapper](src/solidly/SolidlyWrapper.sol)                | Velodrome           | 0xcF13CDdbA3aEf757c52466deC310F221e06238d6 | Optimism                                          |
-
-When a contract requires constructor parameters which vary per network, these are supplied by the
-[Registry](https://github.com/alcueca/registry) deployed at 0x1BFf8Eee6ECF1c8155E81dba8894CE9cF49a220c in each supported
-network.
+## Deployments
+
+| Lender                 | Address                                    | Networks                                          | Gas     | Fees     | Contract                                                        |
+| ---------------------- | ------------------------------------------ | ------------------------------------------------- | ------- | -------- | --------------------------------------------------------------- |
+| Aave v3                | 0x9D4D2C08b29A2Db1c614483cd8971734BFDCC9F2 | Arbitrum One, Optimism, Polygon                   | 212569  | 0.05%    | [AaveWrapper](src/aave/AaveWrapper.sol)                         |
+| Aave v3 (Permissioned) | 0x0c86c636ed5593705b5675d370c831972C787841 | Ethereum, Gnosis                                  | 229742  | 0        | [AaveWrapper](src/aave/AaveWrapper.sol)                         |
+| Spark                  | 0x8cB701df93f2Dae295aE8D7beE5Aa7e4D40CB397 | Ethereum, Gnosis                                  | 212569  | 0        | [AaveWrapper](src/aave/AaveWrapper.sol)                         |
+| Balancer v2            | 0x9E092cb431e5F1aa70e47e052773711d2Ba4917E | Ethereum, Arbitrum One, Optimism, Polygon, Gnosis | 113032  | 0        | [BalancerWrapper](src/balancer/BalancerWrapper.sol)             |
+| Balancer v2            | 0xD534400B0555F8441c5a3e0E9e585615B54fB2F4 | Base                                              | 113032  | 0        | [BalancerWrapper](src/balancer/BalancerWrapper.sol)             |
+| Uniswap v3             | 0x319300462C37AD2D4f26B584C2b67De51F51f289 | Arbitrum One, Optimism, Polygon                   | 94720   | Variable | [UniswapV3Wrapper](src/uniswapV3/UniswapV3Wrapper.sol)          |
+| Balancer + Moonwell    | 0x6207ec38da68902CC60D3760c9fe3EB64B426207 | Base                                              | 1183309 | 0        | [CompoundWrapper](src/compound/CompoundWrapper.sol)             |
+| Balancer + Sonne       | 0x6412183C579a276f467ad38468D19CC8f1F2b5cb | Optimism                                          | 1040313 | 0        | [CompoundWrapper](src/compound/CompoundWrapper.sol)             |
+| Balancer + Silo        | 0x0F9104Fec1a5C91e63632E215e8F5c57C8f32c77 | Arbitrum One                                      | 1025824 | 1        | [SiloWrapper](src/silo/SiloWrapper.sol)                         |
+| Dolomite               | 0x54F1ce5E6bdf027C9a6016C9F52fC5A445b77ed6 | Arbitrum One                                      | 459815  | 0        | [DolomiteWrapper](src/dolomite/DolomiteWrapper.sol)             |
+| MorphoBlue             | 0xa0Cb4e1222d813D6e4dE79f2A7A0B7759209588F | Ethereum                                          | 132114  | 0        | [MorphoBlueWrapper](src/morpho/MorphoBlueWrapper.sol)           |
+| Camelot                | 0x5E8820B2832aD8451f65Fa2CCe2F3Cef29016D0d | Arbitrum One                                      | 80679   | 0.01%    | [AlgebraWrapper](src/algebra/AlgebraWrapper.sol)                |
+| Camelot + Pendle       | 0xC9d66F655b7B35A2B4958bE2FB58E472736Bbc47 | Arbitrum One                                      | 506792  | 0.01%    | [AlgebraPendleWrapper](src/pendle/AlgebraPendleWrapper.sol)     |
+| Balancer + Pendle      | 0xC1Ea6a6df39D991006b39706db7C51f5A1819da7 | Arbitrum One                                      | 525422  | 0        | [BalancerPendleWrapper](src/pendle/BalancerPendleWrapper.sol)   |
+| Uniswap v3 + Pendle    | 0xa353Fd50210786F0E038ddD574A21d0CCefb3163 | Arbitrum One                                      | 497567  | Variable | [UniswapV3PendleWrapper](src/pendle/UniswapV3PendleWrapper.sol) |
+| Aerodrome              | 0x69b6E55f00d908018E2D745c524995bc231D762b | Base                                              | 163919  | Variable | [SolidlyWrapper](src/solidly/SolidlyWrapper.sol)                |
+| Velodrome              | 0xcF13CDdbA3aEf757c52466deC310F221e06238d6 | Optimism                                          | 163919  | Variable | [SolidlyWrapper](src/solidly/SolidlyWrapper.sol)                |
+
+Disclaimer: The gas costs are calculated for calling `flashLoan` on each wrapper for an arbitrarily chosen token and
+loan amount. Calling the underlying flash lender directly may have larger gas savings in some instances than others.
+Requesting loans for different tokens or amounts might have different gas costs. For AMMs the fees often vary according
+to pool parameters and state.
 
 ## Flash Loans
 

diff --git a/foundry.toml b/foundry.toml
@@ -5,6 +5,9 @@
   block_timestamp = 1_680_220_800 # March 31, 2023 at 00:00 GMT
   bytecode_hash = "none"
   cbor_metadata = false
+  evm_version = 'london'
+  ffi = true
+  fs_permissions = [{ access = "read-write", path = ".forge-snapshots/"}]
   fuzz = { runs = 1_000 }
   gas_reports = ["*"]
   libs = ["lib"]
@@ -15,7 +18,6 @@
   solc = "0.8.24"
   src = "src"
   test = "test"
-  evm_version = 'london'
   verbosity = 2
 
 [profile.ci]

diff --git a/lib/forge-gas-snapshot b/lib/forge-gas-snapshot
diff --git a/test/AaveWrapper.v2.t.sol b/test/AaveWrapper.v2.t.sol
@@ -106,4 +106,12 @@ contract AaveWrapperTest is Test {
             params: ""
         });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(dai, loan);
+        IERC20(dai).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(dai, loan, "AaveV2");
+    }
 }
diff --git a/test/AaveWrapper.v3.t.sol b/test/AaveWrapper.v3.t.sol
@@ -99,4 +99,12 @@ contract AaveWrapperTest is Test {
             params: ""
         });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(dai, loan);
+        IERC20(dai).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(dai, loan, "AaveV3");
+    }
 }
diff --git a/test/AlgebraPendleWrapper.Camelot.t.sol b/test/AlgebraPendleWrapper.Camelot.t.sol
@@ -175,4 +175,12 @@ contract AlgebraPendleWrapperCamelotTest is Test {
         vm.expectRevert(AlgebraPendleWrapper.Unauthorized.selector);
         wrapper.algebraFlashCallback({ fee0: 0, fee1: 0, params: abi.encode(underlying, weth, usdc, 0, "") });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(token, loan);
+        IERC20(token).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "AlgebraPendleCamelot");
+    }
 }
diff --git a/test/AlgebraWrapper.Camelot.t.sol b/test/AlgebraWrapper.Camelot.t.sol
@@ -96,4 +96,13 @@ contract AlgebraWrapperCamelotTest is Test {
         vm.expectRevert(AlgebraWrapper.Unauthorized.selector);
         wrapper.algebraFlashCallback({ fee0: 0, fee1: 0, params: abi.encode(weth, usdc, 0, "") });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        address token = wsteth;
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(token, loan);
+        deal(address(token), address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "AlgebraCamelot");
+    }
 }
diff --git a/test/BalancerPendleWrapper.t.sol b/test/BalancerPendleWrapper.t.sol
@@ -101,4 +101,12 @@ contract BalancerPendleWrapperTest is Test {
         vm.expectRevert(BalancerPendleWrapper.HashMismatch.selector);
         wrapper.receiveFlashLoan(address(token).toArray(), uint256(1e18).toArray(), uint256(0).toArray(), "");
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 10e18;
+        uint256 fee = wrapper.flashFee(token, loan);
+        IERC20(token).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "BalancerPendle");
+    }
 }
diff --git a/test/BalancerWrapper.t.sol b/test/BalancerWrapper.t.sol
@@ -92,4 +92,12 @@ contract BalancerWrapperTest is Test {
             params: ""
         });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(dai, loan);
+        IERC20(dai).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(dai, loan, "Balancer");
+    }
 }
diff --git a/test/CompoundWrapper.Moonwell.t.sol b/test/CompoundWrapper.Moonwell.t.sol
@@ -105,4 +105,12 @@ contract MoonwellWrapperTest is Test {
 
         wrapper.setCToken(IERC20(dai), ICToken(address(0x628ff693426583D9a7FB391E54366292F509D457)));
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(dai, loan);
+        IERC20(dai).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(dai, loan, "CompoundMoonwell");
+    }
 }
diff --git a/test/CompoundWrapper.Sonne.t.sol b/test/CompoundWrapper.Sonne.t.sol
@@ -103,4 +103,12 @@ contract SonneWrapperTest is Test {
 
         wrapper.setCToken(IERC20(snx), ICToken(address(0xD7dAabd899D1fAbbC3A9ac162568939CEc0393Cc)));
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(snx, loan);
+        IERC20(snx).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(snx, loan, "CompoundSonne");
+    }
 }
diff --git a/test/DolomiteWrapper.t.sol b/test/DolomiteWrapper.t.sol
@@ -86,4 +86,12 @@ contract DolomiteWrapperTest is Test {
         vm.expectRevert(DolomiteWrapper.HashMismatch.selector);
         wrapper.callFunction(address(wrapper), accountInfo, "");
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 10e8;
+        uint256 fee = wrapper.flashFee(wbtc, loan);
+        IERC20(wbtc).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(wbtc, loan, "Dolomite");
+    }
 }
diff --git a/test/MockBorrower.sol b/test/MockBorrower.sol
@@ -3,6 +3,7 @@ pragma solidity ^0.8.19;
 
 import "erc7399/IERC7399.sol";
 import "src/BaseWrapper.sol";
+import { GasSnapshot } from "forge-gas-snapshot/GasSnapshot.sol";
 
 contract LoanReceiver {
     using SafeERC20 for IERC20;
@@ -13,7 +14,7 @@ contract LoanReceiver {
 }
 
 /// @dev Mock flash loan borrower. It allows to examine the state of the borrower during the callback.
-contract MockBorrower {
+contract MockBorrower is GasSnapshot {
     using SafeERC20 for IERC20;
 
     bytes32 public constant ERC3156PP_CALLBACK_SUCCESS = keccak256("ERC3156PP_CALLBACK_SUCCESS");
@@ -137,6 +138,22 @@ contract MockBorrower {
         return "";
     }
 
+    function onFlashLoanMeasureGas(
+        address,
+        address paymentReceiver,
+        address asset,
+        uint256 amount,
+        uint256 fee,
+        bytes calldata
+    )
+        external
+        returns (bytes memory)
+    {
+        IERC20(asset).safeTransfer(paymentReceiver, amount + fee);
+
+        return "";
+    }
+
     function flashBorrow(address asset, uint256 amount) public returns (bytes memory) {
         return lender.flash(address(loanReceiver), asset, amount, "", this.onFlashLoan);
     }
@@ -158,4 +175,17 @@ contract MockBorrower {
     function flashBorrowVoid(address asset, uint256 amount) public returns (bytes memory) {
         return lender.flash(address(loanReceiver), asset, amount, "", this.onFlashLoanVoid);
     }
+
+    function flashBorrowMeasureGas(
+        address asset,
+        uint256 amount,
+        string calldata tag
+    )
+        public
+        returns (bytes memory out)
+    {
+        snapStart(tag);
+        out = lender.flash(address(this), asset, amount, "", this.onFlashLoanMeasureGas);
+        snapEnd();
+    }
 }
diff --git a/test/MorphoBlueWrapper.t.sol b/test/MorphoBlueWrapper.t.sol
@@ -91,4 +91,12 @@ contract MorphoBlueWrapperTest is Test {
         vm.expectRevert(MorphoBlueWrapper.NotMorpho.selector);
         wrapper.onMorphoFlashLoan({ amount: 0, params: "" });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 100e18;
+        uint256 fee = wrapper.flashFee(wstETH, loan);
+        IERC20(wstETH).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(wstETH, loan, "MorphoBlue");
+    }
 }
diff --git a/test/PermissionedAaveWrapper.v3.t.sol b/test/PermissionedAaveWrapper.v3.t.sol
@@ -120,6 +120,19 @@ contract PermissionedAaveWrapperTest is Test {
         );
         borrower.flashBorrow(dai, 1e18);
     }
+
+    function test_measureFlashLoanGasPermissionedAaveV3() public {
+        console2.log("test_measureFlashLoanGas");
+        vm.mockCall(
+            provider.getACLManager(),
+            abi.encodeWithSelector(IACLManager.isFlashBorrower.selector, wrapper),
+            abi.encode(true)
+        );
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(dai, loan);
+        IERC20(dai).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(dai, loan, "PermissionedAaveV3");
+    }
 }
 
 interface IACLManager {

diff --git a/test/SiloWrapper.t.sol b/test/SiloWrapper.t.sol
@@ -115,4 +115,12 @@ contract SiloWrapperTest is Test {
         vm.expectRevert(SiloWrapper.HashMismatch.selector);
         wrapper.receiveFlashLoan(address(token).toArray(), uint256(1e18).toArray(), uint256(0).toArray(), "");
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 10_999_999_999_999_999;
+        uint256 fee = wrapper.flashFee(token, loan);
+        IERC20(token).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "Silo");
+    }
 }
diff --git a/test/SolidlyWrapper.t.sol b/test/SolidlyWrapper.t.sol
@@ -102,4 +102,13 @@ contract SolidlyWrapperTest is Test {
         vm.expectRevert(SolidlyWrapper.UnknownPool.selector);
         wrapper.hook({ sender: address(wrapper), amount0: 0, amount1: 0, params: abi.encode(weth, usdc, 0, false, "") });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        address token = weth;
+        uint256 fee = wrapper.flashFee(token, loan);
+        deal(address(token), address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "Solidly");
+    }
 }
diff --git a/test/UniswapV3PendleWrapper.t.sol b/test/UniswapV3PendleWrapper.t.sol
@@ -178,4 +178,12 @@ contract UniswapV3PendleWrapperTest is Test {
             params: abi.encode(address(token), address(underlying), uint24(0.0005e6), uint256(0), "")
         });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(token, loan);
+        IERC20(token).safeTransfer(address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "UniswapV3Pendle");
+    }
 }
diff --git a/test/UniswapV3Wrapper.t.sol b/test/UniswapV3Wrapper.t.sol
@@ -108,4 +108,13 @@ contract UniswapV3WrapperTest is Test {
             params: abi.encode(address(usdc), address(usdt), uint24(0.0005e6), uint256(0), "")
         });
     }
+
+    function test_measureFlashLoanGas() public {
+        console2.log("test_measureFlashLoanGas");
+        address token = weth;
+        uint256 loan = 1e18;
+        uint256 fee = wrapper.flashFee(token, loan);
+        deal(address(token), address(borrower), fee);
+        borrower.flashBorrowMeasureGas(token, loan, "UniswapV3");
+    }
 }