From e2e8f996330e594d2e4af43c0696d01ff715c37c Mon Sep 17 00:00:00 2001
From: Perelyn <64838956+Perelyn-sama@users.noreply.github.com>
Date: Mon, 3 Nov 2025 03:15:52 +0100
Subject: [PATCH] add rust test to pda example

---
 Cargo.lock                                    |  7 ++
 .../native/program/Cargo.toml                 | 10 +-
 .../native/program/tests/test.rs              | 98 +++++++++++++++++++
 3 files changed, 114 insertions(+), 1 deletion(-)
 create mode 100644 basics/program-derived-addresses/native/program/tests/test.rs

diff --git a/Cargo.lock b/Cargo.lock
index 1275e977..3b1c44f0 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -2184,8 +2184,15 @@ version = "0.1.0"
 dependencies = [
  "borsh 1.5.7",
  "borsh-derive 1.5.7",
+ "litesvm",
+ "solana-instruction 3.0.0",
+ "solana-keypair",
+ "solana-native-token 3.0.0",
  "solana-program 3.0.0",
+ "solana-pubkey 3.0.0",
+ "solana-rent 3.0.0",
  "solana-system-interface 2.0.0",
+ "solana-transaction",
 ]
 
 [[package]]
diff --git a/basics/program-derived-addresses/native/program/Cargo.toml b/basics/program-derived-addresses/native/program/Cargo.toml
index 9cae908b..41e42c0a 100644
--- a/basics/program-derived-addresses/native/program/Cargo.toml
+++ b/basics/program-derived-addresses/native/program/Cargo.toml
@@ -13,9 +13,17 @@ solana-system-interface.workspace = true
 crate-type = ["cdylib", "lib"]
 
 [features]
-anchor-debug = []
 custom-heap = []
 custom-panic = []
 
 [lints.rust]
 unexpected_cfgs = { level = "warn", check-cfg = ['cfg(target_os, values("solana"))'] }
+
+[dev-dependencies]
+litesvm = "0.8.1"
+solana-instruction = "3.0.0"
+solana-keypair = "3.0.1"
+solana-pubkey = "3.0.0"
+solana-transaction = "3.0.1"
+solana-native-token = "3.0.0"
+solana-rent = "3.0.0"
diff --git a/basics/program-derived-addresses/native/program/tests/test.rs b/basics/program-derived-addresses/native/program/tests/test.rs
new file mode 100644
index 00000000..42d75203
--- /dev/null
+++ b/basics/program-derived-addresses/native/program/tests/test.rs
@@ -0,0 +1,98 @@
+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!("../../../../../target/deploy/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);
+}