Adds more testing.

primev · Mar 5, 2024 · fbc9df7 · fbc9df7
1 parent 12cef51
commit fbc9df7
Show file tree

Hide file tree

Showing 2 changed files with 60 additions and 1 deletion.
diff --git a/contracts/Oracle.sol b/contracts/Oracle.sol
@@ -100,7 +100,7 @@ contract Oracle is Ownable {
     ) external onlyOwner {
         // Check graffiti against registered builder IDs
         address builder = blockBuilderNameToAddress[blockBuilderName];
-
+        require(residualBidAfterDecay <= 100, "Residual bid after decay cannot be greater than 100 percent");
         IPreConfCommitmentStore.PreConfCommitment memory commitment = preConfContract.getCommitment(commitmentIndex);
         if (commitment.commiter == builder && commitment.blockNumber == blockNumber) {
                 processCommitment(commitmentIndex, isSlash, residualBidAfterDecay);

diff --git a/test/PreConfirmationConfTest.sol b/test/PreConfirmationConfTest.sol
@@ -502,6 +502,65 @@ contract TestPreConfCommitmentStore is Test {
         }
     }
 
+
+    function test_InitiateRewardFullyDecayed() public {
+        // Assuming you have a stored commitment
+        {
+            (address bidder, ) = makeAddrAndKey("alice");
+            vm.deal(bidder, 5 ether);
+            vm.prank(bidder);
+            bidderRegistry.prepay{value: 2 ether}();
+
+            // Step 1: Verify that the commitment has not been used before
+            bytes32 bidHash = verifyCommitmentNotUsed(
+                _testCommitmentAliceBob.txnHash,
+                _testCommitmentAliceBob.bid,
+                _testCommitmentAliceBob.blockNumber,
+                _testCommitmentAliceBob.decayStartTimestamp,
+                _testCommitmentAliceBob.decayEndTimestamp,
+                _testCommitmentAliceBob.bidSignature
+            );
+            bytes32 preConfHash = preConfCommitmentStore.getPreConfHash(
+                _testCommitmentAliceBob.txnHash,
+                _testCommitmentAliceBob.bid,
+                _testCommitmentAliceBob.blockNumber,
+                _testCommitmentAliceBob.decayStartTimestamp,
+                _testCommitmentAliceBob.decayEndTimestamp,
+                bidHash,
+                _bytesToHexString(_testCommitmentAliceBob.bidSignature)
+            );
+
+            // Verify that the commitment has not been used before
+            (bool commitmentUsed, , , , , , , , , , , , ) = preConfCommitmentStore
+                .commitments(preConfHash);
+            assert(commitmentUsed == false);
+            bytes32 index = preConfCommitmentStore.storeCommitment(
+                _testCommitmentAliceBob.bid,
+                _testCommitmentAliceBob.blockNumber,
+                _testCommitmentAliceBob.txnHash,
+                _testCommitmentAliceBob.decayStartTimestamp,
+                _testCommitmentAliceBob.decayEndTimestamp,
+                _testCommitmentAliceBob.bidSignature,
+                _testCommitmentAliceBob.commitmentSignature
+            );
+            (address commiter, ) = makeAddrAndKey("bob");
+            vm.deal(commiter, 5 ether);
+            vm.prank(commiter);
+            providerRegistry.registerAndStake{value: 4 ether}();
+            vm.prank(feeRecipient);
+            preConfCommitmentStore.initiateReward(index, 0);
+
+            (commitmentUsed, , , , , , , , , , , , ) = preConfCommitmentStore
+                .commitments(index);
+            // Verify that the commitment has been marked as used
+            assert(commitmentUsed == true);
+            // commitmentHash value is internal to contract and not asserted
+
+            assert(bidderRegistry.bidderPrepaidBalances(bidder) == 2 ether);
+            assert(bidderRegistry.providerAmount(commiter) == 0 ether);
+        }
+
+    }
     function _bytesToHexString(
         bytes memory _bytes
     ) public pure returns (string memory) {