feat(validations): improve error stake validator/withdrawer error

witnet · Oct 30, 2024 · d8d9f7c · d8d9f7c
1 parent b2efe31
commit d8d9f7c
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/validations/src/validations.rs b/validations/src/validations.rs
@@ -1343,10 +1343,19 @@ pub fn validate_stake_transaction<'a>(
     }
 
     // A stake transaction can only stake on an existing validator if the withdrawer address is the same
-    stakes.check_validator_withdrawer(
+    match stakes.check_validator_withdrawer(
         st_tx.body.output.key.validator,
         st_tx.body.output.key.withdrawer,
-    )?;
+    ) {
+        Ok(_) => (),
+        Err(_) => {
+            return Err(TransactionError::NoStakeFound {
+                validator: st_tx.body.output.key.validator,
+                withdrawer: st_tx.body.output.key.withdrawer,
+            }
+            .into());
+        }
+    }
 
     // Check that the amount of coins to stake plus the alread staked amount is equal or smaller than the maximum allowed
     let stakes_key = QueryStakesKey::Key(StakeKey {