admits cancel during claiming

Author: maxcoto

src/DealNFT.sol

@@ -331,7 +331,7 @@ contract DealNFT is ERC721, IDealNFT, ReentrancyGuard {
      */
     function cancel() external {
         if(msg.sender != config.sponsor && msg.sender != config.arbitrator) revert NotAuthorized();
-        if(state() > State.Active) revert CannotCancel();
+        if(state() > State.Claiming) revert CannotCancel();
         config.cancelled = true;
         emit StateUpdated(State.Cancelled);
     }

test/DealNFTCancel.t.sol

@@ -33,10 +33,24 @@ contract DealNFTCancelTest is Test, DealSetup {
         deal.cancel();
     }
 
-    function test_RevertWhen_CancelWhenClaiming() public {
+    function test_CancelWhenClaiming() public {
         skip(15 days);
         assertEq(uint(deal.state()), uint256(DealNFT.State.Claiming));
 
+        vm.prank(sponsor);
+        deal.cancel();
+        
+        assertEq(uint(deal.state()), uint256(DealNFT.State.Cancelled));
+    }
+
+    function test_RevertWhen_CancelWhenCancelled() public {
+        skip(15 days);
+
+        vm.prank(sponsor);
+        deal.cancel();
+
+        assertEq(uint(deal.state()), uint256(DealNFT.State.Cancelled));
+
         vm.expectRevert(DealNFT.CannotCancel.selector);
         vm.prank(sponsor);
         deal.cancel();