Updated to check prizes for both canaries

GenerationSoftware · Apr 2, 2024 · 37d69cc · 37d69cc
1 parent 89baf8a
commit 37d69cc
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/lib/pt-v5-prize-pool b/lib/pt-v5-prize-pool
diff --git a/src/Claimer.sol b/src/Claimer.sol
@@ -242,7 +242,7 @@ contract Claimer {
     }
     uint256 _maxFee = _computeMaxFee(_tier);
     SD59x18 perTimeUnit = LinearVRGDALib.getPerTimeUnit(
-      prizePool.estimatedPrizeCount(),
+      prizePool.estimatedPrizeCountWithBothCanaries(),
       timeToReachMaxFee
     );
     uint256 elapsed = block.timestamp - (prizePool.lastAwardedDrawAwardedAt());

diff --git a/test/Claimer.t.sol b/test/Claimer.t.sol
@@ -332,7 +332,7 @@ contract ClaimerTest is Test {
     uint prizeCount = 10;
     vm.mockCall(
       address(prizePool),
-      abi.encodeWithSignature("estimatedPrizeCount()"),
+      abi.encodeWithSignature("estimatedPrizeCountWithBothCanaries()"),
       abi.encode(prizeCount)
     );
     vm.mockCall(
@@ -358,7 +358,7 @@ contract ClaimerTest is Test {
     uint prizeCount = 10;
     vm.mockCall(
       address(prizePool),
-      abi.encodeWithSignature("estimatedPrizeCount()"),
+      abi.encodeWithSignature("estimatedPrizeCountWithBothCanaries()"),
       abi.encode(prizeCount)
     );
     vm.mockCall(
@@ -390,7 +390,7 @@ contract ClaimerTest is Test {
     );
     vm.mockCall(
       address(prizePool),
-      abi.encodeWithSignature("estimatedPrizeCount()"),
+      abi.encodeWithSignature("estimatedPrizeCountWithBothCanaries()"),
       abi.encodePacked(ESTIMATED_PRIZES)
     );
     vm.mockCall(