Gas Snapshot Testing

# Generate gas snapshot
forge snapshot

# Output stored in .gas-snapshot:
# MyContractTest:test_deposit() (gas: 45231)
# MyContractTest:test_withdraw() (gas: 38912)
# MyContractTest:test_transfer() (gas: 21453)

# Compare current gas against committed snapshot
# Fails if any test uses more gas than recorded
forge snapshot --check

# Allow up to 5% increase
forge snapshot --tolerance 5

# Detailed per-function gas breakdown
forge test --gas-report

# Output:
# | src/Vault.sol:Vault |                 |       |        |       |         |
# |---------------------|-----------------|-------|--------|-------|---------|
# | Function Name       | min             | avg   | median | max   | # calls |
# | deposit             | 45231           | 47892 | 45231  | 53214 | 12      |
# | withdraw            | 38912           | 40123 | 38912  | 42345 | 8       |

# Filter to specific contracts
forge test --gas-report --match-contract Vault

name: Gas Check
on: [pull_request]

jobs:
  gas:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - uses: foundry-rs/foundry-toolchain@v1

      - name: Install dependencies
        run: forge install

      - name: Check gas snapshots
        run: forge snapshot --check --tolerance 3

      - name: Gas report
        run: forge test --gas-report | tee gas-report.txt

      - name: Comment gas report on PR
        uses: actions/github-script@v7
        if: github.event_name == 'pull_request'
        with:
          script: |
            const fs = require('fs');
            const report = fs.readFileSync('gas-report.txt', 'utf8');
            github.rest.issues.createComment({
              issue_number: context.issue.number,
              owner: context.repo.owner,
              repo: context.repo.repo,
              body: `## Gas Report\n\`\`\`\n${report}\n\`\`\``
            });

# 1. Create baseline snapshot
forge snapshot --snap .gas-snapshot-baseline

# 2. Make optimizations
# ... edit contracts ...

# 3. Compare against baseline
forge snapshot --diff .gas-snapshot-baseline

# Output shows differences:
# test_deposit() (gas: 45231 -> 42100) -3131 (-6.92%)
# test_withdraw() (gas: 38912 -> 38900) -12 (-0.03%)

contract GasTest is Test {
    MyContract target;

    function setUp() public {
        target = new MyContract();
        // Warm storage slots for realistic measurements
        target.initialize();
    }

    // Naming convention: gas_ prefix for benchmark tests
    function test_gas_deposit_firstTime() public {
        // Cold storage write — most expensive
        target.deposit(1 ether);
    }

    function test_gas_deposit_subsequent() public {
        target.deposit(1 ether); // warm up
        target.deposit(1 ether); // measure this
    }

    function test_gas_batchDeposit_10() public {
        for (uint256 i = 0; i < 10; i++) {
            target.deposit(1 ether);
        }
    }

    // Measure specific operations with gasleft()
    function test_gas_specificOperation() public {
        // Setup (not measured)
        target.deposit(1 ether);

        uint256 gasBefore = gasleft();
        target.withdraw(0.5 ether);
        uint256 gasUsed = gasBefore - gasleft();

        // Log for manual review
        emit log_named_uint("withdraw gas", gasUsed);

        // Assert gas bound
        assertLt(gasUsed, 50_000, "withdraw too expensive");
    }
}

// Before: 3 SLOADs
function getInfo() external view returns (uint256, uint256, uint256) {
    return (value1, value2, value3);
}

// After: cache in memory
function getInfo() external view returns (uint256 a, uint256 b, uint256 c) {
    a = value1;
    b = value2;
    c = value3;
}

// Unchecked math where overflow is impossible
function distribute(uint256 total, uint256 share) internal pure returns (uint256) {
    unchecked {
        return total / share; // share is validated > 0 by caller
    }
}