add require & test

ubiquity · Feb 16, 2024 · 6ce539b · 6ce539b
1 parent 75c78ba
commit 6ce539b
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/src/NftReward.sol b/src/NftReward.sol
@@ -218,6 +218,7 @@ contract NftReward is Initializable, ERC721Upgradeable, OwnableUpgradeable, Paus
      * @param _nonceValue Nonce value to invalidate
      */
     function invalidateNonce(uint256 _nonceValue) external onlyOwner {
+        require(!nonceRedeemed[_nonceValue], "Already minted");
         nonceRedeemed[_nonceValue] = true;
     }
 

diff --git a/test/NftReward.t.sol b/test/NftReward.t.sol
@@ -571,4 +571,52 @@ contract NftRewardTest is Test {
         assertEq(nftReward.nonceRedeemed(1), true);
         assertEq(nftReward.tokenIdCounter(), 0);
     }
+
+    function testRequireInvalidateNonce_ShouldRevert_IfNonceIsRedeemed() public {
+        // prepare arbitrary data keys
+        bytes32[] memory keys = new bytes32[](1);
+        keys[0] = keccak256("GITHUB_ORGANIZATION_NAME"); 
+        // prepare arbitrary data values
+        string[] memory values = new string[](1);
+        values[0] = "ubiquity";
+        // prepare mint request
+        NftReward.MintRequest memory mintRequest = NftReward.MintRequest({
+            beneficiary: user1,
+            deadline: block.timestamp + 1,
+            keys: keys,
+            nonce: 1,
+            values: values
+        });
+        // get mint request digest which should be signed
+        bytes32 digest = nftReward.getMintRequestDigest(mintRequest);
+        // minter signs mint request digest
+        (uint8 v, bytes32 r, bytes32 s) = vm.sign(minterPrivateKey, digest);
+        // get minter's signature
+        bytes memory signature = abi.encodePacked(r, s, v);
+
+        uint tokenId = 0;
+
+        // before
+        vm.expectRevert();
+        nftReward.ownerOf(tokenId);
+        assertEq(nftReward.nonceRedeemed(1), false);
+        assertEq(nftReward.tokenDataKeyExists(keccak256("GITHUB_ORGANIZATION_NAME")), false);
+
+        // user1 mints
+        vm.prank(user1);
+        nftReward.safeMint(mintRequest, signature);
+
+        // owner try to invalidate
+        vm.prank(owner);
+        vm.expectRevert("Already minted");
+        nftReward.invalidateNonce(1);
+
+        // after
+        assertEq(nftReward.nonceRedeemed(1), true);
+        assertEq(nftReward.tokenDataKeys(0), keccak256("GITHUB_ORGANIZATION_NAME"));
+        assertEq(nftReward.tokenDataKeyExists(keccak256("GITHUB_ORGANIZATION_NAME")), true);
+        assertEq(nftReward.ownerOf(tokenId), user1);
+        assertEq(nftReward.tokenIdCounter(), 1);
+        assertEq(nftReward.tokenData(0, keccak256("GITHUB_ORGANIZATION_NAME")), "ubiquity");
+    }
 }