Tests: Added testcases for AaveVault.sol + mock contracts

Author: craterface77

src/mocks/MockAavePool.sol

@@ -0,0 +1,86 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.10;
+
+import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+import {MockERC20} from "./MockERC20.sol";
+import {SafeERC20} from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
+
+/**
+ * @title MockAavePool
+ * @notice A simplified mock for Aave Pool to test interactions with AaveVault.
+ */
+contract MockAavePool {
+    using SafeERC20 for IERC20;
+
+    struct Reserve {
+        uint256 totalLiquidity; // Total liquidity in the pool
+        address aTokenAddress; // Address of the associated aToken
+    }
+
+    mapping(address => Reserve) public reserves;
+
+    event Supply(address indexed user, address indexed asset, uint256 amount);
+    event Withdraw(address indexed user, address indexed asset, uint256 amount);
+    event YieldGenerated(address indexed user, address indexed asset, uint256 amount);
+
+    /**
+     * @notice Adds support for a new token and links it with an aToken.
+     * @param underlyingToken Address of the underlying token.
+     * @param aToken Address of the associated aToken.
+     */
+    function addReserve(address underlyingToken, address aToken) external {
+        require(reserves[underlyingToken].aTokenAddress == address(0), "MockAavePool: Reserve already exists");
+        reserves[underlyingToken] = Reserve({totalLiquidity: 0, aTokenAddress: aToken});
+    }
+
+    /**
+     * @notice Supplies liquidity to the pool and mints aTokens.
+     * @param asset Address of the token to deposit.
+     * @param amount Amount of tokens to deposit.
+     * @param onBehalfOf Address to mint aTokens for.
+     * @dev referralCode Referral code (ignored in mock implementation).
+     */
+    function supply(address asset, uint256 amount, address onBehalfOf, uint16) external {
+        Reserve storage reserve = reserves[asset];
+        require(reserve.aTokenAddress != address(0), "MockAavePool: Unsupported asset");
+
+        IERC20(asset).safeTransferFrom(msg.sender, address(this), amount);
+        IERC20(reserve.aTokenAddress).safeTransfer(onBehalfOf, amount); // Mint aTokens 1:1
+
+        reserve.totalLiquidity += amount;
+
+        emit Supply(onBehalfOf, asset, amount);
+    }
+
+    /**
+     * @notice Withdraws liquidity from the pool and burns aTokens.
+     * @param asset Address of the token to withdraw.
+     * @param amount Amount of tokens to withdraw.
+     * @param to Address to send the withdrawn tokens to.
+     */
+    function withdraw(address asset, uint256 amount, address to) external {
+        Reserve storage reserve = reserves[asset];
+        require(reserve.aTokenAddress != address(0), "MockAavePool: Unsupported asset");
+
+        IERC20(asset).safeTransfer(to, amount);
+
+        reserve.totalLiquidity -= amount;
+
+        emit Withdraw(msg.sender, asset, amount);
+    }
+
+    /**
+     * @notice Generates yield by minting additional aTokens for a user.
+     * @param asset Address of the underlying token.
+     * @param amount Amount of yield to generate.
+     * @param user Address of the user to mint aTokens for.
+     */
+    function generateYield(address asset, uint256 amount, address user) external {
+        Reserve storage reserve = reserves[asset];
+        require(reserve.aTokenAddress != address(0), "MockAavePool: Unsupported asset");
+
+        IERC20(reserve.aTokenAddress).safeTransfer(user, amount); // Mint yield as aTokens
+
+        emit YieldGenerated(user, asset, amount);
+    }
+}

src/mocks/MockERC20.sol

@@ -9,4 +9,8 @@ contract MockERC20 is ERC20 {
     function mint(address sender, uint256 amount) external {
         _mint(sender, amount);
     }
+
+    function burn(address sender, uint256 amount) external {
+        _burn(sender, amount);
+    }
 }

src/mocks/MockL2AavePool.sol

@@ -0,0 +1,46 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.10;
+
+import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol";
+import {MockERC20} from "./MockERC20.sol";
+import {SafeERC20} from "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol";
+import {CalldataLogic} from "aave-v3-origin/core/contracts/protocol/libraries/logic/CalldataLogic.sol";
+
+/**
+ * @title MockL2AavePool
+ * @notice A simplified mock for Aave L2 Pool to test interactions with AaveVault on L2.
+ */
+contract MockL2AavePool {
+    using SafeERC20 for IERC20;
+
+    struct Reserve {
+        uint256 totalLiquidity;
+        address aTokenAddress;
+        address asset;
+    }
+
+    Reserve public reserve;
+
+    function addReserve(address underlyingToken, address aToken) external {
+        reserve = Reserve({totalLiquidity: 0, aTokenAddress: aToken, asset: underlyingToken});
+    }
+
+    function supply(bytes32 args) external {
+        // Simulate supply logic. Simulate amount just for testing purposes
+        IERC20(reserve.asset).safeTransferFrom(msg.sender, address(this), 100 ether); // Transfer underlying to pool
+        IERC20(reserve.aTokenAddress).safeTransfer(msg.sender, 100 ether); // Mint aTokens to user
+
+        reserve.totalLiquidity += 100 ether;
+    }
+
+    function withdraw(bytes32 args) external {
+        // Simulate withdraw logic. Simulate amount just for testing purposes
+        IERC20(reserve.asset).safeTransfer(msg.sender, 100 ether); // Transfer underlying to user
+
+        reserve.totalLiquidity -= 100 ether;
+    }
+
+    function getTotalLiquidity() external view returns (uint256) {
+        return reserve.totalLiquidity;
+    }
+}

src/mocks/MockL2Encoder.sol

@@ -0,0 +1,41 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.0;
+
+import {IPool} from "aave-v3-origin/core/contracts/interfaces/IPool.sol";
+
+/**
+ * @title MockL2Encoder
+ * @dev A mock implementation of the L2Encoder for testing purposes.
+ */
+contract MockL2Encoder {
+    IPool public pool;
+
+    /**
+     * @notice Constructor to initialize the mock with a reference to the Aave Pool.
+     * @param _pool Address of the Aave Pool.
+     */
+    constructor(address _pool) {
+        pool = IPool(_pool);
+    }
+
+    /**
+     * @notice Encodes the parameters for a supply operation.
+     * @param asset The address of the asset being supplied.
+     * @param amount The amount of the asset being supplied.
+     * @param referralCode The referral code for the supply operation.
+     * @return Encoded parameters as bytes32.
+     */
+    function encodeSupplyParams(address asset, uint256 amount, uint16 referralCode) external pure returns (bytes32) {
+        return keccak256(abi.encodePacked(asset, amount, referralCode));
+    }
+
+    /**
+     * @notice Encodes the parameters for a withdrawal operation.
+     * @param asset The address of the asset being withdrawn.
+     * @param amount The amount of the asset being withdrawn.
+     * @return Encoded parameters as bytes32.
+     */
+    function encodeWithdrawParams(address asset, uint256 amount) external pure returns (bytes32) {
+        return keccak256(abi.encodePacked(asset, amount));
+    }
+}