Added tests

1inch · Jan 18, 2024 · 873402d · 873402d
1 parent 1828667
commit 873402d
Showing 1 changed file with 49 additions and 0 deletions.
diff --git a/test/unit/Escrow.t.sol b/test/unit/Escrow.t.sol
@@ -15,6 +15,38 @@ contract EscrowTest is BaseSetup {
     }
 
     /* solhint-disable func-name-mixedcase */
+    // Only resolver can withdraw
+    function test_NoWithdrawalByAnyone() public {
+        // deploy escrow
+        (
+            IOrderMixin.Order memory order,
+            bytes32 orderHash,
+            bytes memory extraData,
+            /* bytes memory extension */,
+            Escrow srcClone
+        ) = _prepareDataSrc(SECRET, MAKING_AMOUNT, TAKING_AMOUNT, true);
+
+        (bool success,) = address(srcClone).call{value: SRC_SAFETY_DEPOSIT}("");
+        assertEq(success, true);
+        usdc.transfer(address(srcClone), MAKING_AMOUNT);
+
+        vm.prank(address(limitOrderProtocol));
+        escrowFactory.postInteraction(
+            order,
+            "", // extension
+            orderHash,
+            bob.addr, // taker
+            MAKING_AMOUNT,
+            TAKING_AMOUNT,
+            0, // remainingMakingAmount
+            extraData
+        );
+
+        // withdraw
+        vm.expectRevert(IEscrow.InvalidCaller.selector);
+        srcClone.withdrawSrc(SECRET);
+    }
+
     function test_NoWithdrawalDuringFinalityLockSrc() public {
         // deploy escrow
         (
@@ -465,6 +497,23 @@ contract EscrowTest is BaseSetup {
         assertEq(address(dstClone).balance, balanceEscrowNative - DST_SAFETY_DEPOSIT);
     }
 
+    // Only resolver can cancel
+    function test_NoCancelByAnyoneDst() public {
+        (
+            IEscrowFactory.DstEscrowImmutablesCreation memory immutables,
+            Escrow dstClone
+        ) = _prepareDataDst(SECRET, TAKING_AMOUNT, alice.addr, bob.addr, address(dai));
+
+        // deploy escrow
+        vm.prank(bob.addr);
+        escrowFactory.createEscrow{value: DST_SAFETY_DEPOSIT}(immutables);
+
+        // cancel
+        vm.warp(block.timestamp + dstTimelocks.finality + dstTimelocks.unlock + dstTimelocks.publicUnlock + 100);
+        vm.expectRevert(IEscrow.InvalidCaller.selector);
+        dstClone.cancelDst();
+    }
+
     function test_NoCancelDuringResolverUnlockDst() public {
         (
             IEscrowFactory.DstEscrowImmutablesCreation memory immutables,