saved files

jito-foundation · Jul 16, 2024 · 675c92c · 675c92c
1 parent 9042131
commit 675c92c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 10 deletions.
diff --git a/tests/src/steward_fixtures.rs b/tests/src/steward_fixtures.rs
@@ -934,10 +934,8 @@ impl Default for StateMachineFixtures {
             stake_deposit_unstake_total: 0,
             delegations: [Delegation::default(); MAX_VALIDATORS],
             instant_unstake: BitMask::default(),
-            compute_delegations_completed: false.into(),
-            rebalance_completed: false.into(),
+            status_flags: 0,
             validators_added: 0,
-            checked_validators_removed_from_list: false.into(),
             validators_to_remove: BitMask::default(),
             _padding0: [0; STATE_PADDING_0_SIZE],
         };

diff --git a/tests/tests/steward/test_state_transitions.rs b/tests/tests/steward/test_state_transitions.rs
@@ -3,7 +3,7 @@
     These tests cover all possible state transitions when calling the `transition` method on the `StewardState` struct.
 */
 
-use jito_steward::{constants::MAX_VALIDATORS, Delegation, StewardStateEnum};
+use jito_steward::{constants::MAX_VALIDATORS, Delegation, StewardStateEnum, REBALANCE};
 use tests::steward_fixtures::StateMachineFixtures;
 
 #[test]
@@ -214,7 +214,7 @@ pub fn test_idle_noop() {
 
     // Case 2: still after instant_unstake_epoch_progress but after rebalance is completed
     clock.slot = epoch_schedule.get_last_slot_in_epoch(clock.epoch);
-    state.rebalance_completed = true.into();
+    state.set_flag(REBALANCE);
     let res = state.transition(clock, parameters, epoch_schedule);
     assert!(res.is_ok());
     assert!(matches!(state.state_tag, StewardStateEnum::Idle));

diff --git a/utils/steward-cli/src/commands/info/view_state.rs b/utils/steward-cli/src/commands/info/view_state.rs
@@ -165,11 +165,7 @@ fn _print_default_state(
         "Stake Deposit Unstake Total: {}\n",
         state.stake_deposit_unstake_total
     );
-    formatted_string += &format!(
-        "Compute Delegations Completed: {:?}\n",
-        state.compute_delegations_completed
-    );
-    formatted_string += &format!("Rebalance Completed: {:?}\n", state.rebalance_completed);
+
     formatted_string += &format!("Padding0 Length: {}\n", state._padding0.len());
     formatted_string += "---------------------";