forked from lidofinance/dual-governance
-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Adapt tests from model to client code
- Loading branch information
Showing
12 changed files
with
444 additions
and
204 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,9 @@ | ||
| pragma solidity 0.8.23; | ||
|
|
||
| contract WithdrawalQueueModel { | ||
| uint256 _lastFinalizedRequestId; | ||
|
|
||
| function getLastFinalizedRequestId() external view returns (uint256) { | ||
| return _lastFinalizedRequestId; | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,119 @@ | ||
| // SPDX-FileCopyrightText: 2021 Lido <[email protected]> | ||
|
|
||
| // SPDX-License-Identifier: GPL-3.0 | ||
|
|
||
| pragma solidity 0.8.23; | ||
|
|
||
| import "@openzeppelin/contracts/token/ERC20/extensions/ERC20Permit.sol"; | ||
| import "contracts/interfaces/IStETH.sol"; | ||
|
|
||
| /** | ||
| * @title StETH token wrapper with static balances. | ||
| * @dev It's an ERC20 token that represents the account's share of the total | ||
| * supply of stETH tokens. WstETH token's balance only changes on transfers, | ||
| * unlike StETH that is also changed when oracles report staking rewards and | ||
| * penalties. It's a "power user" token for DeFi protocols which don't | ||
| * support rebasable tokens. | ||
| * | ||
| * The contract is also a trustless wrapper that accepts stETH tokens and mints | ||
| * wstETH in return. Then the user unwraps, the contract burns user's wstETH | ||
| * and sends user locked stETH in return. | ||
| * | ||
| * The contract provides the staking shortcut: user can send ETH with regular | ||
| * transfer and get wstETH in return. The contract will send ETH to Lido submit | ||
| * method, staking it and wrapping the received stETH. | ||
| * | ||
| */ | ||
| contract WstETHAdapted is ERC20Permit { | ||
| IStETH public stETH; | ||
|
|
||
| /** | ||
| * @param _stETH address of the StETH token to wrap | ||
| */ | ||
| constructor(IStETH _stETH) | ||
| public | ||
| ERC20Permit("Wrapped liquid staked Ether 2.0") | ||
| ERC20("Wrapped liquid staked Ether 2.0", "wstETH") | ||
| { | ||
| stETH = _stETH; | ||
| } | ||
|
|
||
| /** | ||
| * @notice Exchanges stETH to wstETH | ||
| * @param _stETHAmount amount of stETH to wrap in exchange for wstETH | ||
| * @dev Requirements: | ||
| * - `_stETHAmount` must be non-zero | ||
| * - msg.sender must approve at least `_stETHAmount` stETH to this | ||
| * contract. | ||
| * - msg.sender must have at least `_stETHAmount` of stETH. | ||
| * User should first approve _stETHAmount to the WstETH contract | ||
| * @return Amount of wstETH user receives after wrap | ||
| */ | ||
| function wrap(uint256 _stETHAmount) external returns (uint256) { | ||
| require(_stETHAmount > 0, "wstETH: can't wrap zero stETH"); | ||
| uint256 wstETHAmount = stETH.getSharesByPooledEth(_stETHAmount); | ||
| _mint(msg.sender, wstETHAmount); | ||
| stETH.transferFrom(msg.sender, address(this), _stETHAmount); | ||
| return wstETHAmount; | ||
| } | ||
|
|
||
| /** | ||
| * @notice Exchanges wstETH to stETH | ||
| * @param _wstETHAmount amount of wstETH to uwrap in exchange for stETH | ||
| * @dev Requirements: | ||
| * - `_wstETHAmount` must be non-zero | ||
| * - msg.sender must have at least `_wstETHAmount` wstETH. | ||
| * @return Amount of stETH user receives after unwrap | ||
| */ | ||
| function unwrap(uint256 _wstETHAmount) external returns (uint256) { | ||
| require(_wstETHAmount > 0, "wstETH: zero amount unwrap not allowed"); | ||
| uint256 stETHAmount = stETH.getPooledEthByShares(_wstETHAmount); | ||
| _burn(msg.sender, _wstETHAmount); | ||
| stETH.transfer(msg.sender, stETHAmount); | ||
| return stETHAmount; | ||
| } | ||
|
|
||
| /** | ||
| * @notice Shortcut to stake ETH and auto-wrap returned stETH | ||
| */ | ||
| /* | ||
| receive() external payable { | ||
| uint256 shares = stETH.submit{value: msg.value}(address(0)); | ||
| _mint(msg.sender, shares); | ||
| } | ||
| */ | ||
|
|
||
| /** | ||
| * @notice Get amount of wstETH for a given amount of stETH | ||
| * @param _stETHAmount amount of stETH | ||
| * @return Amount of wstETH for a given stETH amount | ||
| */ | ||
| function getWstETHByStETH(uint256 _stETHAmount) external view returns (uint256) { | ||
| return stETH.getSharesByPooledEth(_stETHAmount); | ||
| } | ||
|
|
||
| /** | ||
| * @notice Get amount of stETH for a given amount of wstETH | ||
| * @param _wstETHAmount amount of wstETH | ||
| * @return Amount of stETH for a given wstETH amount | ||
| */ | ||
| function getStETHByWstETH(uint256 _wstETHAmount) external view returns (uint256) { | ||
| return stETH.getPooledEthByShares(_wstETHAmount); | ||
| } | ||
|
|
||
| /** | ||
| * @notice Get amount of stETH for a one wstETH | ||
| * @return Amount of stETH for 1 wstETH | ||
| */ | ||
| function stEthPerToken() external view returns (uint256) { | ||
| return stETH.getPooledEthByShares(1 ether); | ||
| } | ||
|
|
||
| /** | ||
| * @notice Get amount of wstETH for a one stETH | ||
| * @return Amount of wstETH for a 1 stETH | ||
| */ | ||
| function tokensPerStEth() external view returns (uint256) { | ||
| return stETH.getSharesByPooledEth(1 ether); | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.