125 lines
3.7 KiB
Rust
125 lines
3.7 KiB
Rust
use bdk_chain::{Indexer, SpkTxOutIndex};
|
|
use bitcoin::{
|
|
absolute, transaction, Amount, OutPoint, ScriptBuf, SignedAmount, Transaction, TxIn, TxOut,
|
|
};
|
|
|
|
#[test]
|
|
fn spk_txout_sent_and_received() {
|
|
let spk1 = ScriptBuf::from_hex("001404f1e52ce2bab3423c6a8c63b7cd730d8f12542c").unwrap();
|
|
let spk2 = ScriptBuf::from_hex("00142b57404ae14f08c3a0c903feb2af7830605eb00f").unwrap();
|
|
|
|
let mut index = SpkTxOutIndex::default();
|
|
index.insert_spk(0, spk1.clone());
|
|
index.insert_spk(1, spk2.clone());
|
|
|
|
let tx1 = Transaction {
|
|
version: transaction::Version::TWO,
|
|
lock_time: absolute::LockTime::ZERO,
|
|
input: vec![],
|
|
output: vec![TxOut {
|
|
value: Amount::from_sat(42_000),
|
|
script_pubkey: spk1.clone(),
|
|
}],
|
|
};
|
|
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, ..),
|
|
(Amount::from_sat(0), Amount::from_sat(42_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, ..1),
|
|
(Amount::from_sat(0), Amount::from_sat(42_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, 1..),
|
|
(Amount::from_sat(0), Amount::from_sat(0))
|
|
);
|
|
assert_eq!(index.net_value(&tx1, ..), SignedAmount::from_sat(42_000));
|
|
index.index_tx(&tx1);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx1, ..),
|
|
(Amount::from_sat(0), Amount::from_sat(42_000)),
|
|
"shouldn't change after scanning"
|
|
);
|
|
|
|
let tx2 = Transaction {
|
|
version: transaction::Version::ONE,
|
|
lock_time: absolute::LockTime::ZERO,
|
|
input: vec![TxIn {
|
|
previous_output: OutPoint {
|
|
txid: tx1.compute_txid(),
|
|
vout: 0,
|
|
},
|
|
..Default::default()
|
|
}],
|
|
output: vec![
|
|
TxOut {
|
|
value: Amount::from_sat(20_000),
|
|
script_pubkey: spk2,
|
|
},
|
|
TxOut {
|
|
script_pubkey: spk1,
|
|
value: Amount::from_sat(30_000),
|
|
},
|
|
],
|
|
};
|
|
|
|
assert_eq!(
|
|
index.sent_and_received(&tx2, ..),
|
|
(Amount::from_sat(42_000), Amount::from_sat(50_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx2, ..1),
|
|
(Amount::from_sat(42_000), Amount::from_sat(30_000))
|
|
);
|
|
assert_eq!(
|
|
index.sent_and_received(&tx2, 1..),
|
|
(Amount::from_sat(0), Amount::from_sat(20_000))
|
|
);
|
|
assert_eq!(index.net_value(&tx2, ..), SignedAmount::from_sat(8_000));
|
|
}
|
|
|
|
#[test]
|
|
fn mark_used() {
|
|
let spk1 = ScriptBuf::from_hex("001404f1e52ce2bab3423c6a8c63b7cd730d8f12542c").unwrap();
|
|
let spk2 = ScriptBuf::from_hex("00142b57404ae14f08c3a0c903feb2af7830605eb00f").unwrap();
|
|
|
|
let mut spk_index = SpkTxOutIndex::default();
|
|
spk_index.insert_spk(1, spk1.clone());
|
|
spk_index.insert_spk(2, spk2);
|
|
|
|
assert!(!spk_index.is_used(&1));
|
|
spk_index.mark_used(&1);
|
|
assert!(spk_index.is_used(&1));
|
|
spk_index.unmark_used(&1);
|
|
assert!(!spk_index.is_used(&1));
|
|
spk_index.mark_used(&1);
|
|
assert!(spk_index.is_used(&1));
|
|
|
|
let tx1 = Transaction {
|
|
version: transaction::Version::TWO,
|
|
lock_time: absolute::LockTime::ZERO,
|
|
input: vec![],
|
|
output: vec![TxOut {
|
|
value: Amount::from_sat(42_000),
|
|
script_pubkey: spk1,
|
|
}],
|
|
};
|
|
|
|
spk_index.index_tx(&tx1);
|
|
spk_index.unmark_used(&1);
|
|
assert!(
|
|
spk_index.is_used(&1),
|
|
"even though we unmark_used it doesn't matter because there was a tx scanned that used it"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn unmark_used_does_not_result_in_invalid_representation() {
|
|
let mut spk_index = SpkTxOutIndex::default();
|
|
assert!(!spk_index.unmark_used(&0));
|
|
assert!(!spk_index.unmark_used(&1));
|
|
assert!(!spk_index.unmark_used(&2));
|
|
assert!(spk_index.unused_spks(..).collect::<Vec<_>>().is_empty());
|
|
}
|