From 63cca0f30f921ccaf3e9e86d4f475ec0de21431c Mon Sep 17 00:00:00 2001
From: Sergio Chouhy <sergio.chouhy@gmail.com>
Date: Wed, 13 Aug 2025 01:55:56 -0300
Subject: [PATCH] add tests

---
 nssa/src/signature/private_key.rs | 10 ++++++++++
 nssa/src/state.rs                 |  9 ++++-----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/nssa/src/signature/private_key.rs b/nssa/src/signature/private_key.rs
index a0a7c0c..b35d7da 100644
--- a/nssa/src/signature/private_key.rs
+++ b/nssa/src/signature/private_key.rs
@@ -24,3 +24,13 @@ impl PrivateKey {
         &self.0
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    #[test]
+    fn test_value_getter() {
+        let key = PrivateKey::try_new([1; 32]).unwrap();
+        assert_eq!(key.value(), &key.0);
+    }
+}
diff --git a/nssa/src/state.rs b/nssa/src/state.rs
index 9bcd0aa..f6d932e 100644
--- a/nssa/src/state.rs
+++ b/nssa/src/state.rs
@@ -73,11 +73,6 @@ impl V01State {
     pub(crate) fn builtin_programs(&self) -> &HashMap<ProgramId, Program> {
         &self.builtin_programs
     }
-
-    #[cfg(test)]
-    pub fn force_insert_account(&mut self, address: Address, account: Account) {
-        self.public_state.insert(address, account);
-    }
 }
 
 #[cfg(test)]
@@ -283,6 +278,10 @@ mod tests {
     }
 
     impl V01State {
+        pub fn force_insert_account(&mut self, address: Address, account: Account) {
+            self.public_state.insert(address, account);
+        }
+
         /// Include test programs in the builtin programs map
         pub fn with_test_programs(mut self) -> Self {
             self.insert_program(Program::nonce_changer_program());