90 lines
2.5 KiB
Rust
90 lines
2.5 KiB
Rust
#![cfg(feature = "test-bpf")]
|
|
|
|
mod helpers;
|
|
|
|
use {
|
|
crate::helpers::TEST_STAKE_AMOUNT,
|
|
helpers::*,
|
|
solana_program::{native_token, pubkey::Pubkey},
|
|
solana_program_test::*,
|
|
solana_sdk::signature::Signer,
|
|
spl_stake_pool::{borsh::try_from_slice_unchecked, stake, state},
|
|
};
|
|
|
|
async fn get_list_sum(banks_client: &mut BanksClient, validator_list_key: &Pubkey) -> u64 {
|
|
let validator_list = banks_client
|
|
.get_account(*validator_list_key)
|
|
.await
|
|
.expect("get_account")
|
|
.expect("validator stake list not none");
|
|
let validator_list =
|
|
try_from_slice_unchecked::<state::ValidatorList>(validator_list.data.as_slice()).unwrap();
|
|
|
|
validator_list
|
|
.validators
|
|
.iter()
|
|
.map(|info| info.balance)
|
|
.sum()
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_update_validator_list_balance() {
|
|
let (mut banks_client, payer, recent_blockhash) = program_test().start().await;
|
|
let stake_pool_accounts = StakePoolAccounts::new();
|
|
stake_pool_accounts
|
|
.initialize_stake_pool(&mut banks_client, &payer, &recent_blockhash)
|
|
.await
|
|
.unwrap();
|
|
|
|
// Add several accounts
|
|
let mut stake_accounts: Vec<ValidatorStakeAccount> = vec![];
|
|
const STAKE_ACCOUNTS: u64 = 3;
|
|
for _ in 0..STAKE_ACCOUNTS {
|
|
stake_accounts.push(
|
|
simple_add_validator_to_pool(
|
|
&mut banks_client,
|
|
&payer,
|
|
&recent_blockhash,
|
|
&stake_pool_accounts,
|
|
)
|
|
.await,
|
|
);
|
|
}
|
|
|
|
// Add stake extra funds
|
|
const EXTRA_STAKE: u64 = 1_000_000;
|
|
|
|
for stake_account in stake_accounts {
|
|
transfer(
|
|
&mut banks_client,
|
|
&payer,
|
|
&recent_blockhash,
|
|
&stake_account.stake_account,
|
|
EXTRA_STAKE,
|
|
)
|
|
.await;
|
|
}
|
|
|
|
let rent = banks_client.get_rent().await.unwrap();
|
|
let stake_rent = rent.minimum_balance(std::mem::size_of::<stake::StakeState>())
|
|
+ native_token::sol_to_lamports(1.0);
|
|
|
|
// Check current balance in the list
|
|
assert_eq!(
|
|
get_list_sum(
|
|
&mut banks_client,
|
|
&stake_pool_accounts.validator_list.pubkey()
|
|
)
|
|
.await,
|
|
STAKE_ACCOUNTS * (stake_rent + TEST_STAKE_AMOUNT)
|
|
);
|
|
|
|
// TODO: Execute update list with updated clock
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_update_validator_list_balance_with_uninitialized_validator_list() {} // TODO
|
|
|
|
#[tokio::test]
|
|
async fn test_update_validator_list_balance_with_wrong_stake_state() {} // TODO
|