add revert check in the fuzz test

PufferFinance · Sep 23, 2024 · f8a4bd4 · f8a4bd4
1 parent 23f2c7f
commit f8a4bd4
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/mainnet-contracts/test/unit/PufferWithdrawalManager.t.sol b/mainnet-contracts/test/unit/PufferWithdrawalManager.t.sol
@@ -149,6 +149,13 @@ contract PufferWithdrawalManagerTest is UnitTestHelper {
 
             withdrawalManager.completeQueuedWithdrawal(i);
         }
+
+        uint256[] memory batches = new uint256[](1);
+        batches[0] = 1;
+
+        // If there is no excess ETH, this should revert
+        vm.expectRevert(IPufferWithdrawalManager.AlreadyReturned.selector);
+        withdrawalManager.returnExcessETHToVault(batches);
     }
 
     /**