From 5ab4f32e006411dd4b4fc9a421422a4f63bebdc6 Mon Sep 17 00:00:00 2001
From: TanvirDeol <tanvir@interoplabs.io>
Date: Sat, 18 Jan 2025 18:19:52 -0500
Subject: [PATCH] added unit tests for to_string_bytes

---
 packages/axelar-std/tests/test.rs | 61 +++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 packages/axelar-std/tests/test.rs

diff --git a/packages/axelar-std/tests/test.rs b/packages/axelar-std/tests/test.rs
new file mode 100644
index 00000000..d79b746f
--- /dev/null
+++ b/packages/axelar-std/tests/test.rs
@@ -0,0 +1,61 @@
+use soroban_sdk::testutils::Address as _;
+use soroban_sdk::{Address, Env, String};
+use stellar_axelar_std::address::AddressExt;
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    fn address_to_bytes_and_back(env: &Env, address: &Address) -> Address {
+        let address_bytes = address.to_string_bytes(env);
+        Address::from_string_bytes(&address_bytes)
+    }
+
+    #[test]
+    fn generated_address_to_string_bytes() {
+        let env = Env::default();
+        let original_address = Address::generate(&env);
+        let converted_address = address_to_bytes_and_back(&env, &original_address);
+        assert_eq!(original_address, converted_address);
+    }
+
+    #[test]
+    fn zero_address_to_string_bytes() {
+        let env = Env::default();
+        let zero_address = Address::zero(&env);
+        let converted_address = address_to_bytes_and_back(&env, &zero_address);
+        assert_eq!(zero_address, converted_address);
+    }
+
+    #[test]
+    fn account_keys_to_string_bytes() {
+        let env = Env::default();
+        let account_keys: [&str; 3] = [
+            "GC7OHFPWPSWXL4HMN6TXAG54MTZSMJIASWHO6KVRQNHNCXEAHWDSGGC3",
+            "GC65CUPW2IMTJJY6CII7F3OBPVG4YGASEPBBLM4V3LBKX62P6LA24OFV",
+            "GCXQYWGBS5QIXSURFRL3EQIIY556F2TKUYBNWZKEUPNJAHEVIGGPWX3Y",
+        ];
+
+        for key in &account_keys {
+            let original_address = Address::from_string(&String::from_str(&env, key));
+            let converted_address = address_to_bytes_and_back(&env, &original_address);
+            assert_eq!(original_address, converted_address);
+        }
+    }
+
+    #[test]
+    fn contract_keys_to_string_bytes() {
+        let env = Env::default();
+        let contract_keys: [&str; 3] = [
+            "CB6743BTQ2TZHTUCRSUFAH2X5ICOZGI6I3UCQBY3VFSSJ7IERGXUM7TX",
+            "CCNPLLAHDRCYOY2RGUGBYAEWXEPCSQSZZGYYCQWHEC2KBYBWLVKJAU4D",
+            "CD7I2MTBYIQJ6KWR5FLILJLDBJGDV3FTIV24XRIXAEMSR72SRF4AQQCF",
+        ];
+
+        for key in &contract_keys {
+            let original_address = Address::from_string(&String::from_str(&env, key));
+            let converted_address = address_to_bytes_and_back(&env, &original_address);
+            assert_eq!(original_address, converted_address);
+        }
+    }
+}