Added tests for ExtendedPrivKey & ExtendedPubKey

Author: dhruv-1001

src/lib.rs

@@ -556,12 +556,12 @@ uniffi::deps::static_assertions::assert_impl_all!(Wallet: Sync, Send);
 // crate.
 #[cfg(test)]
 mod test {
-    use crate::{TxBuilder, Wallet};
-    use bdk::bitcoin::Address;
+    use crate::{DerivationPath, ExtendedPrivKey, TxBuilder, Wallet};
     use bdk::bitcoin::Network::Testnet;
+    use bdk::bitcoin::{Address, Network};
     use bdk::wallet::get_funded_wallet;
     use std::str::FromStr;
-    use std::sync::Mutex;
+    use std::sync::{Arc, Mutex};
 
     #[test]
     fn test_drain_wallet() {
@@ -616,4 +616,38 @@ mod test {
         let output_value = psbt.unsigned_tx.output.get(0).cloned().unwrap().value;
         assert_eq!(output_value, 49_890 as u64); // input - fee
     }
+
+    #[test]
+    fn test_generate_extended_priv_key() {
+        let mnemonic =
+            "option hub fatigue fork harbor cement fatal couple kit cloth life toilet".to_string();
+        let extended_priv_key = ExtendedPrivKey::new(Network::Testnet, mnemonic, None).unwrap();
+        assert_eq!(extended_priv_key.to_string(), "tprv8ZgxMBicQKsPf1bXwktHW7GQDBgYBXnpUu5LdYckxpcQcqUEh2YFEad9jYB1TVo41vGRoUSjoQ5MwWonJYqXNMUbXrGPX96yKB8zdzjXKCK".to_string())
+    }
+
+    #[test]
+    fn test_derive_extended_priv_key() {
+        let mnemonic =
+            "option hub fatigue fork harbor cement fatal couple kit cloth life toilet".to_string();
+        let extended_priv_key = ExtendedPrivKey::new(Network::Testnet, mnemonic, None).unwrap();
+        let derivation_path = DerivationPath::new("m/84'/1'/0'/0".to_string()).unwrap();
+        let derived_extended_priv_key =
+            extended_priv_key.derive_xprv(Some(Arc::new(derivation_path)));
+        assert_eq!(derived_extended_priv_key.to_string(), "tprv8hstRa2WtkiWegA8t9VMah1C7BZhG2KkENt7Fr4GT4SpTPKCfX7WWbeihv2ZDkzWoLx2wouKXYsj2eRo54UNGCbVog8QS2kLNvj2b3TmMEx".to_string());
+        let derived_extended_priv_key_no_path = extended_priv_key.derive_xprv(None);
+        assert_eq!(derived_extended_priv_key_no_path.to_string(), "tprv8ZgxMBicQKsPf1bXwktHW7GQDBgYBXnpUu5LdYckxpcQcqUEh2YFEad9jYB1TVo41vGRoUSjoQ5MwWonJYqXNMUbXrGPX96yKB8zdzjXKCK".to_string());
+    }
+
+    #[test]
+    fn test_derive_extended_pub_key() {
+        let mnemonic =
+            "option hub fatigue fork harbor cement fatal couple kit cloth life toilet".to_string();
+        let extended_priv_key = ExtendedPrivKey::new(Network::Testnet, mnemonic, None).unwrap();
+        let derivation_path = DerivationPath::new("m/84'/1'/0'/0".to_string()).unwrap();
+        let derived_extended_pub_key =
+            extended_priv_key.derive_xpub(Some(Arc::new(derivation_path)));
+        assert_eq!(derived_extended_pub_key.to_string(), "tpubDEZvZz4m38QBY9Bvmo9wz6fJgD5dRMWeogUtYN6ZsLFDHsZyHuw6h6Gat43Ltiq1wbeoJLzb3dQpWEnr5uf2gBDjduqsn1ikTCqye3vYcy5");
+        let derived_extended_pub_key_no_path = extended_priv_key.derive_xpub(None);
+        assert_eq!(derived_extended_pub_key_no_path.to_string(), "tpubD6NzVbkrYhZ4YUdKqQYsuWvWnDCULryj4Cg7v4f4P6QoTKj1KRMqR5F1ugxoXDcdaTMGqvsK61dQe8npmn85fmaJ4uEHACGpWsMuEv2upVi".to_string());
+    }
 }