add test_create_empty_store, test_register_account and test_unregister_account

node_core/src/storage/accounts_store.rs

@@ -26,3 +26,56 @@ impl Default for NodeAccountsStore {
         Self::new()
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use accounts::account_core::Account;
+    /// Helper function to create a sample account
+    fn create_sample_account(balance: u64) -> Account {
+        Account::new_with_balance(balance)
+    }
+
+    fn pad_to_32(slice: &[u8]) -> [u8; 32] {
+        let mut padded = [0u8; 32];
+        let len = slice.len().min(32); 
+        padded[..len].copy_from_slice(&slice[..len]);
+        padded
+    }
+
+
+    #[test]
+    fn test_create_empty_store() {
+        let store = NodeAccountsStore::new();
+        assert!(store.accounts.is_empty());
+    }
+
+    #[test]
+    fn test_register_account() {
+        let mut store = NodeAccountsStore::new();
+
+        let account = create_sample_account(100);
+        let account_addr = account.address.clone();
+
+        store.register_account(account);
+
+        assert_eq!(store.accounts.len(), 1);
+        let stored_account = store.accounts.get(&account_addr).unwrap();
+        assert_eq!(stored_account.balance, 100);
+    }
+
+    #[test]
+    fn test_unregister_account() {
+        let mut store = NodeAccountsStore::new();
+
+        let account = create_sample_account(100);
+        let account_addr = account.address.clone();
+        store.register_account(account);
+
+        assert_eq!(store.accounts.len(), 1);
+
+        store.unregister_account(account_addr);
+        assert!(store.accounts.is_empty());
+    }
+
+}