Erc6909claims tests

src/test/MockERC6909Claims.sol → src/test/ERC6909ClaimsImplementation.sol

@@ -6,7 +6,7 @@ import {ERC6909Claims} from "../ERC6909Claims.sol";
 import {CurrencyLibrary, Currency} from "../types/Currency.sol";
 
 /// @notice Mock contract for testing ERC6909Claims
-contract MockERC6909Claims is ERC6909Claims {
+contract ERC6909ClaimsImplementation is ERC6909Claims {
     using CurrencyLibrary for Currency;
 
     /// @notice mocked mint logic

test/ERC6909Claims.t.sol → test/ERC6909ClaimsImplementation.t.sol

@@ -3,43 +3,58 @@ pragma solidity ^0.8.15;
 
 import {Test} from "forge-std/Test.sol";
 import {CurrencyLibrary, Currency} from "../src/types/Currency.sol";
-import {MockERC6909Claims} from "../src/test/MockERC6909Claims.sol";
+import {ERC6909ClaimsImplementation} from "../src/test/ERC6909ClaimsImplementation.sol";
 import {GasSnapshot} from "forge-gas-snapshot/GasSnapshot.sol";
 
 contract ERC6909ClaimsTest is Test, GasSnapshot {
     using CurrencyLibrary for Currency;
 
-    MockERC6909Claims token;
+    ERC6909ClaimsImplementation token;
 
     function setUp() public {
-        token = new MockERC6909Claims();
+        token = new ERC6909ClaimsImplementation();
     }
 
-    function test_burnFrom_withApproval(address sender, uint256 id, uint256 mintAmount, uint256 transferAmount)
-        public
-    {
+    function test_burnFrom_withApproval(address sender, uint256 id, uint256 mintAmount, uint256 burnAmount) public {
         token.mint(sender, id, mintAmount);
 
         vm.prank(sender);
         token.approve(address(this), id, mintAmount);
 
-        if (transferAmount > mintAmount) {
+        if (burnAmount > mintAmount) {
             vm.expectRevert();
         }
-        token.burnFrom(sender, id, transferAmount);
+        token.burnFrom(sender, id, burnAmount);
 
-        if (transferAmount <= mintAmount) {
+        if (burnAmount <= mintAmount) {
             if (mintAmount == type(uint256).max) {
                 assertEq(token.allowance(sender, address(this), id), type(uint256).max);
             } else {
                 if (sender != address(this)) {
-                    assertEq(token.allowance(sender, address(this), id), mintAmount - transferAmount);
+                    assertEq(token.allowance(sender, address(this), id), mintAmount - burnAmount);
                 } else {
                     assertEq(token.allowance(sender, address(this), id), mintAmount);
                 }
             }
-            assertEq(token.balanceOf(sender, id), mintAmount - transferAmount);
+            assertEq(token.balanceOf(sender, id), mintAmount - burnAmount);
+        }
+    }
+
+    function test_burnFrom_withOperator(address sender, uint256 id, uint256 mintAmount, uint256 burnAmount) public {
+        token.mint(sender, id, mintAmount);
+
+        vm.prank(sender);
+        token.setOperator(address(this), true);
+
+        if (burnAmount > mintAmount) {
+            vm.expectRevert();
+        }
+        token.burnFrom(sender, id, burnAmount);
+
+        if (burnAmount <= mintAmount) {
+            assertEq(token.balanceOf(sender, id), mintAmount - burnAmount);
         }
+        assertEq(token.allowance(sender, address(this), id), 0);
     }
 
     function test_burnFrom_revertsWithNoApproval() public {