Fix: add missing unit tests for LockingProcessor (partly failing)

synapsecns · Feb 28, 2024 · fe200d3 · fe200d3
1 parent 8f04487
commit fe200d3
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/packages/contracts-interchain/test/processors/LockingProcessor.t.sol b/packages/contracts-interchain/test/processors/LockingProcessor.t.sol
@@ -33,6 +33,18 @@ contract LockingProcessorTest is AbstractProcessorTest {
         factory.deployLockingProcessor(address(icToken), address(0));
     }
 
+    function test_calculateSwap_exactlyProcessorBalance() public {
+        assertEq(processor.calculateSwap(0, 1, START_BALANCE), START_BALANCE);
+        assertEq(processor.calculateSwap(1, 0, START_BALANCE), START_BALANCE);
+    }
+
+    function test_calculateSwap_overProcessorBalance() public {
+        // Should not be possible to unlock more than the processor balance
+        assertEq(processor.calculateSwap(0, 1, START_BALANCE + 1), 0);
+        // Could lock any amount of tokens
+        assertEq(processor.calculateSwap(1, 0, START_BALANCE + 1), START_BALANCE + 1);
+    }
+
     // Lock token: token (1) -> icToken (0)
     function test_swap_lockUnderlyingToken() public {
         uint256 amount = 100;