Updated to check prizes for both canaries

GenerationSoftware · Apr 1, 2024 · f333899 · f333899
1 parent f14a0da
commit f333899
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
@@ -252,7 +252,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
@@ -323,7 +323,7 @@ contract ClaimerTest is Test {
     uint prizeCount = 10;
     vm.mockCall(
       address(prizePool),
-      abi.encodeWithSignature("estimatedPrizeCount()"),
+      abi.encodeWithSignature("estimatedPrizeCountWithBothCanaries()"),
       abi.encode(prizeCount)
     );
     vm.mockCall(
@@ -349,7 +349,7 @@ contract ClaimerTest is Test {
     uint prizeCount = 10;
     vm.mockCall(
       address(prizePool),
-      abi.encodeWithSignature("estimatedPrizeCount()"),
+      abi.encodeWithSignature("estimatedPrizeCountWithBothCanaries()"),
       abi.encode(prizeCount)
     );
     vm.mockCall(
@@ -381,7 +381,7 @@ contract ClaimerTest is Test {
     );
     vm.mockCall(
       address(prizePool),
-      abi.encodeWithSignature("estimatedPrizeCount()"),
+      abi.encodeWithSignature("estimatedPrizeCountWithBothCanaries()"),
       abi.encodePacked(ESTIMATED_PRIZES)
     );
     vm.mockCall(