anza-xyz-agave/cli/tests/address_lookup_table.rs

use {
    solana_cli::{
        address_lookup_table::{
            AddressLookupTableCliCommand, DEACTIVATE_LOOKUP_TABLE_WARNING,
            FREEZE_LOOKUP_TABLE_WARNING,
        },
        cli::{process_command, CliCommand, CliConfig},
    },
    solana_cli_output::{CliAddressLookupTable, CliAddressLookupTableCreated, OutputFormat},
    solana_faucet::faucet::run_local_faucet_with_unique_port_for_tests,
    solana_keypair::Keypair,
    solana_native_token::LAMPORTS_PER_SOL,
    solana_net_utils::SocketAddrSpace,
    solana_pubkey::Pubkey,
    solana_signer::Signer,
    solana_test_validator::TestValidator,
    std::str::FromStr,
};

#[tokio::test(flavor = "multi_thread", worker_threads = 1)]
async fn test_cli_create_extend_and_freeze_address_lookup_table() {
    let mint_keypair = Keypair::new();
    let faucet_addr = run_local_faucet_with_unique_port_for_tests(mint_keypair.insecure_clone());
    let test_validator = TestValidator::async_with_no_fees(
        &mint_keypair,
        Some(faucet_addr),
        SocketAddrSpace::Unspecified,
    )
    .await;

    let mut config = CliConfig::recent_for_tests();
    let keypair = Keypair::new();
    config.json_rpc_url = test_validator.rpc_url();
    config.signers = vec![&keypair];
    config.output_format = OutputFormat::JsonCompact;

    // Airdrop SOL for transaction fees
    config.command = CliCommand::Airdrop {
        pubkey: None,
        lamports: 10 * LAMPORTS_PER_SOL,
    };
    process_command(&config).await.unwrap();

    // Create lookup table
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::CreateLookupTable {
            authority_pubkey: keypair.pubkey(),
            payer_signer_index: 0,
        });
    let response: CliAddressLookupTableCreated =
        serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
    let lookup_table_pubkey = Pubkey::from_str(&response.lookup_table_address).unwrap();

    // Validate created lookup table
    {
        config.command =
            CliCommand::AddressLookupTable(AddressLookupTableCliCommand::ShowLookupTable {
                lookup_table_pubkey,
            });
        let response: CliAddressLookupTable =
            serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
        assert_eq!(
            response,
            CliAddressLookupTable {
                lookup_table_address: lookup_table_pubkey.to_string(),
                authority: Some(keypair.pubkey().to_string()),
                deactivation_slot: u64::MAX,
                last_extended_slot: 0,
                addresses: vec![],
            }
        );
    }

    // Extend lookup table
    let new_addresses: Vec<Pubkey> = (0..5).map(|_| Pubkey::new_unique()).collect();
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::ExtendLookupTable {
            lookup_table_pubkey,
            authority_signer_index: 0,
            payer_signer_index: 0,
            new_addresses: new_addresses.clone(),
        });
    process_command(&config).await.unwrap();

    // Validate extended lookup table
    {
        config.command =
            CliCommand::AddressLookupTable(AddressLookupTableCliCommand::ShowLookupTable {
                lookup_table_pubkey,
            });
        let CliAddressLookupTable {
            addresses,
            last_extended_slot,
            ..
        } = serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
        assert_eq!(
            addresses
                .into_iter()
                .map(|address| Pubkey::from_str(&address).unwrap())
                .collect::<Vec<Pubkey>>(),
            new_addresses
        );
        assert!(last_extended_slot > 0);
    }

    // Freeze lookup table w/o bypass
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::FreezeLookupTable {
            lookup_table_pubkey,
            authority_signer_index: 0,
            bypass_warning: false,
        });
    let process_err = process_command(&config).await.unwrap_err();
    assert_eq!(process_err.to_string(), FREEZE_LOOKUP_TABLE_WARNING);

    // Freeze lookup table w/ bypass
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::FreezeLookupTable {
            lookup_table_pubkey,
            authority_signer_index: 0,
            bypass_warning: true,
        });
    process_command(&config).await.unwrap();

    // Validate frozen lookup table
    {
        config.command =
            CliCommand::AddressLookupTable(AddressLookupTableCliCommand::ShowLookupTable {
                lookup_table_pubkey,
            });
        let CliAddressLookupTable { authority, .. } =
            serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
        assert!(authority.is_none());
    }
}

#[tokio::test(flavor = "multi_thread", worker_threads = 1)]
async fn test_cli_create_and_deactivate_address_lookup_table() {
    let mint_keypair = Keypair::new();
    let faucet_addr = run_local_faucet_with_unique_port_for_tests(mint_keypair.insecure_clone());
    let test_validator = TestValidator::async_with_no_fees(
        &mint_keypair,
        Some(faucet_addr),
        SocketAddrSpace::Unspecified,
    )
    .await;

    let mut config = CliConfig::recent_for_tests();
    let keypair = Keypair::new();
    config.json_rpc_url = test_validator.rpc_url();
    config.signers = vec![&keypair];
    config.output_format = OutputFormat::JsonCompact;

    // Airdrop SOL for transaction fees
    config.command = CliCommand::Airdrop {
        pubkey: None,
        lamports: 10 * LAMPORTS_PER_SOL,
    };
    process_command(&config).await.unwrap();

    // Create lookup table
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::CreateLookupTable {
            authority_pubkey: keypair.pubkey(),
            payer_signer_index: 0,
        });
    let response: CliAddressLookupTableCreated =
        serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
    let lookup_table_pubkey = Pubkey::from_str(&response.lookup_table_address).unwrap();

    // Validate created lookup table
    {
        config.command =
            CliCommand::AddressLookupTable(AddressLookupTableCliCommand::ShowLookupTable {
                lookup_table_pubkey,
            });
        let response: CliAddressLookupTable =
            serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
        assert_eq!(
            response,
            CliAddressLookupTable {
                lookup_table_address: lookup_table_pubkey.to_string(),
                authority: Some(keypair.pubkey().to_string()),
                deactivation_slot: u64::MAX,
                last_extended_slot: 0,
                addresses: vec![],
            }
        );
    }

    // Deactivate lookup table w/o bypass
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::DeactivateLookupTable {
            lookup_table_pubkey,
            authority_signer_index: 0,
            bypass_warning: false,
        });
    let process_err = process_command(&config).await.unwrap_err();
    assert_eq!(process_err.to_string(), DEACTIVATE_LOOKUP_TABLE_WARNING);

    // Deactivate lookup table w/ bypass
    config.command =
        CliCommand::AddressLookupTable(AddressLookupTableCliCommand::DeactivateLookupTable {
            lookup_table_pubkey,
            authority_signer_index: 0,
            bypass_warning: true,
        });
    process_command(&config).await.unwrap();

    // Validate deactivated lookup table
    {
        config.command =
            CliCommand::AddressLookupTable(AddressLookupTableCliCommand::ShowLookupTable {
                lookup_table_pubkey,
            });
        let CliAddressLookupTable {
            deactivation_slot, ..
        } = serde_json::from_str(&process_command(&config).await.unwrap()).unwrap();
        assert_ne!(deactivation_slot, u64::MAX);
    }
}