solana-program-examples/basics/program-derived-addresses/native/program/tests/test.rs

use borsh::BorshDeserialize;
use litesvm::LiteSVM;
use program_derived_addresses_native_program::state::{IncrementPageVisits, PageVisits};
use solana_instruction::{AccountMeta, Instruction};
use solana_keypair::{Keypair, Signer};
use solana_native_token::LAMPORTS_PER_SOL;
use solana_pubkey::Pubkey;
use solana_rent::Rent;
use solana_system_interface::instruction::create_account;
use solana_transaction::Transaction;

#[test]
fn test_pda() {
    let mut svm = LiteSVM::new();

    let program_id = Pubkey::new_unique();
    let program_bytes =
        include_bytes!("../../tests/fixtures/program_derived_addresses_native_program.so");
    svm.add_program(program_id, program_bytes).unwrap();

    let payer = Keypair::new();
    svm.airdrop(&payer.pubkey(), LAMPORTS_PER_SOL * 10).unwrap();

    let test_user = Keypair::new();

    let rent = Rent::default();

    let create_ix = create_account(
        &payer.pubkey(),
        &test_user.pubkey(),
        solana_rent::Rent::minimum_balance(&rent, 0),
        0,
        &solana_system_interface::program::ID,
    );

    let tx = Transaction::new_signed_with_payer(
        &[create_ix],
        Some(&payer.pubkey()),
        &[&payer, &test_user],
        svm.latest_blockhash(),
    );

    let _ = svm.send_transaction(tx).is_ok();

    let (pda, bump) =
        Pubkey::find_program_address(&[b"page_visits", test_user.pubkey().as_ref()], &program_id);

    let data = borsh::to_vec(&PageVisits {
        page_visits: 0,
        bump,
    })
    .unwrap();

    let ix = Instruction {
        program_id,
        accounts: vec![
            AccountMeta::new(pda, false),
            AccountMeta::new(test_user.pubkey(), false),
            AccountMeta::new(payer.pubkey(), true),
            AccountMeta::new(solana_system_interface::program::ID, false),
        ],
        data,
    };

    let tx = Transaction::new_signed_with_payer(
        &[ix],
        Some(&payer.pubkey()),
        &[&payer],
        svm.latest_blockhash(),
    );

    let _ = svm.send_transaction(tx).is_ok();

    let data = borsh::to_vec(&IncrementPageVisits {}).unwrap();

    let ix = Instruction {
        program_id,
        accounts: vec![
            AccountMeta::new(pda, false),
            AccountMeta::new(payer.pubkey(), true),
        ],
        data,
    };

    let tx = Transaction::new_signed_with_payer(
        &[ix],
        Some(&payer.pubkey()),
        &[&payer],
        svm.latest_blockhash(),
    );

    let _ = svm.send_transaction(tx).is_ok();

    // read page visits
    let account_info = svm.get_account(&pda).unwrap();
    let read_page_visits = PageVisits::try_from_slice(&account_info.data).unwrap();
    assert_eq!(read_page_visits.page_visits, 1);
}