chore: simplify test

KintoXYZ · Jan 27, 2024 · 2c1a6c7 · 2c1a6c7
1 parent 3cb92a9
commit 2c1a6c7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/src/Faucet.sol b/src/Faucet.sol
@@ -19,9 +19,11 @@ contract Faucet is Initializable, UUPSUpgradeable, OwnableUpgradeable, IFaucet {
     using SignatureChecker for address;
 
     /* ============ Events ============ */
+
     event Claim(address indexed _to, uint256 _timestamp);
 
     /* ============ Constants ============ */
+
     uint256 public constant CLAIM_AMOUNT = 1 ether / 2500;
     uint256 public constant FAUCET_AMOUNT = 1 ether;
 

diff --git a/test/Faucet.t.sol b/test/Faucet.t.sol
@@ -130,10 +130,12 @@ contract FaucetTest is SharedSetup {
         vm.prank(_owner);
         _faucet.startFaucet{value: 1 ether}();
 
-        for (uint256 i = 1; i <= 2500; i++) {
-            vm.prank(vm.addr(i));
-            _faucet.claimKintoETH();
-        }
+        // reduce faucet balance to CLAIM AMOUNT
+        vm.deal(address(_faucet), _faucet.CLAIM_AMOUNT());
+
+        vm.prank(_user);
+        _faucet.claimKintoETH();
+
         // assert faucet is deactivated
         assertEq(address(_faucet).balance, 0);
         assertEq(_faucet.active(), false);