From 1dcc6c7b372c078469fe4318d62506c90bac70b4 Mon Sep 17 00:00:00 2001
From: Ebina Perelyn Okoh <okohebina@gmail.com>
Date: Mon, 27 Oct 2025 20:18:48 +0100
Subject: [PATCH] add litesvm test for counter example

---
 Cargo.lock                                  |  8 +++
 basics/counter/native/program/Cargo.toml    | 10 ++++
 basics/counter/native/program/src/lib.rs    |  2 +-
 basics/counter/native/program/tests/test.rs | 63 +++++++++++++++++++++
 4 files changed, 82 insertions(+), 1 deletion(-)
 create mode 100644 basics/counter/native/program/tests/test.rs

diff --git a/Cargo.lock b/Cargo.lock
index f4696c4ad..1d6db5be2 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -947,7 +947,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/counter/native/program/Cargo.toml b/basics/counter/native/program/Cargo.toml
index ca1c19f0b..5bb4e4865 100644
--- a/basics/counter/native/program/Cargo.toml
+++ b/basics/counter/native/program/Cargo.toml
@@ -21,3 +21,13 @@ solana-program.workspace = true
 
 [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-native-token = "3.0.0"
+solana-pubkey = "3.0.0"
+solana-transaction = "3.0.1"
+solana-rent = "3.0.0"
+solana-system-interface.workspace = true
diff --git a/basics/counter/native/program/src/lib.rs b/basics/counter/native/program/src/lib.rs
index f1b557354..f8bded9dc 100644
--- a/basics/counter/native/program/src/lib.rs
+++ b/basics/counter/native/program/src/lib.rs
@@ -9,7 +9,7 @@ use solana_program::{
 };
 
 mod state;
-use state::*;
+pub use state::*;
 
 declare_id!("Fg6PaFpoGXkYsidMpWTK6W2BeZ7FEfcYkg476zPFsLnS");
 
diff --git a/basics/counter/native/program/tests/test.rs b/basics/counter/native/program/tests/test.rs
new file mode 100644
index 000000000..8c0d348db
--- /dev/null
+++ b/basics/counter/native/program/tests/test.rs
@@ -0,0 +1,63 @@
+use borsh::BorshDeserialize;
+use counter_solana_native::Counter;
+use litesvm::LiteSVM;
+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_counter() {
+    let program_id = Pubkey::new_unique();
+    let program_bytes = include_bytes!("../../../../../target/deploy/counter_solana_native.so");
+
+    let mut svm = LiteSVM::new();
+    svm.add_program(program_id, program_bytes).unwrap();
+
+    let payer = Keypair::new();
+    let counter_account = Keypair::new();
+
+    svm.airdrop(&payer.pubkey(), LAMPORTS_PER_SOL * 10).unwrap();
+
+    let counter_account_size = std::mem::size_of::<Counter>();
+
+    let create_ix = create_account(
+        &payer.pubkey(),
+        &counter_account.pubkey(),
+        Rent::default().minimum_balance(counter_account_size),
+        counter_account_size as u64,
+        &program_id,
+    );
+
+    let tx = Transaction::new_signed_with_payer(
+        &[create_ix],
+        Some(&payer.pubkey()),
+        &[&payer, &counter_account],
+        svm.latest_blockhash(),
+    );
+    svm.send_transaction(tx).unwrap();
+
+    let ix = Instruction {
+        program_id,
+        accounts: vec![AccountMeta::new(counter_account.pubkey(), false)],
+        data: vec![0],
+    };
+
+    let tx = Transaction::new_signed_with_payer(
+        &[ix],
+        Some(&payer.pubkey()),
+        &[payer],
+        svm.latest_blockhash(),
+    );
+
+    dbg!(&[0].split_at(1));
+
+    let _ = svm.send_transaction(tx).is_ok();
+
+    let counter_account_data = svm.get_account(&counter_account.pubkey()).unwrap().data;
+    let couneter = Counter::try_from_slice(&counter_account_data).unwrap();
+    assert_eq!(couneter.count, 1);
+}