Test add more rewards after the contract is created

ChorusOne · Nov 27, 2023 · 231de98 · 231de98
1 parent 3fd9291
commit 231de98
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/FeeRewardsManager.t.sol b/test/FeeRewardsManager.t.sol
@@ -68,6 +68,12 @@ contract FeeRewardsTest is Test {
         // We've got the ether.
         assertEq(address(feeRewardsManager).balance, 0 ether);
         assertEq(address(101).balance, 2.8 ether);
+
+        vm.deal(derivedAddr, 10 ether);
+        RewardsCollector(payable(addr)).collectRewards();
+        assertEq(addr.balance, 0 ether);
+        assertEq(withdrawalCredential.balance, 7.2 ether + 7.2 ether);
+        assertEq(address(101).balance, 2.8 ether);
     }
 
     function createWithdrawalSimulateRewards(