setPostOpCost tests

bcnmy · Jul 4, 2024 · 199d312 · 199d312
1 parent 5da6fbe
commit 199d312
Show file tree

Hide file tree

Showing 5 changed files with 39 additions and 81 deletions.
diff --git a/contracts/test/Foo.sol b/contracts/test/Foo.sol
diff --git a/contracts/test/Lock.sol b/contracts/test/Lock.sol
diff --git a/scripts/foundry/Deploy.s.sol b/scripts/foundry/Deploy.s.sol
diff --git a/test/foundry/unit/concrete/TestSponsorshipPaymasterWithPremiumTest.t.sol b/test/foundry/unit/concrete/TestSponsorshipPaymasterWithPremiumTest.t.sol
@@ -1,6 +1,7 @@
 // SPDX-License-Identifier: Unlicensed
 pragma solidity ^0.8.26;
 
+import { console2 } from "forge-std/src/console2.sol";
 import { NexusTestBase } from "../../base/NexusTestBase.sol";
 import { IBiconomySponsorshipPaymaster } from "../../../../contracts/interfaces/IBiconomySponsorshipPaymaster.sol";
 import { BiconomySponsorshipPaymaster } from "../../../../contracts/sponsorship/SponsorshipPaymasterWithPremium.sol";
@@ -25,13 +26,15 @@ contract TestSponsorshipPaymasterWithPremium is NexusTestBase {
         assertEq(address(testArtifact.entryPoint()), ENTRYPOINT_ADDRESS);
         assertEq(testArtifact.verifyingSigner(), PAYMASTER_SIGNER.addr);
         assertEq(testArtifact.feeCollector(), PAYMASTER_FEE_COLLECTOR.addr);
+        assertEq(testArtifact.postopCost(), 0 wei);
     }
 
     function test_CheckInitialPaymasterState() external view {
         assertEq(bicoPaymaster.owner(), PAYMASTER_OWNER.addr);
         assertEq(address(bicoPaymaster.entryPoint()), ENTRYPOINT_ADDRESS);
         assertEq(bicoPaymaster.verifyingSigner(), PAYMASTER_SIGNER.addr);
         assertEq(bicoPaymaster.feeCollector(), PAYMASTER_FEE_COLLECTOR.addr);
+        assertEq(bicoPaymaster.postopCost(), 0 wei);
     }
 
     function test_OwnershipTransfer() external prankModifier(PAYMASTER_OWNER.addr) {
@@ -261,4 +264,26 @@ contract TestSponsorshipPaymasterWithPremium is NexusTestBase {
         vm.expectRevert("withdraw failed");
         bicoPaymaster.withdrawEth(payable(ALICE_ADDRESS), ethAmount);
     }
+
+    function test_SetPostopCost() external prankModifier(PAYMASTER_OWNER.addr) {
+        uint48 initialPostopCost = bicoPaymaster.postopCost();
+        assertEq(initialPostopCost, 0 wei);
+        uint48 newPostopCost = initialPostopCost + 1 wei;
+
+        vm.expectEmit(true, true, false, true, address(bicoPaymaster));
+        emit IBiconomySponsorshipPaymaster.PostopCostChanged(initialPostopCost, newPostopCost);
+        bicoPaymaster.setPostopCost(newPostopCost);
+
+        uint48 resultingPostopCost = bicoPaymaster.postopCost();
+        assertEq(resultingPostopCost, newPostopCost);
+    }
+
+    function test_RevertIf_SetPostopCostToHigh() external prankModifier(PAYMASTER_OWNER.addr) {
+        uint48 initialPostopCost = bicoPaymaster.postopCost();
+        assertEq(initialPostopCost, 0 wei);
+        uint48 newPostopCost = initialPostopCost + 200_001 wei;
+
+        vm.expectRevert("Gas overhead too high");
+        bicoPaymaster.setPostopCost(newPostopCost);
+    }
 }
diff --git a/test/foundry/unit/fuzz/TestFuzz_TestSponsorshipPaymasterWithPremium.t.sol b/test/foundry/unit/fuzz/TestFuzz_TestSponsorshipPaymasterWithPremium.t.sol
@@ -76,4 +76,18 @@ contract TestFuzz_SponsorshipPaymasterWithPremium is NexusTestBase {
         assertEq(ALICE_ADDRESS.balance, initialAliceBalance + ethAmount);
         assertEq(address(bicoPaymaster).balance, 0 ether);
     }
+
+    function testFuzz_SetPostopCost(uint48 value) external prankModifier(PAYMASTER_OWNER.addr) {
+        vm.assume(value <= 200_000 wei);
+        uint48 initialPostopCost = bicoPaymaster.postopCost();
+        assertEq(initialPostopCost, 0 wei);
+        uint48 newPostopCost = value;
+
+        vm.expectEmit(true, true, false, true, address(bicoPaymaster));
+        emit IBiconomySponsorshipPaymaster.PostopCostChanged(initialPostopCost, newPostopCost);
+        bicoPaymaster.setPostopCost(newPostopCost);
+
+        uint48 resultingPostopCost = bicoPaymaster.postopCost();
+        assertEq(resultingPostopCost, newPostopCost);
+    }
 }