Test coverage back to 100% (#69)

* add test for token that reverts but doesn't return any data

* back to 100% coverage

Author: amiecorso

test/src/PaymentEscrow/capture.t.sol

@@ -7,6 +7,7 @@ import {PaymentEscrow} from "../../../src/PaymentEscrow.sol";
 import {MockRevertOnTransferToken} from "../../mocks/MockRevertOnTransferToken.sol";
 import {MockFailOnTransferToken} from "../../mocks/MockFailOnTransferToken.sol";
 import {PaymentEscrowBase} from "../../base/PaymentEscrowBase.sol";
+import {MockERC3009Token} from "../../mocks/MockERC3009Token.sol";
 
 contract CaptureTest is PaymentEscrowBase {
     function test_reverts_whenNotOperator(uint120 authorizedAmount, address sender) public {
@@ -335,7 +336,9 @@ contract CaptureTest is PaymentEscrowBase {
         paymentEscrow.capture(paymentInfo, authorizedAmount, captureFeeBps, invalidFeeReceiver);
     }
 
-    function test_reverts_ifSendTokensReverts(uint120 authorizedAmount, bytes calldata revertData) public {
+    function test_reverts_ifSendTokensReverts_undeployedTokenStore(uint120 authorizedAmount, bytes calldata revertData)
+        public
+    {
         vm.assume(authorizedAmount > 0);
         address revertingToken =
             address(new MockRevertOnTransferToken(address(preApprovalPaymentCollector), revertData));
@@ -357,6 +360,45 @@ contract CaptureTest is PaymentEscrowBase {
         vm.stopPrank();
     }
 
+    function test_reverts_ifSendTokensReverts_deployedTokenStore(uint120 authorizedAmount, bytes calldata revertData)
+        public
+    {
+        vm.assume(authorizedAmount > 0);
+        // process a payment first to make sure the token store is deployed
+        PaymentEscrow.PaymentInfo memory initialPaymentInfo = _createPaymentInfo(payerEOA, authorizedAmount);
+        bytes memory initialSignature = _signERC3009ReceiveWithAuthorizationStruct(initialPaymentInfo, payer_EOA_PK);
+        mockERC3009Token.mint(payerEOA, authorizedAmount);
+        vm.prank(initialPaymentInfo.operator);
+        paymentEscrow.charge(
+            initialPaymentInfo,
+            authorizedAmount,
+            address(erc3009PaymentCollector),
+            initialSignature,
+            initialPaymentInfo.minFeeBps,
+            initialPaymentInfo.feeReceiver
+        );
+
+        // create a new token that reverts on transfer
+        address revertingToken =
+            address(new MockRevertOnTransferToken(address(preApprovalPaymentCollector), revertData));
+        MockRevertOnTransferToken(revertingToken).mint(payerEOA, authorizedAmount);
+
+        PaymentEscrow.PaymentInfo memory paymentInfo = _createPaymentInfo(payerEOA, authorizedAmount);
+        paymentInfo.token = revertingToken;
+
+        vm.prank(payerEOA);
+        MockRevertOnTransferToken(revertingToken).approve(address(preApprovalPaymentCollector), authorizedAmount);
+        vm.prank(payerEOA);
+        preApprovalPaymentCollector.preApprove(paymentInfo);
+
+        vm.startPrank(paymentInfo.operator);
+        paymentEscrow.authorize(paymentInfo, authorizedAmount, address(preApprovalPaymentCollector), "");
+
+        vm.expectRevert(abi.encodeWithSelector(MockRevertOnTransferToken.CustomRevert.selector, revertData));
+        paymentEscrow.capture(paymentInfo, authorizedAmount, paymentInfo.minFeeBps, paymentInfo.feeReceiver);
+        vm.stopPrank();
+    }
+
     function test_reverts_ifSendTokensFails(uint120 authorizedAmount) public {
         vm.assume(authorizedAmount > 0);
         address revertingToken = address(new MockFailOnTransferToken(address(preApprovalPaymentCollector)));