Test trying to change contract's owner with fallback

ChorusOne · Nov 27, 2023 · 5cc5aab · 5cc5aab
1 parent 66d3865
commit 5cc5aab
Showing 1 changed file with 2 additions and 16 deletions.
diff --git a/test/FeeRewardsManager.t.sol b/test/FeeRewardsManager.t.sol
@@ -177,28 +177,14 @@ contract FeeRewardsTest is Test {
     }
 
     function testSendToContractWithdrawalCredential() public {
-        WithdrawalContract withdrawalCredentialContract = new WithdrawalContract();
-        address addr = address(
-            createWithdrawalSimulateRewards(
-                address(withdrawalCredentialContract),
-                10 ether
-            )
-        );
-        RewardsCollector(payable(addr)).collectRewards();
-        assertEq(address(feeRewardsManager).balance, 2.8 ether);
-        assertEq(address(withdrawalCredentialContract).balance, 7.2 ether);
-    }
-
-    function testChangeOwnerWithContract() public {
-        WithdrawalContract withdrawalCredentialContract = new WithdrawalContract();
+        ChangeOwnerContract withdrawalCredentialContract = new ChangeOwnerContract();
         address addr = address(
             createWithdrawalSimulateRewards(
                 address(withdrawalCredentialContract),
                 10 ether
             )
         );
+        vm.expectRevert("Failed to send Ether back to withdrawal credential");
         RewardsCollector(payable(addr)).collectRewards();
-        assertEq(address(feeRewardsManager).balance, 2.8 ether);
-        assertEq(address(withdrawalCredentialContract).balance, 7.2 ether);
     }
 }