Improve maxCycles test coverage (#336)

drips-network · Oct 11, 2023 · dbce0cc · dbce0cc
1 parent 6743435
commit dbce0cc
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/test/Streams.t.sol b/test/Streams.t.sol
@@ -1372,6 +1372,25 @@ contract StreamsTest is Test, PseudoRandomUtils, Streams {
         receiveStreams(receiver, _cycleSecs);
     }
 
+    function testReceiveAllStreamsCyclesWithLowMaxCycles() public {
+        // Enough for 3 cycles
+        uint128 amt = _cycleSecs * 3;
+        skipToCycleEnd();
+        setStreams(sender, 0, amt, recv(receiver, 1), _cycleSecs * 3);
+        skipToCycleEnd();
+        skipToCycleEnd();
+        skipToCycleEnd();
+        receiveStreams({
+            accountId: receiver,
+            maxCycles: 4,
+            expectedReceivedAmt: _cycleSecs * 3,
+            expectedReceivedCycles: 3,
+            expectedAmtAfter: 0,
+            expectedCyclesAfter: 0
+        });
+        receiveStreams(receiver, 0);
+    }
+
     function testSenderCanStreamToThemselves() public {
         uint128 amt = _cycleSecs * 3;
         skipToCycleEnd();