test: Add fuzz test to timestamp in claimReferralFees

rabbitholegg · Apr 9, 2024 · 14e515b · 14e515b
1 parent 1c52379
commit 14e515b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/Quest.t.sol b/test/Quest.t.sol
@@ -330,7 +330,8 @@ contract TestQuest is Test, TestUtils, Errors, Events {
                         CLAIM REFERRAL FEES
     //////////////////////////////////////////////////////////////*/
 
-    function testFuzz_claimReferralFees() public {
+    function test_fuzz_claimReferralFees(uint96 timestamp) public {
+        vm.assume(timestamp > END_TIME);
         QuestFactoryMock(questFactoryMock).setMintFee(CLAIM_FEE);
         QuestFactoryMock(questFactoryMock).setNumberMinted(TOTAL_PARTICIPANTS);
 
@@ -360,7 +361,7 @@ contract TestQuest is Test, TestUtils, Errors, Events {
             "referral claims for all referrers should equal the reward amount (single claim)"
         );
 
-        vm.warp(END_TIME);
+        vm.warp(timestamp);
         vm.prank(referrer);
         quest.claimReferralFees(referrer);