From 68371bb7ad6e7510b21f5eb81bcb7bf15e98d211 Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Thu, 26 Jun 2025 07:19:44 -0400
Subject: [PATCH] test for same functionality

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

diff --git a/accounts/src/account_core/mod.rs b/accounts/src/account_core/mod.rs
index 079e725..337f476 100644
--- a/accounts/src/account_core/mod.rs
+++ b/accounts/src/account_core/mod.rs
@@ -99,7 +99,7 @@ impl AccountPublicMask {
         viewing_public_key_receiver: AffinePoint,
         data: &[u8],
     ) -> (CipherText, Nonce) {
-        ephemeral_key_holder.encrypt_data(viewing_public_key_receiver, data)
+        Account::encrypt_data(ephemeral_key_holder, viewing_public_key_receiver, data)
     }
 
     pub fn make_tag(&self) -> Tag {
@@ -236,6 +236,17 @@ mod tests {
         assert!(account.key_holder.address != [0u8; 32]); // Check if the address is not empty
     }
 
+    #[test]
+    fn test_new_mask_account() {
+        let account = Account::new();
+
+        let mask = account.make_account_public_mask();
+
+        assert_eq!(mask.balance, 0);
+        assert_eq!(mask.address, account.key_holder.address);
+        assert_eq!(mask.make_tag(), account.make_tag());
+    }
+
     #[test]
     fn test_add_new_utxo_outputs() {
         let mut account = Account::new();