test/tx_redelegate: prevent redelegation delegator from being validator

anoma · Jan 26, 2024 · cd43f1a · cd43f1a
1 parent aa1f5e5
commit cd43f1a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/wasm/wasm_source/src/tx_redelegate.rs b/wasm/wasm_source/src/tx_redelegate.rs
@@ -403,9 +403,15 @@ mod tests {
             token::testing::arb_amount_non_zero_ceiled(max_amount),
         )
             .prop_filter_map(
-                "Src and dest validator must not be the same",
+                "Src and dest validator must not be the same and owner must \
+                 not be a validator",
                 |(src_validator, dest_validator, owner, amount)| {
-                    if src_validator == dest_validator {
+                    let is_owner_validator = matches!(
+                        &owner,
+                        Address::Established(owner)
+                        if owner == &src_validator || owner == &dest_validator
+                    );
+                    if is_owner_validator || src_validator == dest_validator {
                         None
                     } else {
                         let src_validator = Address::Established(src_validator);