139 lines
3.8 KiB
Rust
139 lines
3.8 KiB
Rust
#![allow(clippy::integer_arithmetic)]
|
|
use {
|
|
serial_test::serial,
|
|
solana_bench_tps::{
|
|
bench::{do_bench_tps, generate_and_fund_keypairs},
|
|
cli::Config,
|
|
},
|
|
solana_client::{
|
|
rpc_client::RpcClient,
|
|
thin_client::create_client,
|
|
tpu_client::{TpuClient, TpuClientConfig},
|
|
},
|
|
solana_core::validator::ValidatorConfig,
|
|
solana_faucet::faucet::run_local_faucet,
|
|
solana_local_cluster::{
|
|
local_cluster::{ClusterConfig, LocalCluster},
|
|
validator_configs::make_identical_validator_configs,
|
|
},
|
|
solana_rpc::rpc::JsonRpcConfig,
|
|
solana_sdk::{
|
|
commitment_config::CommitmentConfig,
|
|
signature::{Keypair, Signer},
|
|
},
|
|
solana_streamer::socket::SocketAddrSpace,
|
|
solana_test_validator::TestValidator,
|
|
std::{sync::Arc, time::Duration},
|
|
};
|
|
|
|
fn test_bench_tps_local_cluster(config: Config) {
|
|
let native_instruction_processors = vec![];
|
|
|
|
solana_logger::setup();
|
|
|
|
let faucet_keypair = Keypair::new();
|
|
let faucet_pubkey = faucet_keypair.pubkey();
|
|
let faucet_addr = run_local_faucet(faucet_keypair, None);
|
|
|
|
const NUM_NODES: usize = 1;
|
|
let cluster = LocalCluster::new(
|
|
&mut ClusterConfig {
|
|
node_stakes: vec![999_990; NUM_NODES],
|
|
cluster_lamports: 200_000_000,
|
|
validator_configs: make_identical_validator_configs(
|
|
&ValidatorConfig {
|
|
rpc_config: JsonRpcConfig {
|
|
faucet_addr: Some(faucet_addr),
|
|
..JsonRpcConfig::default_for_test()
|
|
},
|
|
..ValidatorConfig::default_for_test()
|
|
},
|
|
NUM_NODES,
|
|
),
|
|
native_instruction_processors,
|
|
..ClusterConfig::default()
|
|
},
|
|
SocketAddrSpace::Unspecified,
|
|
);
|
|
|
|
cluster.transfer(&cluster.funding_keypair, &faucet_pubkey, 100_000_000);
|
|
|
|
let client = Arc::new(create_client(
|
|
cluster.entry_point_info.rpc,
|
|
cluster.entry_point_info.tpu,
|
|
));
|
|
|
|
let lamports_per_account = 100;
|
|
|
|
let keypair_count = config.tx_count * config.keypair_multiplier;
|
|
let keypairs = generate_and_fund_keypairs(
|
|
client.clone(),
|
|
&config.id,
|
|
keypair_count,
|
|
lamports_per_account,
|
|
)
|
|
.unwrap();
|
|
|
|
let _total = do_bench_tps(client, config, keypairs);
|
|
|
|
#[cfg(not(debug_assertions))]
|
|
assert!(_total > 100);
|
|
}
|
|
|
|
fn test_bench_tps_test_validator(config: Config) {
|
|
solana_logger::setup();
|
|
|
|
let mint_keypair = Keypair::new();
|
|
let mint_pubkey = mint_keypair.pubkey();
|
|
|
|
let faucet_addr = run_local_faucet(mint_keypair, None);
|
|
|
|
let test_validator =
|
|
TestValidator::with_no_fees(mint_pubkey, Some(faucet_addr), SocketAddrSpace::Unspecified);
|
|
|
|
let rpc_client = Arc::new(RpcClient::new_with_commitment(
|
|
test_validator.rpc_url(),
|
|
CommitmentConfig::processed(),
|
|
));
|
|
let websocket_url = test_validator.rpc_pubsub_url();
|
|
|
|
let client =
|
|
Arc::new(TpuClient::new(rpc_client, &websocket_url, TpuClientConfig::default()).unwrap());
|
|
|
|
let lamports_per_account = 100;
|
|
|
|
let keypair_count = config.tx_count * config.keypair_multiplier;
|
|
let keypairs = generate_and_fund_keypairs(
|
|
client.clone(),
|
|
&config.id,
|
|
keypair_count,
|
|
lamports_per_account,
|
|
)
|
|
.unwrap();
|
|
|
|
let _total = do_bench_tps(client, config, keypairs);
|
|
|
|
#[cfg(not(debug_assertions))]
|
|
assert!(_total > 100);
|
|
}
|
|
|
|
#[test]
|
|
#[serial]
|
|
fn test_bench_tps_local_cluster_solana() {
|
|
test_bench_tps_local_cluster(Config {
|
|
tx_count: 100,
|
|
duration: Duration::from_secs(10),
|
|
..Config::default()
|
|
});
|
|
}
|
|
|
|
#[test]
|
|
#[serial]
|
|
fn test_bench_tps_tpu_client() {
|
|
test_bench_tps_test_validator(Config {
|
|
tx_count: 100,
|
|
duration: Duration::from_secs(10),
|
|
..Config::default()
|
|
});
|
|
}
|