//! Feature Gate program compute unit benchmark testing.

use {
    mollusk_svm::{program::keyed_account_for_system_program, Mollusk},
    mollusk_svm_bencher::{Bench, MolluskComputeUnitBencher},
    solana_feature_gate_program::instruction::revoke_pending_activation,
    solana_sdk::{account::AccountSharedData, feature::Feature, incinerator, pubkey::Pubkey},
};

fn main() {
    std::env::set_var("SBF_OUT_DIR", "../target/deploy");
    let mollusk = Mollusk::new(&solana_sdk::feature::id(), "solana_feature_gate_program");

    let feature = Pubkey::new_unique();

    let bench: Bench = (
        "revoke_pending_activation",
        &revoke_pending_activation(&feature),
        &[
            (
                feature,
                AccountSharedData::new_data(
                    42,
                    &Feature { activated_at: None },
                    &solana_sdk::feature::id(),
                )
                .unwrap(),
            ),
            (incinerator::id(), AccountSharedData::default()),
            keyed_account_for_system_program(),
        ],
    );

    MolluskComputeUnitBencher::new(mollusk)
        .bench(bench)
        .must_pass(true)
        .out_dir("./benches")
        .execute();
}