chore: use Amount on tests

luisschwab · ValuedMammal · commit 104a301cee06 · 2025-05-28T11:09:46.000-04:00
diff --git a/wallet/src/test_utils.rs b/wallet/src/test_utils.rs
@@ -251,7 +251,7 @@ impl From<ConfirmationBlockTime> for ReceiveTo {
 }
 
 /// Receive a tx output with the given value in the latest block
-pub fn receive_output_in_latest_block(wallet: &mut Wallet, value: u64) -> OutPoint {
+pub fn receive_output_in_latest_block(wallet: &mut Wallet, value: Amount) -> OutPoint {
     let latest_cp = wallet.latest_checkpoint();
     let height = latest_cp.height();
     assert!(height > 0, "cannot receive tx into genesis block");
@@ -268,7 +268,7 @@ pub fn receive_output_in_latest_block(wallet: &mut Wallet, value: u64) -> OutPoi
 /// Receive a tx output with the given value and chain position
 pub fn receive_output(
     wallet: &mut Wallet,
-    value: u64,
+    value: Amount,
     receive_to: impl Into<ReceiveTo>,
 ) -> OutPoint {
     let addr = wallet.next_unused_address(KeychainKind::External).address;
@@ -279,7 +279,7 @@ pub fn receive_output(
 pub fn receive_output_to_address(
     wallet: &mut Wallet,
     addr: Address,
-    value: u64,
+    value: Amount,
     receive_to: impl Into<ReceiveTo>,
 ) -> OutPoint {
     let tx = Transaction {
@@ -288,7 +288,7 @@ pub fn receive_output_to_address(
         input: vec![],
         output: vec![TxOut {
             script_pubkey: addr.script_pubkey(),
-            value: Amount::from_sat(value),
+            value,
         }],
     };
 
diff --git a/wallet/src/wallet/export.rs b/wallet/src/wallet/export.rs
@@ -214,6 +214,7 @@ impl FullyNodedExport {
 #[cfg(test)]
 mod test {
     use alloc::string::ToString;
+    use bitcoin::Amount;
     use core::str::FromStr;
 
     use bdk_chain::BlockId;
@@ -233,7 +234,7 @@ mod test {
             hash: BlockHash::all_zeros(),
         };
         insert_checkpoint(&mut wallet, block);
-        receive_output_in_latest_block(&mut wallet, 10_000);
+        receive_output_in_latest_block(&mut wallet, Amount::from_sat(10_000));
 
         wallet
     }
diff --git a/wallet/src/wallet/tx_builder.rs b/wallet/src/wallet/tx_builder.rs
@@ -1104,7 +1104,7 @@ mod test {
             },
         );
 
-        receive_output_in_latest_block(&mut wallet, Amount::ONE_BTC.to_sat());
+        receive_output_in_latest_block(&mut wallet, Amount::ONE_BTC);
 
         // tx1 sending 15k sat to a recipient
         let recip = ScriptBuf::from_hex(
diff --git a/wallet/tests/wallet.rs b/wallet/tests/wallet.rs
@@ -1576,12 +1576,12 @@ fn test_create_tx_increment_change_index() {
         descriptor: &'static str,
         change_descriptor: Option<&'static str>,
         // amount to send
-        to_send: u64,
+        to_send: Amount,
         // (derivation index, next unused index) of *change keychain*
         expect: (Option<u32>, u32),
     }
     // total wallet funds
-    let amount = 10_000;
+    let amount = Amount::from_sat(10_000);
     let recipient = Address::from_str("bcrt1q3qtze4ys45tgdvguj66zrk4fu6hq3a3v9pfly5")
         .unwrap()
         .assume_checked()
@@ -1592,55 +1592,55 @@ fn test_create_tx_increment_change_index() {
             name: "two wildcard, builder error",
             descriptor: desc,
             change_descriptor: Some(change_desc),
-            to_send: amount + 1,
+            to_send: amount + Amount::from_sat(1),
             // should not use or derive change index
             expect: (None, 0),
         },
         TestCase {
             name: "two wildcard, create change",
             descriptor: desc,
             change_descriptor: Some(change_desc),
-            to_send: 5_000,
+            to_send: Amount::from_sat(5_000),
             // should use change index
             expect: (Some(0), 1),
         },
         TestCase {
             name: "two wildcard, no change",
             descriptor: desc,
             change_descriptor: Some(change_desc),
-            to_send: 9_850,
+            to_send: Amount::from_sat(9_850),
             // should not use change index
             expect: (None, 0),
         },
         TestCase {
             name: "one wildcard, create change",
             descriptor: desc,
             change_descriptor: None,
-            to_send: 5_000,
+            to_send: Amount::from_sat(5_000),
             // should use change index of external keychain
             expect: (Some(1), 2),
         },
         TestCase {
             name: "one wildcard, no change",
             descriptor: desc,
             change_descriptor: None,
-            to_send: 9_850,
+            to_send: Amount::from_sat(9_850),
             // should not use change index
             expect: (Some(0), 1),
         },
         TestCase {
             name: "single key, create change",
             descriptor: get_test_tr_single_sig(),
             change_descriptor: None,
-            to_send: 5_000,
+            to_send: Amount::from_sat(5_000),
             // single key only has one derivation index (0)
             expect: (Some(0), 0),
         },
         TestCase {
             name: "single key, no change",
             descriptor: get_test_tr_single_sig(),
             change_descriptor: None,
-            to_send: 9_850,
+            to_send: Amount::from_sat(9_850),
             expect: (Some(0), 0),
         },
     ]
@@ -1665,7 +1665,7 @@ fn test_create_tx_increment_change_index() {
         receive_output(&mut wallet, amount, ReceiveTo::Mempool(0));
         // create tx
         let mut builder = wallet.build_tx();
-        builder.add_recipient(recipient.clone(), Amount::from_sat(test.to_send));
+        builder.add_recipient(recipient.clone(), test.to_send);
         let res = builder.finish();
         if !test.name.contains("error") {
             assert!(res.is_ok());
@@ -2557,7 +2557,7 @@ fn test_legacy_bump_fee_add_input() {
 #[test]
 fn test_bump_fee_absolute_add_input() {
     let (mut wallet, _) = get_funded_wallet_wpkh();
-    receive_output_in_latest_block(&mut wallet, 25_000);
+    receive_output_in_latest_block(&mut wallet, Amount::from_sat(25_000));
     let addr = Address::from_str("2N1Ffz3WaNzbeLFBb51xyFMHYSEUXcbiSoX")
         .unwrap()
         .assume_checked();
@@ -2605,7 +2605,7 @@ fn test_bump_fee_absolute_add_input() {
 #[test]
 fn test_legacy_bump_fee_absolute_add_input() {
     let (mut wallet, _) = get_funded_wallet_single(get_test_pkh());
-    receive_output_in_latest_block(&mut wallet, 25_000);
+    receive_output_in_latest_block(&mut wallet, Amount::from_sat(25_000));
     let addr = Address::from_str("2N1Ffz3WaNzbeLFBb51xyFMHYSEUXcbiSoX")
         .unwrap()
         .assume_checked();
@@ -2653,7 +2653,7 @@ fn test_legacy_bump_fee_absolute_add_input() {
 #[test]
 fn test_bump_fee_no_change_add_input_and_change() {
     let (mut wallet, _) = get_funded_wallet_wpkh();
-    let op = receive_output_in_latest_block(&mut wallet, 25_000);
+    let op = receive_output_in_latest_block(&mut wallet, Amount::from_sat(25_000));
 
     // initially make a tx without change by using `drain_to`
     let addr = Address::from_str("2N1Ffz3WaNzbeLFBb51xyFMHYSEUXcbiSoX")
@@ -2716,7 +2716,7 @@ fn test_bump_fee_no_change_add_input_and_change() {
 #[test]
 fn test_bump_fee_add_input_change_dust() {
     let (mut wallet, _) = get_funded_wallet_wpkh();
-    receive_output_in_latest_block(&mut wallet, 25_000);
+    receive_output_in_latest_block(&mut wallet, Amount::from_sat(25_000));
     let addr = Address::from_str("2N1Ffz3WaNzbeLFBb51xyFMHYSEUXcbiSoX")
         .unwrap()
         .assume_checked();
@@ -2786,7 +2786,7 @@ fn test_bump_fee_add_input_change_dust() {
 #[test]
 fn test_bump_fee_force_add_input() {
     let (mut wallet, _) = get_funded_wallet_wpkh();
-    let incoming_op = receive_output_in_latest_block(&mut wallet, 25_000);
+    let incoming_op = receive_output_in_latest_block(&mut wallet, Amount::from_sat(25_000));
 
     let addr = Address::from_str("2N1Ffz3WaNzbeLFBb51xyFMHYSEUXcbiSoX")
         .unwrap()
@@ -2843,7 +2843,7 @@ fn test_bump_fee_force_add_input() {
 #[test]
 fn test_bump_fee_absolute_force_add_input() {
     let (mut wallet, _) = get_funded_wallet_wpkh();
-    let incoming_op = receive_output_in_latest_block(&mut wallet, 25_000);
+    let incoming_op = receive_output_in_latest_block(&mut wallet, Amount::from_sat(25_000));
 
     let addr = Address::from_str("2N1Ffz3WaNzbeLFBb51xyFMHYSEUXcbiSoX")
         .unwrap()
@@ -2917,7 +2917,7 @@ fn test_bump_fee_unconfirmed_inputs_only() {
     let psbt = builder.finish().unwrap();
     // Now we receive one transaction with 0 confirmations. We won't be able to use that for
     // fee bumping, as it's still unconfirmed!
-    receive_output(&mut wallet, 25_000, ReceiveTo::Mempool(0));
+    receive_output(&mut wallet, Amount::from_sat(25_000), ReceiveTo::Mempool(0));
     let mut tx = psbt.extract_tx().expect("failed to extract tx");
     let txid = tx.compute_txid();
     for txin in &mut tx.input {
@@ -2943,7 +2943,7 @@ fn test_bump_fee_unconfirmed_input() {
         .assume_checked();
     // We receive a tx with 0 confirmations, which will be used as an input
     // in the drain tx.
-    receive_output(&mut wallet, 25_000, ReceiveTo::Mempool(0));
+    receive_output(&mut wallet, Amount::from_sat(25_000), ReceiveTo::Mempool(0));
     let mut builder = wallet.build_tx();
     builder.drain_wallet().drain_to(addr.script_pubkey());
     let psbt = builder.finish().unwrap();
@@ -2982,7 +2982,7 @@ fn test_fee_amount_negative_drain_val() {
         .unwrap()
         .assume_checked();
     let fee_rate = FeeRate::from_sat_per_kwu(500);
-    let incoming_op = receive_output_in_latest_block(&mut wallet, 8859);
+    let incoming_op = receive_output_in_latest_block(&mut wallet, Amount::from_sat(8859));
 
     let mut builder = wallet.build_tx();
     builder
@@ -3354,7 +3354,7 @@ fn test_next_unused_address() {
     assert_eq!(next_unused_addr.index, 0);
 
     // use the above address
-    receive_output(&mut wallet, 25_000, ReceiveTo::Mempool(0));
+    receive_output(&mut wallet, Amount::from_sat(25_000), ReceiveTo::Mempool(0));
 
     assert_eq!(
         wallet
@@ -4431,7 +4431,7 @@ fn test_keychains_with_overlapping_spks() {
         },
         confirmation_time: 0,
     };
-    let _outpoint = receive_output_to_address(&mut wallet, addr, 8000, anchor);
+    let _outpoint = receive_output_to_address(&mut wallet, addr, Amount::from_sat(8000), anchor);
     assert_eq!(wallet.balance().confirmed, Amount::from_sat(58000));
 }
 
@@ -4521,14 +4521,14 @@ fn single_descriptor_wallet_can_create_tx_and_receive_change() {
         .create_wallet_no_persist()
         .unwrap();
     assert_eq!(wallet.keychains().count(), 1);
-    let amt = Amount::from_sat(5_000);
-    receive_output(&mut wallet, 2 * amt.to_sat(), ReceiveTo::Mempool(2));
+    let amount = Amount::from_sat(5_000);
+    receive_output(&mut wallet, amount * 2, ReceiveTo::Mempool(2));
     // create spend tx that produces a change output
     let addr = Address::from_str("bcrt1qc6fweuf4xjvz4x3gx3t9e0fh4hvqyu2qw4wvxm")
         .unwrap()
         .assume_checked();
     let mut builder = wallet.build_tx();
-    builder.add_recipient(addr.script_pubkey(), amt);
+    builder.add_recipient(addr.script_pubkey(), amount);
     let mut psbt = builder.finish().unwrap();
     assert!(wallet.sign(&mut psbt, SignOptions::default()).unwrap());
     let tx = psbt.extract_tx().unwrap();
@@ -4537,7 +4537,7 @@ fn single_descriptor_wallet_can_create_tx_and_receive_change() {
     let unspent: Vec<_> = wallet.list_unspent().collect();
     assert_eq!(unspent.len(), 1);
     let utxo = unspent.first().unwrap();
-    assert!(utxo.txout.value < amt);
+    assert!(utxo.txout.value < amount);
     assert_eq!(
         utxo.keychain,
         KeychainKind::External,
@@ -4548,7 +4548,7 @@ fn single_descriptor_wallet_can_create_tx_and_receive_change() {
 #[test]
 fn test_transactions_sort_by() {
     let (mut wallet, _txid) = get_funded_wallet_wpkh();
-    receive_output(&mut wallet, 25_000, ReceiveTo::Mempool(0));
+    receive_output(&mut wallet, Amount::from_sat(25_000), ReceiveTo::Mempool(0));
 
     // sort by chain position, unconfirmed then confirmed by descending block height
     let sorted_txs: Vec<WalletTx> =
