From a57aeaa64ee9c7b68e510b15ccd647c635a4a119 Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Mon, 9 Dec 2024 04:09:27 +0100
Subject: [PATCH] add test_add_new_utxo_outputs, test_update_public_balance,
 test_add_asset

---
 accounts/src/account_core/mod.rs | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/accounts/src/account_core/mod.rs b/accounts/src/account_core/mod.rs
index 05e4a925..5105d4c6 100644
--- a/accounts/src/account_core/mod.rs
+++ b/accounts/src/account_core/mod.rs
@@ -164,4 +164,35 @@ mod tests {
         assert!(account.utxo_tree.store.get(&account.address).is_none());
     }
 
+    #[test]
+    fn test_add_new_utxo_outputs() {
+        let mut account = Account::new();
+        let utxo1 = generate_dummy_utxo(account.address, 100);
+        let utxo2 = generate_dummy_utxo(account.address, 200);
+
+        let result = account.add_new_utxo_outputs(vec![utxo1.clone(), utxo2.clone()]);
+
+        assert!(result.is_ok());
+        assert_eq!(account.utxo_tree.store.len(), 2);
+    }
+
+    #[test]
+    fn test_update_public_balance() {
+        let mut account = Account::new();
+        account.update_public_balance(500);
+
+        assert_eq!(account.balance, 500);
+    }
+
+    #[test]
+    fn test_add_asset() {
+        let mut account = Account::new();
+        let asset = "dummy_asset";
+        let amount = 1000u128;
+
+        let result = account.add_asset(asset, amount, false);
+
+        assert!(result.is_ok());
+        assert_eq!(account.utxo_tree.store.len(), 1);
+    }
 }