solana/bench-exchange/tests/bench_exchange.rs

use {
    log::*,
    solana_bench_exchange::bench::{airdrop_lamports, do_bench_exchange, Config},
    solana_core::validator::ValidatorConfig,
    solana_exchange_program::{
        exchange_processor::process_instruction, id, solana_exchange_program,
    },
    solana_faucet::faucet::run_local_faucet_with_port,
    solana_gossip::gossip_service::{discover_cluster, get_multi_client},
    solana_local_cluster::{
        local_cluster::{ClusterConfig, LocalCluster},
        validator_configs::make_identical_validator_configs,
    },
    solana_runtime::{bank::Bank, bank_client::BankClient},
    solana_sdk::{
        genesis_config::create_genesis_config,
        signature::{Keypair, Signer},
    },
    solana_streamer::socket::SocketAddrSpace,
    std::{process::exit, sync::mpsc::channel, time::Duration},
};

#[test]
#[ignore]
fn test_exchange_local_cluster() {
    solana_logger::setup();

    const NUM_NODES: usize = 1;

    let config = Config {
        identity: Keypair::new(),
        duration: Duration::from_secs(1),
        fund_amount: 100_000,
        threads: 1,
        transfer_delay: 20, // 15
        batch_size: 100,    // 1000
        chunk_size: 10,     // 200
        account_groups: 1,  // 10
        ..Config::default()
    };
    let Config {
        fund_amount,
        batch_size,
        account_groups,
        ..
    } = config;
    let accounts_in_groups = batch_size * account_groups;

    let cluster = LocalCluster::new(
        &mut ClusterConfig {
            node_stakes: vec![100_000; NUM_NODES],
            cluster_lamports: 100_000_000_000_000,
            validator_configs: make_identical_validator_configs(
                &ValidatorConfig::default_for_test(),
                NUM_NODES,
            ),
            native_instruction_processors: [solana_exchange_program!()].to_vec(),
            ..ClusterConfig::default()
        },
        SocketAddrSpace::Unspecified,
    );

    let faucet_keypair = Keypair::new();
    cluster.transfer(
        &cluster.funding_keypair,
        &faucet_keypair.pubkey(),
        2_000_000_000_000,
    );

    let (addr_sender, addr_receiver) = channel();
    run_local_faucet_with_port(faucet_keypair, addr_sender, Some(1_000_000_000_000), 0);
    let faucet_addr = addr_receiver
        .recv_timeout(Duration::from_secs(2))
        .expect("run_local_faucet")
        .expect("faucet_addr");

    info!("Connecting to the cluster");
    let nodes = discover_cluster(
        &cluster.entry_point_info.gossip,
        NUM_NODES,
        SocketAddrSpace::Unspecified,
    )
    .unwrap_or_else(|err| {
        error!("Failed to discover {} nodes: {:?}", NUM_NODES, err);
        exit(1);
    });

    let (client, num_clients) = get_multi_client(&nodes, &SocketAddrSpace::Unspecified);

    info!("clients: {}", num_clients);
    assert!(num_clients >= NUM_NODES);

    const NUM_SIGNERS: u64 = 2;
    airdrop_lamports(
        &client,
        &faucet_addr,
        &config.identity,
        fund_amount * (accounts_in_groups + 1) as u64 * NUM_SIGNERS,
    );

    do_bench_exchange(vec![client], config);
}

#[test]
fn test_exchange_bank_client() {
    solana_logger::setup();
    let (genesis_config, identity) = create_genesis_config(100_000_000_000_000);
    let mut bank = Bank::new(&genesis_config);
    bank.add_builtin("exchange_program", id(), process_instruction);
    let clients = vec![BankClient::new(bank)];

    do_bench_exchange(
        clients,
        Config {
            identity,
            duration: Duration::from_secs(1),
            fund_amount: 100_000,
            threads: 1,
            transfer_delay: 20, // 0;
            batch_size: 100,    // 1500;
            chunk_size: 10,     // 1500;
            account_groups: 1,  // 50;
            ..Config::default()
        },
    );
}