From a4f9a0094ec754443201e6daf2b0a43e3607181c Mon Sep 17 00:00:00 2001
From: BitcoinZavior <BitcoinZavior@GMail.Com>
Date: Tue, 5 Nov 2024 14:32:44 -0500
Subject: [PATCH] test: add test for tx_builder::policy_path method

---
 bdk-ffi/src/tx_builder.rs | 89 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)

diff --git a/bdk-ffi/src/tx_builder.rs b/bdk-ffi/src/tx_builder.rs
index e71cf325..da5f9628 100644
--- a/bdk-ffi/src/tx_builder.rs
+++ b/bdk-ffi/src/tx_builder.rs
@@ -283,3 +283,92 @@ impl BumpFeeTxBuilder {
         Ok(Arc::new(psbt.into()))
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use std::sync::Arc;
+
+    use bitcoin_ffi::Network;
+
+    use crate::{
+        descriptor::Descriptor, esplora::EsploraClient, store::Connection,
+        types::FullScanScriptInspector, wallet::Wallet,
+    };
+
+    struct FullScanInspector;
+    impl FullScanScriptInspector for FullScanInspector {
+        fn inspect(&self, _: bdk_wallet::KeychainKind, _: u32, _: Arc<bitcoin_ffi::Script>) {}
+    }
+
+    #[test]
+    fn test_policy_path() {
+        let wallet = create_and_sync_wallet();
+        let address = wallet
+            .next_unused_address(bdk_wallet::KeychainKind::External)
+            .address;
+        println!("Wallet address: {:?}", address);
+
+        let ext_policy = wallet.policies(bdk_wallet::KeychainKind::External);
+        let int_policy = wallet.policies(bdk_wallet::KeychainKind::Internal);
+
+        if let (Ok(Some(ext_policy)), Ok(Some(int_policy))) = (ext_policy, int_policy) {
+            let ext_path = vec![(ext_policy.id().clone(), vec![0, 1])]
+                .into_iter()
+                .collect();
+            println!("External Policy path : {:?}\n", ext_path);
+            let int_path = vec![(int_policy.id().clone(), vec![0, 1])]
+                .into_iter()
+                .collect();
+            println!("Internal Policy Path: {:?}\n", int_path);
+
+            match crate::tx_builder::TxBuilder::new()
+                .add_recipient(
+                    &(*address.script_pubkey()).to_owned(),
+                    Arc::new(bitcoin_ffi::Amount::from_sat(1000)),
+                )
+                .do_not_spend_change()
+                .policy_path(int_path, bdk_wallet::KeychainKind::Internal)
+                .policy_path(ext_path, bdk_wallet::KeychainKind::External)
+                .finish(&Arc::new(wallet))
+            {
+                Ok(tx) => println!("Transaction serialized: {}\n", tx.serialize()),
+                Err(e) => println!("Error: {}", e.to_string()),
+            }
+        } else {
+            println!("Failed to retrieve valid policies for keychains.");
+        }
+    }
+
+    fn create_and_sync_wallet() -> Wallet {
+        let external_descriptor = format!(
+            "wsh(thresh(2,pk({}/0/*),sj:and_v(v:pk({}/0/*),n:older(6)),snj:and_v(v:pk({}/0/*),after(630000))))",
+            "tpubD6NzVbkrYhZ4XJBfEJ6gt9DiVdfWJijsQTCE3jtXByW3Tk6AVGQ3vL1NNxg3SjB7QkJAuutACCQjrXD8zdZSM1ZmBENszCqy49ECEHmD6rf",
+            "tpubD6NzVbkrYhZ4YfAr3jCBRk4SpqB9L1Hh442y83njwfMaker7EqZd7fHMqyTWrfRYJ1e5t2ue6BYjW5i5yQnmwqbzY1a3kfqNxog1AFcD1aE",
+            "tprv8ZgxMBicQKsPeitVUz3s6cfyCECovNP7t82FaKPa4UKqV1kssWcXgLkMDjzDbgG9GWoza4pL7z727QitfzkiwX99E1Has3T3a1MKHvYWmQZ"
+        );
+        let internal_descriptor = format!(
+            "wsh(thresh(2,pk({}/1/*),sj:and_v(v:pk({}/1/*),n:older(6)),snj:and_v(v:pk({}/1/*),after(630000))))",
+            "tpubD6NzVbkrYhZ4XJBfEJ6gt9DiVdfWJijsQTCE3jtXByW3Tk6AVGQ3vL1NNxg3SjB7QkJAuutACCQjrXD8zdZSM1ZmBENszCqy49ECEHmD6rf",
+            "tpubD6NzVbkrYhZ4YfAr3jCBRk4SpqB9L1Hh442y83njwfMaker7EqZd7fHMqyTWrfRYJ1e5t2ue6BYjW5i5yQnmwqbzY1a3kfqNxog1AFcD1aE",
+            "tprv8ZgxMBicQKsPeitVUz3s6cfyCECovNP7t82FaKPa4UKqV1kssWcXgLkMDjzDbgG9GWoza4pL7z727QitfzkiwX99E1Has3T3a1MKHvYWmQZ"
+        );
+        let wallet = Wallet::new(
+            Arc::new(Descriptor::new(external_descriptor, Network::Signet).unwrap()),
+            Arc::new(Descriptor::new(internal_descriptor, Network::Signet).unwrap()),
+            Network::Signet,
+            Arc::new(Connection::new_in_memory().unwrap()),
+        )
+        .unwrap();
+        let client = EsploraClient::new("https://mutinynet.com/api/".to_string());
+        let full_scan_builder = wallet.start_full_scan();
+        let full_scan_request = full_scan_builder
+            .inspect_spks_for_all_keychains(Arc::new(FullScanInspector))
+            .unwrap()
+            .build()
+            .unwrap();
+        let update = client.full_scan(full_scan_request, 10, 10).unwrap();
+        wallet.apply_update(update).unwrap();
+        println!("Wallet balance: {:?}", wallet.balance().total.to_sat());
+        wallet
+    }
+}