stake-contract: allow partial reward withdrawal

dusk-network · Aug 31, 2024 · 424aba2 · 424aba2
1 parent b08e899
commit 424aba2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/contracts/stake/src/state.rs b/contracts/stake/src/state.rs
@@ -190,8 +190,8 @@ impl StakeState {
             panic!("There is no reward available to withdraw");
         }
 
-        if value != loaded_stake.reward {
-            panic!("Value withdrawn different from available reward");
+        if value > loaded_stake.reward {
+            panic!("Value withdrawn higher than available reward");
         }
 
         // check signature is correct
@@ -207,7 +207,7 @@ impl StakeState {
                 .expect("Withdrawing reward should succeed");
 
         // update the state accordingly
-        loaded_stake.reward = 0;
+        loaded_stake.reward -= value;
         rusk_abi::emit(
             "withdraw",
             StakeWithReceiverEvent {