From 5a350de55ddb3ba502e92a901975644e6ca0e644 Mon Sep 17 00:00:00 2001
From: ququzone <xueping.yang@gmail.com>
Date: Thu, 24 Oct 2024 13:46:41 +0800
Subject: [PATCH] enhance fixed rewards pool tests

---
 test/FixedRewardPool.t.sol | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/FixedRewardPool.t.sol b/test/FixedRewardPool.t.sol
index e6cd0c7..8d03d8a 100644
--- a/test/FixedRewardPool.t.sol
+++ b/test/FixedRewardPool.t.sol
@@ -110,13 +110,16 @@ contract TestFixedRewardPool is Test {
     fixedRewardPool.withdraw(1);
     assertEq(bob.balance, 0.225 ether);
     assertEq(testNFT.ownerOf(1), bob);
-    fixedRewardPool.claimRewards(alice);
-    assertEq(alice.balance, 0.075 ether);
+    assertEq(alice.balance, 0.05 ether);
     assertEq(fixedRewardPool.totalStakedWeight(), 100);
     assertEq(fixedRewardPool.tokenWeight(1), 0);
 
     vm.roll(15);
     assertEq(fixedRewardPool.pendingReward(bob), 0);
+    fixedRewardPool.claimRewards(alice);
+    assertEq(alice.balance, 0.075 ether);
+
+    vm.roll(20);
     vm.prank(alice);
     fixedRewardPool.withdraw(3);
     assertEq(alice.balance, 0.075 ether);