Merge pull request #119 from GenerationSoftware/gen-1855-test-reth-in…

…tegration-with-pt

test wreth integration

.gitmodules

@@ -40,3 +40,6 @@
 [submodule "lib/ExcessivelySafeCall"]
 	path = lib/ExcessivelySafeCall
 	url = https://github.com/nomad-xyz/ExcessivelySafeCall
+[submodule "lib/rETHERC4626"]
+	path = lib/rETHERC4626
+	url = https://github.com/rocket-pool/rETHERC4626

remappings.txt

@@ -7,6 +7,7 @@ openzeppelin/=lib/openzeppelin-contracts/contracts/
 owner-manager-contracts/=lib/owner-manager-contracts/contracts/
 yield-daddy/=lib/yield-daddy/src/
 excessively-safe-call/=lib/ExcessivelySafeCall/src/
+rETHERC4626/=lib/rETHERC4626/src/
 
 pt-v5-liquidator-interfaces/=lib/pt-v5-liquidator-interfaces/src/interfaces/
 pt-v5-twab-controller=lib/pt-v5-twab-controller/src/

test/integration/rocket-pool/wrETHOptimism.t.sol

@@ -0,0 +1,91 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.24;
+
+import { BaseIntegration, IERC20, IERC4626 } from "../BaseIntegration.t.sol";
+import { WRETH, IERC20 as RETH_IERC20 } from "rETHERC4626/WRETH.sol";
+import { MockOracle } from "rETHERC4626/mock/MockOracle.sol";
+
+import { ERC4626, ERC20, IERC20 as TokenVaultIERC20 } from "../../../lib/rETHERC4626/lib/openzeppelin-contracts/contracts/token/ERC20/extensions/ERC4626.sol";
+contract TokenVault is ERC4626 {
+    constructor(
+        TokenVaultIERC20 asset_,
+        string memory name_,
+        string memory symbol_
+    ) ERC4626(asset_) ERC20(name_, symbol_) { }
+}
+
+contract wrETHOptimismIntegrationTest is BaseIntegration {
+    uint256 fork;
+    uint256 forkBlock = 122853185;
+    uint256 forkBlockTimestamp = 1721305147;
+
+    address internal _reth = address(0x9Bcef72be871e61ED4fBbc7630889beE758eb81D);
+    address internal _rethWhale = address(0xBA12222222228d8Ba445958a75a0704d566BF2C8);
+    address internal _wrethWhale;
+
+    MockOracle internal _oracle;
+    WRETH internal _wreth;
+    TokenVault internal _yieldVault;
+
+    /* ============ setup ============ */
+
+    function setUpUnderlyingAsset() public virtual override returns (IERC20 asset, uint8 decimals, uint256 approxAssetUsdExchangeRate) {
+        _oracle = new MockOracle(1 ether); // setup oracle for rETH with initial price at 1 ETH
+        _wreth = new WRETH(RETH_IERC20(_reth), _oracle);
+        return (IERC20(address(_wreth)), 18, 3.4e18);
+    }
+
+    function setUpYieldVault() public virtual override returns (IERC4626) {
+        _yieldVault = new TokenVault(_wreth, "wRETH Vault", "vwRETH");
+
+        _wrethWhale = makeAddr("wrethWhale");
+        vm.startPrank(_rethWhale);
+        IERC20(_reth).approve(address(_wreth), type(uint256).max);
+        _wreth.mint(IERC20(_reth).balanceOf(_rethWhale));
+        _wreth.transfer(_wrethWhale, _wreth.balanceOf(_rethWhale));
+        require(_wreth.balanceOf(_wrethWhale) > 0, "no wreth balance");
+        vm.stopPrank();
+
+        return IERC4626(address(_yieldVault));
+    }
+
+    function setUpFork() public virtual override {
+        fork = vm.createFork(vm.rpcUrl("optimism"), forkBlock);
+        vm.selectFork(fork);
+        vm.warp(forkBlockTimestamp);
+    }
+
+    function beforeSetup() public virtual override {
+        lowGasPriceEstimate = 0.05 gwei; // just L2 gas, we ignore L1 costs for a super low estimate
+        roundingErrorOnTransfer = 1; // 1 wei rounding error on transfer is common with wrETH
+    }
+
+    function afterSetup() public virtual override { }
+
+    /* ============ helpers to override ============ */
+
+    /// @dev The max amount of assets than can be dealt.
+    function maxDeal() public virtual override returns (uint256) {
+        return _wreth.balanceOf(_wrethWhale);
+    }
+
+    /// @dev May revert if the amount requested exceeds the amount available to deal.
+    function dealAssets(address to, uint256 amount) public virtual override prankception(_wrethWhale) {
+        _wreth.transfer(to, amount);
+    }
+
+    /// @dev Accrues yield by increasing the oracle price
+    function _accrueYield() internal virtual override {
+        uint256 _rate = _oracle.rate();
+        _oracle.setRate((_rate * (1e18 + (uint256(0.03e18) / 365))) / 1e18); // approx daily rate increase for 3% APR 
+        _wreth.rebase();
+    }
+
+    /// @dev Simulates loss by decreasing the oracle price
+    function _simulateLoss() internal virtual override {
+        uint256 _rate = _oracle.rate();
+        _oracle.setRate(_rate / 2);
+        _wreth.rebase();
+    }
+
+}