ledger: Use Cell in add/get UTXO's.

chainwayxyz · Jun 10, 2024 · 5659cea · 5659cea
1 parent 514acbe
commit 5659cea
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 44 deletions.
diff --git a/src/ledger/mod.rs b/src/ledger/mod.rs
@@ -34,6 +34,13 @@ impl Ledger {
 
         self.addresses.set(addresses);
     }
+    /// Returns address of the user.
+    pub fn get_addresses(&self) -> Vec<Address> {
+        let addresses = self.addresses.take();
+        self.addresses.set(addresses.clone());
+
+        addresses
+    }
 }
 
 #[cfg(test)]

diff --git a/src/ledger/transactions.rs b/src/ledger/transactions.rs
@@ -5,54 +5,54 @@ use bitcoin::TxOut;
 
 impl Ledger {
     /// Adds a new UTXO to user's UTXO's.
-    pub fn add_utxo(&self, utxo: TxOut) -> Self {
-        // let mut ledger = self.clone().to_owned();
+    pub fn add_utxo(&self, utxo: TxOut) {
+        let mut utxos = self.utxos.take();
+        utxos.push(utxo);
 
-        // ledger.utxos.push(utxo);
-
-        // ledger
-        todo!()
+        self.utxos.set(utxos);
     }
     /// Returns UTXO's of the user.
     pub fn get_utxos(&self) -> Vec<TxOut> {
-        // self.utxos.clone()
-        todo!()
+        let utxos = self.utxos.take();
+        self.utxos.set(utxos.clone());
+
+        utxos
     }
 }
 
-// #[cfg(test)]
-// mod tests {
-//     use crate::ledger::Ledger;
-//     use bitcoin::{Address, Amount, Network, TxOut, XOnlyPublicKey};
-//     use secp256k1::Secp256k1;
-
-//     #[test]
-//     fn add_utxo() {
-//         let mut ledger = Ledger::new();
-
-//         assert_eq!(ledger.utxos.len(), 0);
-//         assert_eq!(ledger.addresses.take().len(), 0);
-
-//         // Generate a random address.
-//         let secp = Secp256k1::new();
-//         let xonly_public_key = XOnlyPublicKey::from_slice(&[
-//             0x78u8, 0x19u8, 0x90u8, 0xd7u8, 0xe2u8, 0x11u8, 0x8cu8, 0xc3u8, 0x61u8, 0xa9u8, 0x3au8,
-//             0x6fu8, 0xccu8, 0x54u8, 0xceu8, 0x61u8, 0x1du8, 0x6du8, 0xf3u8, 0x81u8, 0x68u8, 0xd6u8,
-//             0xb1u8, 0xedu8, 0xfbu8, 0x55u8, 0x65u8, 0x35u8, 0xf2u8, 0x20u8, 0x0cu8, 0x4b,
-//         ])
-//         .unwrap();
-//         let address = Address::p2tr(&secp, xonly_public_key, None, Network::Regtest);
-//         ledger.add_address(address);
-//         assert_eq!(ledger.addresses.take().len(), 1);
-
-//         // Insert a dummy UTXO.
-//         let utxo = TxOut {
-//             value: Amount::from_sat(0x45),
-//             script_pubkey: ledger.addresses.take()[0].script_pubkey(),
-//         };
-//         ledger = ledger.add_utxo(utxo);
-
-//         assert_eq!(ledger.utxos.len(), 1);
-//         assert_eq!(ledger.utxos[0].value, Amount::from_sat(0x45));
-//     }
-// }
+#[cfg(test)]
+mod tests {
+    use crate::ledger::Ledger;
+    use bitcoin::{Address, Amount, Network, TxOut, XOnlyPublicKey};
+    use secp256k1::Secp256k1;
+
+    #[test]
+    fn add_utxo() {
+        let ledger = Ledger::new();
+
+        assert_eq!(ledger.get_utxos().len(), 0);
+        assert_eq!(ledger.get_addresses().len(), 0);
+
+        // Generate a random address.
+        let secp = Secp256k1::new();
+        let xonly_public_key = XOnlyPublicKey::from_slice(&[
+            0x78u8, 0x19u8, 0x90u8, 0xd7u8, 0xe2u8, 0x11u8, 0x8cu8, 0xc3u8, 0x61u8, 0xa9u8, 0x3au8,
+            0x6fu8, 0xccu8, 0x54u8, 0xceu8, 0x61u8, 0x1du8, 0x6du8, 0xf3u8, 0x81u8, 0x68u8, 0xd6u8,
+            0xb1u8, 0xedu8, 0xfbu8, 0x55u8, 0x65u8, 0x35u8, 0xf2u8, 0x20u8, 0x0cu8, 0x4b,
+        ])
+        .unwrap();
+        let address = Address::p2tr(&secp, xonly_public_key, None, Network::Regtest);
+        ledger.add_address(address);
+        assert_eq!(ledger.get_addresses().len(), 1);
+
+        // Insert a dummy UTXO.
+        let utxo = TxOut {
+            value: Amount::from_sat(0x45),
+            script_pubkey: ledger.get_addresses()[0].script_pubkey(),
+        };
+        ledger.add_utxo(utxo);
+
+        assert_eq!(ledger.get_utxos().len(), 1);
+        assert_eq!(ledger.get_utxos()[0].value, Amount::from_sat(0x45));
+    }
+}