Finish refactoring tests

anoma · Dec 8, 2023 · c14bec8 · c14bec8
1 parent 82c96d7
commit c14bec8
Show file tree

Hide file tree

Showing 6 changed files with 3,170 additions and 3,603 deletions.
diff --git a/proof_of_stake/src/tests.rs b/proof_of_stake/src/tests.rs
@@ -2,5 +2,7 @@ mod helpers;
 mod state_machine;
 mod state_machine_v2;
 mod test_helper_fns;
+mod test_pos;
+mod test_slash_and_redel;
 mod test_validator;
 mod utils;
diff --git a/proof_of_stake/src/tests/helpers.rs b/proof_of_stake/src/tests/helpers.rs
@@ -77,7 +77,7 @@ pub fn advance_epoch(s: &mut TestWlStorage, params: &PosParams) -> Epoch {
     current_epoch
 }
 
-fn arb_genesis_validators(
+pub fn arb_genesis_validators(
     size: Range<usize>,
     threshold: Option<token::Amount>,
 ) -> impl Strategy<Value = Vec<GenesisValidator>> {