merged in with evan

jito-foundation · Jul 23, 2024 · 17dcc07 · 17dcc07
2 parents 7b18620 + 4369738
commit 17dcc07
Showing 1 changed file with 12 additions and 10 deletions.
diff --git a/tests/tests/steward/test_integration.rs b/tests/tests/steward/test_integration.rs
@@ -965,15 +965,6 @@ async fn test_rebalance_increase() {
         &serialized_stake_pool_account(stake_pool_spl, std::mem::size_of::<StakePool>()).into(),
     );
 
-    let mut steward_state_account: StewardStateAccount =
-        fixture.load_and_deserialize(&fixture.steward_state).await;
-
-    steward_state_account.state.num_pool_validators += 1;
-    ctx.borrow_mut().set_account(
-        &fixture.steward_state,
-        &serialized_steward_state_account(steward_state_account).into(),
-    );
-
     let (stake_account_address, transient_stake_account_address, withdraw_authority) =
         fixture.stake_accounts_for_validator(vote_account).await;
 
@@ -1052,6 +1043,17 @@ async fn test_rebalance_increase() {
         &serialized_steward_state_account(steward_state_account).into(),
     );
 
+    let mut steward_state_account: StewardStateAccount =
+        fixture.load_and_deserialize(&fixture.steward_state).await;
+
+    // Force validator into the active set, don't wait for next cycle
+    steward_state_account.state.num_pool_validators += 1;
+    steward_state_account.state.validators_added -= 1;
+    ctx.borrow_mut().set_account(
+        &fixture.steward_state,
+        &serialized_steward_state_account(steward_state_account).into(),
+    );
+
     let reserve_before_rebalance = fixture.get_account(&fixture.stake_pool_meta.reserve).await;
 
     let tx = Transaction::new_signed_with_payer(
@@ -1264,7 +1266,7 @@ async fn test_rebalance_decrease() {
         fixture.load_and_deserialize(&fixture.steward_state).await;
 
     // Force validator into the active set, don't wait for next cycle
-    steward_state_account.state.num_pool_validators += 1;
+    // steward_state_account.state.num_pool_validators += 1;
     steward_state_account.state.validators_added -= 1;
     ctx.borrow_mut().set_account(
         &fixture.steward_state,