test(escrow-contract): add test for cancelling thawing escrow account…

… funds

Signed-off-by: Bryan Cole <[email protected]>

test/Escrow.t.sol

@@ -106,10 +106,29 @@ contract EscrowContractTest is Test {
         // Simulate passing the freeze period
         vm.warp(block.timestamp + WITHDRAW_ESCROW_FREEZE_PERIOD + 1);
 
+        // Cancel thaw and attempt to withdraw (expect revert)
+        vm.startPrank(SENDER_ADDRESS);
+        escrowContract.cancelThaw(receiverAddress);
+
         uint256 senderBalanceBeforeWithdraw = mockERC20.balanceOf(SENDER_ADDRESS);
+        vm.expectRevert(Escrow.EscrowNotThawing.selector);
         escrowContract.withdraw(receiverAddress);
         uint256 senderBalanceAfterWithdraw = mockERC20.balanceOf(SENDER_ADDRESS);
 
+        assertEq(senderBalanceAfterWithdraw - senderBalanceBeforeWithdraw, 0, "Incorrect removed amount");
+
+        // Sets msg.sender address for next contract calls until stop is called
+        vm.startPrank(SENDER_ADDRESS);
+        escrowContract.thaw(receiverAddress, ESCROW_AMOUNT);
+
+        // Simulate passing the freeze period
+        vm.warp(block.timestamp + WITHDRAW_ESCROW_FREEZE_PERIOD + 1);
+
+        senderBalanceBeforeWithdraw = mockERC20.balanceOf(SENDER_ADDRESS);
+        vm.expectRevert(Escrow.EscrowNotThawing.selector);
+        escrowContract.withdraw(receiverAddress);
+        senderBalanceAfterWithdraw = mockERC20.balanceOf(SENDER_ADDRESS);
+
         uint256 removedAmount = senderBalanceAfterWithdraw - senderBalanceBeforeWithdraw;
 
         assertEq(removedAmount, ESCROW_AMOUNT, "Incorrect removed amount");