feat: adding gas snapshots to integration tests

defi-wonderland · May 10, 2024 · 849095f · 849095f
1 parent 1a87b42
commit 849095f
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 1 deletion.
diff --git a/.forge-snapshots/swapExactAmountIn.snap b/.forge-snapshots/swapExactAmountIn.snap
@@ -0,0 +1 @@
+143158
diff --git a/foundry.toml b/foundry.toml
@@ -12,6 +12,8 @@ sort_imports = true
 solc_version = '0.8.23'
 libs = ["node_modules", "lib"]
 optimizer_runs = 10_000
+ffi = true
+fs_permissions = [{ access = "read-write", path = ".forge-snapshots/"}]
 
 [profile.optimized]
 via_ir = true

diff --git a/package.json b/package.json
@@ -45,6 +45,7 @@
     "@defi-wonderland/natspec-smells": "1.1.1",
     "@defi-wonderland/smock-foundry": "1.5.0",
     "forge-std": "github:foundry-rs/forge-std#5475f85",
+    "forge-gas-snapshot": "github:marktoda/forge-gas-snapshot#9161f7c",
     "husky": ">=8",
     "lint-staged": ">=10",
     "solhint": "github:solhint-community/solhint-community#v4.0.0-rc01",

diff --git a/remappings.txt b/remappings.txt
@@ -1,5 +1,6 @@
 ds-test/=node_modules/ds-test/src
 forge-std/=node_modules/forge-std/src
+forge-gas-snapshot/=node_modules/forge-gas-snapshot/src
 solmate/=node_modules/solmate/src
 
 contracts/=src/contracts
diff --git a/test/integration/PoolSwap.t.sol b/test/integration/PoolSwap.t.sol
@@ -6,7 +6,9 @@ import {BFactory} from 'contracts/BFactory.sol';
 import {BPool} from 'contracts/BPool.sol';
 import {IERC20} from 'contracts/BToken.sol';
 
-abstract contract PoolSwapIntegrationTest is Test {
+import {GasSnapshot} from "forge-gas-snapshot/GasSnapshot.sol";
+
+abstract contract PoolSwapIntegrationTest is Test, GasSnapshot {
   BFactory public factory;
   BPool public pool;
 
@@ -66,7 +68,9 @@ contract DirectPoolSwapIntegrationTest is PoolSwapIntegrationTest {
     tokenA.approve(address(pool), type(uint256).max);
 
     // swap 0.5 tokenA for tokenB
+    snapStart('swapExactAmountIn');
     pool.swapExactAmountIn(address(tokenA), 0.5e18, address(tokenB), 0, type(uint256).max);
+    snapEnd();
 
     vm.stopPrank();
   }

diff --git a/yarn.lock b/yarn.lock
@@ -812,6 +812,10 @@ for-each@^0.3.3:
   dependencies:
     is-callable "^1.1.3"
 
+"forge-gas-snapshot@github:marktoda/forge-gas-snapshot#9161f7c":
+  version "0.0.0"
+  resolved "https://codeload.github.com/marktoda/forge-gas-snapshot/tar.gz/9161f7c0b6c6788a89081e2b3b9c67592b71e689"
+
 "forge-std@github:foundry-rs/forge-std#5475f85":
   version "1.7.6"
   resolved "https://codeload.github.com/foundry-rs/forge-std/tar.gz/5475f852e3f530d7e25dfb4596aa1f9baa8ffdfc"