refactor(contracts): ensure cancel_stream reverts when empty/claimed

ApeWorX · Sep 6, 2024 · ce90246 · ce90246
1 parent 56b2e7b
commit ce90246
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/contracts/StreamManager.vy b/contracts/StreamManager.vy
@@ -224,6 +224,7 @@ def cancel_stream(
 
     funded_amount: uint256 = self.streams[creator][stream_id].funded_amount
     amount_locked: uint256 = funded_amount - self._amount_unlocked(creator, stream_id)
+    assert amount_locked > 0  # NOTE: reverts if stream doesn't exist, or already cancelled
     self.streams[creator][stream_id].funded_amount = funded_amount - amount_locked
 
     token: ERC20 = self.streams[creator][stream_id].token