Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
c93e196
commit e9251e7
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
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,183 @@ | ||
| // SPDX-License-Identifier: GPL-3.0-or-later | ||
| pragma solidity 0.8.16; | ||
|
|
||
| import '../../contracts/library/UtilLib.sol'; | ||
|
|
||
| import '../../contracts/StaderConfig.sol'; | ||
| import '../../contracts/SDUtilityPool.sol'; | ||
| import '../../contracts/SDIncentiveController.sol'; | ||
|
|
||
| import '../mocks/SDCollateralMock.sol'; | ||
| import '../mocks/StaderTokenMock.sol'; | ||
| import '../mocks/SDIncentiveControllerMock.sol'; | ||
| import '../mocks/OperatorRewardsCollectorMock.sol'; | ||
| import '../mocks/PoolUtilsMock.sol'; | ||
|
|
||
| import 'forge-std/Test.sol'; | ||
| import '@openzeppelin/contracts/proxy/transparent/TransparentUpgradeableProxy.sol'; | ||
| import '@openzeppelin/contracts/proxy/transparent/ProxyAdmin.sol'; | ||
|
|
||
| contract SDIncentiveControllerTest is Test { | ||
| address staderAdmin; | ||
| address staderManager; | ||
| address staderTreasury; | ||
|
|
||
| StaderConfig staderConfig; | ||
| SDUtilityPool sdUtilityPool; | ||
| SDIncentiveController sdIncentiveController; | ||
| StaderTokenMock staderToken; | ||
| SDCollateralMock sdCollateral; | ||
| OperatorRewardsCollectorMock operatorRewardsCollector; | ||
| PoolUtilsMock poolUtils; | ||
|
|
||
| function setUp() public { | ||
| staderAdmin = vm.addr(100); | ||
| staderManager = vm.addr(101); | ||
| address ethDepositAddr = vm.addr(102); | ||
| staderTreasury = vm.addr(105); | ||
|
|
||
| staderToken = new StaderTokenMock(); | ||
| ProxyAdmin admin = new ProxyAdmin(); | ||
|
|
||
| StaderConfig configImpl = new StaderConfig(); | ||
| TransparentUpgradeableProxy configProxy = new TransparentUpgradeableProxy( | ||
| address(configImpl), | ||
| address(admin), | ||
| '' | ||
| ); | ||
| staderConfig = StaderConfig(address(configProxy)); | ||
| staderConfig.initialize(staderAdmin, ethDepositAddr); | ||
|
|
||
| sdCollateral = new SDCollateralMock(); | ||
| operatorRewardsCollector = new OperatorRewardsCollectorMock(); | ||
| poolUtils = new PoolUtilsMock(address(staderConfig)); | ||
|
|
||
| vm.startPrank(staderAdmin); | ||
| staderConfig.updateStaderToken(address(staderToken)); | ||
| staderConfig.updateSDCollateral(address(sdCollateral)); | ||
| staderConfig.updateOperatorRewardsCollector(address(operatorRewardsCollector)); | ||
| staderConfig.updatePoolUtils(address(poolUtils)); | ||
| staderConfig.grantRole(staderConfig.MANAGER(), staderManager); | ||
| vm.stopPrank(); | ||
|
|
||
| vm.prank(staderManager); | ||
| staderConfig.updateStaderTreasury(staderTreasury); | ||
|
|
||
| SDUtilityPool sdUtilityPoolImpl = new SDUtilityPool(); | ||
| TransparentUpgradeableProxy proxy = new TransparentUpgradeableProxy( | ||
| address(sdUtilityPoolImpl), | ||
| address(admin), | ||
| '' | ||
| ); | ||
| sdUtilityPool = SDUtilityPool(address(proxy)); | ||
| sdUtilityPool.initialize(staderAdmin, address(staderConfig)); | ||
|
|
||
| vm.prank(staderAdmin); | ||
| sdUtilityPool.updateRiskConfig(70, 30, 5, 50); | ||
|
|
||
| SDIncentiveController sdIncentiveControllerImpl = new SDIncentiveController(); | ||
| TransparentUpgradeableProxy sdIncentiveControllerProxy = new TransparentUpgradeableProxy( | ||
| address(sdIncentiveControllerImpl), | ||
| address(admin), | ||
| '' | ||
| ); | ||
| sdIncentiveController = SDIncentiveController(address(sdIncentiveControllerProxy)); | ||
| sdIncentiveController.initialize(staderAdmin, address(staderConfig)); | ||
|
|
||
| vm.startPrank(staderAdmin); | ||
| staderConfig.updateSDIncentiveController(address(sdIncentiveController)); | ||
| staderConfig.updateSDUtilityPool(address(sdUtilityPool)); | ||
| vm.stopPrank(); | ||
| } | ||
|
|
||
| function setupIncentive(uint256 incentiveAmount, uint256 duration) public returns(uint256, uint256) { | ||
| vm.assume(incentiveAmount > 0); | ||
| vm.assume(duration > 0); | ||
|
|
||
| incentiveAmount = ((incentiveAmount % 10000) + 1) * 1e18; | ||
| duration = ((duration%10) + 1) * 100; | ||
| incentiveAmount = (incentiveAmount / duration) * duration; | ||
| staderToken.transfer(staderManager, incentiveAmount); | ||
|
|
||
| vm.startPrank(staderManager); | ||
| staderToken.approve(address(sdIncentiveController), incentiveAmount); | ||
| sdIncentiveController.start(incentiveAmount, duration); | ||
| vm.stopPrank(); | ||
|
|
||
| return (incentiveAmount, duration); | ||
| } | ||
|
|
||
| function test_Initialize() public { | ||
| ProxyAdmin admin = new ProxyAdmin(); | ||
| SDIncentiveController sdIncentiveControllerImpl = new SDIncentiveController(); | ||
| TransparentUpgradeableProxy sdIncentiveControllerProxy = new TransparentUpgradeableProxy( | ||
| address(sdIncentiveControllerImpl), | ||
| address(admin), | ||
| '' | ||
| ); | ||
| SDIncentiveController sdIncentiveController2 = SDIncentiveController(address(sdIncentiveControllerProxy)); | ||
| sdIncentiveController2.initialize(staderAdmin, address(staderConfig)); | ||
| } | ||
|
|
||
| function test_VerifyInitialize() public { | ||
| assertEq(address(sdIncentiveController.staderConfig()), address(staderConfig)); | ||
| assertEq(sdIncentiveController.emissionPerBlock(), 0); | ||
| assertEq(sdIncentiveController.rewardEndBlock(), 0); | ||
| assertEq(sdIncentiveController.lastUpdateBlockNumber(), 0); | ||
| assertEq(sdIncentiveController.rewardPerTokenStored(), 0); | ||
| assertEq(sdIncentiveController.DECIMAL(), 1e18); | ||
| assertTrue(sdIncentiveController.hasRole(sdIncentiveController.DEFAULT_ADMIN_ROLE(), staderAdmin)); | ||
| } | ||
|
|
||
| function test_Simple(uint128 sdAmount, uint16 randomSeed, uint256 incentiveAmount, uint256 duration) public { | ||
| vm.assume(randomSeed > 1); | ||
|
|
||
| (incentiveAmount, duration) = setupIncentive(incentiveAmount, duration); | ||
|
|
||
| uint256 deployerSDBalance = staderToken.balanceOf(address(this)); | ||
| vm.assume(sdAmount <= deployerSDBalance / 4 && sdAmount > 1000e18); | ||
|
|
||
| address user = vm.addr(randomSeed); | ||
| address user2 = vm.addr(randomSeed - 1); | ||
|
|
||
| staderToken.transfer(user, sdAmount); | ||
| staderToken.transfer(user2, sdAmount); | ||
|
|
||
| assertEq(sdIncentiveController.earned(user), 0); | ||
| assertEq(sdIncentiveController.earned(user2), 0); | ||
|
|
||
| vm.startPrank(user2); | ||
| staderToken.approve(address(sdUtilityPool), sdAmount); | ||
| sdUtilityPool.delegate(sdAmount); | ||
| vm.stopPrank(); | ||
|
|
||
| vm.roll(block.number + 10); | ||
|
|
||
| assertApproxEqAbs(sdIncentiveController.earned(user2), incentiveAmount/duration*10, 1e9); | ||
|
|
||
| vm.startPrank(user); | ||
| staderToken.approve(address(sdUtilityPool), sdAmount); | ||
| sdUtilityPool.delegate(sdAmount); | ||
| vm.stopPrank(); | ||
|
|
||
| vm.roll(block.number + 20); | ||
|
|
||
| vm.startPrank(user2); | ||
| sdUtilityPool.requestWithdrawWithSDAmount(sdAmount/2); | ||
| vm.stopPrank(); | ||
|
|
||
| vm.startPrank(staderAdmin); | ||
| sdUtilityPool.updateMinBlockDelayToFinalizeRequest(0); | ||
| sdUtilityPool.finalizeDelegatorWithdrawalRequest(); | ||
| vm.stopPrank(); | ||
|
|
||
| vm.startPrank(user2); | ||
| uint256 earned = sdIncentiveController.earned(user2); | ||
| sdUtilityPool.claim(1); | ||
| assertEq(staderToken.balanceOf(user2), earned + sdAmount/2); | ||
| vm.stopPrank(); | ||
|
|
||
| vm.roll(block.number + duration); | ||
| assertApproxEqAbs(earned + sdIncentiveController.earned(user2) + sdIncentiveController.earned(user), incentiveAmount, 1e9); | ||
| } | ||
| } | ||
