use bitcoin::network::constants::Network;
use electrsd::bitcoind::bitcoincore_rpc::bitcoincore_rpc_json::AddressType;
use bitcoind::bitcoincore_rpc::RpcApi;
+use bdk_macros::maybe_await;
use std::env;
use std::sync::Mutex;
}
}
-#[test]
-#[cfg(feature = "esplora-blocking")]
-fn test_esplora_syncs() {
- let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
- generate_blocks_and_wait(&bitcoind, &electrsd, 101);
- let mut logger = TestLogger::new();
- let esplora_url = format!("http://{}", electrsd.esplora_url.as_ref().unwrap());
- let tx_sync = EsploraSyncClient::new(esplora_url, &mut logger);
- let confirmable = TestConfirmable::new();
+macro_rules! test_syncing {
+ ($tx_sync: expr, $confirmable: expr, $bitcoind: expr, $electrsd: expr) => {{
+ // Check we pick up on new best blocks
+ assert_eq!($confirmable.best_block.lock().unwrap().1, 0);
- // Check we pick up on new best blocks
- assert_eq!(confirmable.best_block.lock().unwrap().1, 0);
+ maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
+ assert_eq!($confirmable.best_block.lock().unwrap().1, 102);
- tx_sync.sync(vec![&confirmable]).unwrap();
- assert_eq!(confirmable.best_block.lock().unwrap().1, 102);
+ let events = std::mem::take(&mut *$confirmable.events.lock().unwrap());
+ assert_eq!(events.len(), 1);
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 1);
-
- // Check registered confirmed transactions are marked confirmed
- let new_address = bitcoind.client.get_new_address(Some("test"),
+ // Check registered confirmed transactions are marked confirmed
+ let new_address = $bitcoind.client.get_new_address(Some("test"),
Some(AddressType::Legacy)).unwrap().assume_checked();
- let txid = bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None, None,
+ let txid = $bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None, None,
None, None, None, None).unwrap();
- let second_txid = bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None,
+ let second_txid = $bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None,
None, None, None, None, None).unwrap();
- tx_sync.register_tx(&txid, &new_address.script_pubkey());
-
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 0);
- assert!(confirmable.confirmed_txs.lock().unwrap().is_empty());
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 2);
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Now take an arbitrary output of the second transaction and check we'll confirm its spend.
- let tx_res = bitcoind.client.get_transaction(&second_txid, None).unwrap();
- let block_hash = tx_res.info.blockhash.unwrap();
- let tx = tx_res.transaction().unwrap();
- let prev_outpoint = tx.input.first().unwrap().previous_output;
- let prev_tx = bitcoind.client.get_transaction(&prev_outpoint.txid, None).unwrap().transaction()
- .unwrap();
- let prev_script_pubkey = prev_tx.output[prev_outpoint.vout as usize].script_pubkey.clone();
- let output = WatchedOutput {
- block_hash: Some(block_hash),
- outpoint: OutPoint { txid: prev_outpoint.txid, index: prev_outpoint.vout as u16 },
- script_pubkey: prev_script_pubkey
- };
+ $tx_sync.register_tx(&txid, &new_address.payload.script_pubkey());
+
+ maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
+
+ let events = std::mem::take(&mut *$confirmable.events.lock().unwrap());
+ assert_eq!(events.len(), 0);
+ assert!($confirmable.confirmed_txs.lock().unwrap().is_empty());
+ assert!($confirmable.unconfirmed_txs.lock().unwrap().is_empty());
+
+ generate_blocks_and_wait(&$bitcoind, &$electrsd, 1);
+ maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
+
+ let events = std::mem::take(&mut *$confirmable.events.lock().unwrap());
+ assert_eq!(events.len(), 2);
+ assert!($confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
+ assert!($confirmable.unconfirmed_txs.lock().unwrap().is_empty());
+
+ // Now take an arbitrary output of the second transaction and check we'll confirm its spend.
+ let tx_res = $bitcoind.client.get_transaction(&second_txid, None).unwrap();
+ let block_hash = tx_res.info.blockhash.unwrap();
+ let tx = tx_res.transaction().unwrap();
+ let prev_outpoint = tx.input.first().unwrap().previous_output;
+ let prev_tx = $bitcoind.client.get_transaction(&prev_outpoint.txid, None).unwrap().transaction()
+ .unwrap();
+ let prev_script_pubkey = prev_tx.output[prev_outpoint.vout as usize].script_pubkey.clone();
+ let output = WatchedOutput {
+ block_hash: Some(block_hash),
+ outpoint: OutPoint { txid: prev_outpoint.txid, index: prev_outpoint.vout as u16 },
+ script_pubkey: prev_script_pubkey
+ };
+
+ $tx_sync.register_output(output);
+ maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
+
+ let events = std::mem::take(&mut *$confirmable.events.lock().unwrap());
+ assert_eq!(events.len(), 1);
+ assert!($confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
+ assert_eq!($confirmable.confirmed_txs.lock().unwrap().len(), 2);
+ assert!($confirmable.unconfirmed_txs.lock().unwrap().is_empty());
+
+ // Check previously confirmed transactions are marked unconfirmed when they are reorged.
+ let best_block_hash = $bitcoind.client.get_best_block_hash().unwrap();
+ $bitcoind.client.invalidate_block(&best_block_hash).unwrap();
+
+ // We're getting back to the previous height with a new tip, but best block shouldn't change.
+ generate_blocks_and_wait(&$bitcoind, &$electrsd, 1);
+ assert_ne!($bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
+ maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
+ let events = std::mem::take(&mut *$confirmable.events.lock().unwrap());
+ assert_eq!(events.len(), 0);
+
+ // Now we're surpassing previous height, getting new tip.
+ generate_blocks_and_wait(&$bitcoind, &$electrsd, 1);
+ assert_ne!($bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
+ maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
+
+ // Transactions still confirmed but under new tip.
+ assert!($confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
+ assert!($confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
+ assert!($confirmable.unconfirmed_txs.lock().unwrap().is_empty());
+
+ // Check we got unconfirmed, then reconfirmed in the meantime.
+ let mut seen_txids = HashSet::new();
+ let events = std::mem::take(&mut *$confirmable.events.lock().unwrap());
+ assert_eq!(events.len(), 5);
+
+ match events[0] {
+ TestConfirmableEvent::Unconfirmed(t) => {
+ assert!(t == txid || t == second_txid);
+ assert!(seen_txids.insert(t));
+ },
+ _ => panic!("Unexpected event"),
+ }
- tx_sync.register_output(output);
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 1);
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
- assert_eq!(confirmable.confirmed_txs.lock().unwrap().len(), 2);
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Check previously confirmed transactions are marked unconfirmed when they are reorged.
- let best_block_hash = bitcoind.client.get_best_block_hash().unwrap();
- bitcoind.client.invalidate_block(&best_block_hash).unwrap();
-
- // We're getting back to the previous height with a new tip, but best block shouldn't change.
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- assert_ne!(bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
- tx_sync.sync(vec![&confirmable]).unwrap();
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 0);
-
- // Now we're surpassing previous height, getting new tip.
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- assert_ne!(bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- // Transactions still confirmed but under new tip.
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Check we got unconfirmed, then reconfirmed in the meantime.
- let mut seen_txids = HashSet::new();
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 5);
-
- match events[0] {
- TestConfirmableEvent::Unconfirmed(t) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.insert(t));
- },
- _ => panic!("Unexpected event"),
- }
+ match events[1] {
+ TestConfirmableEvent::Unconfirmed(t) => {
+ assert!(t == txid || t == second_txid);
+ assert!(seen_txids.insert(t));
+ },
+ _ => panic!("Unexpected event"),
+ }
- match events[1] {
- TestConfirmableEvent::Unconfirmed(t) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.insert(t));
- },
- _ => panic!("Unexpected event"),
- }
+ match events[2] {
+ TestConfirmableEvent::BestBlockUpdated(..) => {},
+ _ => panic!("Unexpected event"),
+ }
- match events[2] {
- TestConfirmableEvent::BestBlockUpdated(..) => {},
- _ => panic!("Unexpected event"),
- }
+ match events[3] {
+ TestConfirmableEvent::Confirmed(t, _, _) => {
+ assert!(t == txid || t == second_txid);
+ assert!(seen_txids.remove(&t));
+ },
+ _ => panic!("Unexpected event"),
+ }
- match events[3] {
- TestConfirmableEvent::Confirmed(t, _, _) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.remove(&t));
- },
- _ => panic!("Unexpected event"),
- }
+ match events[4] {
+ TestConfirmableEvent::Confirmed(t, _, _) => {
+ assert!(t == txid || t == second_txid);
+ assert!(seen_txids.remove(&t));
+ },
+ _ => panic!("Unexpected event"),
+ }
- match events[4] {
- TestConfirmableEvent::Confirmed(t, _, _) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.remove(&t));
- },
- _ => panic!("Unexpected event"),
- }
+ assert_eq!(seen_txids.len(), 0);
+ }};
+}
- assert_eq!(seen_txids.len(), 0);
+#[test]
+#[cfg(feature = "esplora-blocking")]
+fn test_esplora_syncs() {
+ let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
+ generate_blocks_and_wait(&bitcoind, &electrsd, 101);
+ let mut logger = TestLogger::new();
+ let esplora_url = format!("http://{}", electrsd.esplora_url.as_ref().unwrap());
+ let tx_sync = EsploraSyncClient::new(esplora_url, &mut logger);
+ let confirmable = TestConfirmable::new();
+
+ test_syncing!(tx_sync, confirmable, bitcoind, electrsd);
}
#[tokio::test]
let tx_sync = EsploraSyncClient::new(esplora_url, &mut logger);
let confirmable = TestConfirmable::new();
- // Check we pick up on new best blocks
- assert_eq!(confirmable.best_block.lock().unwrap().1, 0);
-
- tx_sync.sync(vec![&confirmable]).await.unwrap();
- assert_eq!(confirmable.best_block.lock().unwrap().1, 102);
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 1);
-
- // Check registered confirmed transactions are marked confirmed
- let new_address = bitcoind.client.get_new_address(Some("test"),
- Some(AddressType::Legacy)).unwrap().assume_checked();
- let txid = bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None, None,
- None, None, None, None).unwrap();
- let second_txid = bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None,
- None, None, None, None, None).unwrap();
- tx_sync.register_tx(&txid, &new_address.script_pubkey());
-
- tx_sync.sync(vec![&confirmable]).await.unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 0);
- assert!(confirmable.confirmed_txs.lock().unwrap().is_empty());
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- tx_sync.sync(vec![&confirmable]).await.unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 2);
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Now take an arbitrary output of the second transaction and check we'll confirm its spend.
- let tx_res = bitcoind.client.get_transaction(&second_txid, None).unwrap();
- let block_hash = tx_res.info.blockhash.unwrap();
- let tx = tx_res.transaction().unwrap();
- let prev_outpoint = tx.input.first().unwrap().previous_output;
- let prev_tx = bitcoind.client.get_transaction(&prev_outpoint.txid, None).unwrap().transaction()
- .unwrap();
- let prev_script_pubkey = prev_tx.output[prev_outpoint.vout as usize].script_pubkey.clone();
- let output = WatchedOutput {
- block_hash: Some(block_hash),
- outpoint: OutPoint { txid: prev_outpoint.txid, index: prev_outpoint.vout as u16 },
- script_pubkey: prev_script_pubkey
- };
-
- tx_sync.register_output(output);
- tx_sync.sync(vec![&confirmable]).await.unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 1);
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
- assert_eq!(confirmable.confirmed_txs.lock().unwrap().len(), 2);
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Check previously confirmed transactions are marked unconfirmed when they are reorged.
- let best_block_hash = bitcoind.client.get_best_block_hash().unwrap();
- bitcoind.client.invalidate_block(&best_block_hash).unwrap();
-
- // We're getting back to the previous height with a new tip, but best block shouldn't change.
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- assert_ne!(bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
- tx_sync.sync(vec![&confirmable]).await.unwrap();
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 0);
-
- // Now we're surpassing previous height, getting new tip.
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- assert_ne!(bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
- tx_sync.sync(vec![&confirmable]).await.unwrap();
-
- // Transactions still confirmed but under new tip.
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Check we got unconfirmed, then reconfirmed in the meantime.
- let mut seen_txids = HashSet::new();
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 5);
-
- match events[0] {
- TestConfirmableEvent::Unconfirmed(t) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.insert(t));
- },
- _ => panic!("Unexpected event"),
- }
-
- match events[1] {
- TestConfirmableEvent::Unconfirmed(t) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.insert(t));
- },
- _ => panic!("Unexpected event"),
- }
-
- match events[2] {
- TestConfirmableEvent::BestBlockUpdated(..) => {},
- _ => panic!("Unexpected event"),
- }
-
- match events[3] {
- TestConfirmableEvent::Confirmed(t, _, _) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.remove(&t));
- },
- _ => panic!("Unexpected event"),
- }
-
- match events[4] {
- TestConfirmableEvent::Confirmed(t, _, _) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.remove(&t));
- },
- _ => panic!("Unexpected event"),
- }
-
- assert_eq!(seen_txids.len(), 0);
+ test_syncing!(tx_sync, confirmable, bitcoind, electrsd);
}
#[test]
let electrum_url = format!("tcp://{}", electrsd.electrum_url);
let tx_sync = ElectrumSyncClient::new(electrum_url, &mut logger).unwrap();
let confirmable = TestConfirmable::new();
-
- // Check we pick up on new best blocks
- assert_eq!(confirmable.best_block.lock().unwrap().1, 0);
-
- tx_sync.sync(vec![&confirmable]).unwrap();
- assert_eq!(confirmable.best_block.lock().unwrap().1, 102);
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 1);
-
- // Check registered confirmed transactions are marked confirmed
- let new_address = bitcoind.client.get_new_address(Some("test"),
- Some(AddressType::Legacy)).unwrap().assume_checked();
- let txid = bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None, None,
- None, None, None, None).unwrap();
- let second_txid = bitcoind.client.send_to_address(&new_address, Amount::from_sat(5000), None,
- None, None, None, None, None).unwrap();
- tx_sync.register_tx(&txid, &new_address.script_pubkey());
-
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 0);
- assert!(confirmable.confirmed_txs.lock().unwrap().is_empty());
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 2);
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Now take an arbitrary output of the second transaction and check we'll confirm its spend.
- let tx_res = bitcoind.client.get_transaction(&second_txid, None).unwrap();
- let block_hash = tx_res.info.blockhash.unwrap();
- let tx = tx_res.transaction().unwrap();
- let prev_outpoint = tx.input.first().unwrap().previous_output;
- let prev_tx = bitcoind.client.get_transaction(&prev_outpoint.txid, None).unwrap().transaction()
- .unwrap();
- let prev_script_pubkey = prev_tx.output[prev_outpoint.vout as usize].script_pubkey.clone();
- let output = WatchedOutput {
- block_hash: Some(block_hash),
- outpoint: OutPoint { txid: prev_outpoint.txid, index: prev_outpoint.vout as u16 },
- script_pubkey: prev_script_pubkey
- };
-
- tx_sync.register_output(output);
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 1);
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
- assert_eq!(confirmable.confirmed_txs.lock().unwrap().len(), 2);
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Check previously confirmed transactions are marked unconfirmed when they are reorged.
- let best_block_hash = bitcoind.client.get_best_block_hash().unwrap();
- bitcoind.client.invalidate_block(&best_block_hash).unwrap();
-
- // We're getting back to the previous height with a new tip, but best block shouldn't change.
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- assert_ne!(bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
- tx_sync.sync(vec![&confirmable]).unwrap();
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 0);
-
- // Now we're surpassing previous height, getting new tip.
- generate_blocks_and_wait(&bitcoind, &electrsd, 1);
- assert_ne!(bitcoind.client.get_best_block_hash().unwrap(), best_block_hash);
- tx_sync.sync(vec![&confirmable]).unwrap();
-
- // Transactions still confirmed but under new tip.
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&txid));
- assert!(confirmable.confirmed_txs.lock().unwrap().contains_key(&second_txid));
- assert!(confirmable.unconfirmed_txs.lock().unwrap().is_empty());
-
- // Check we got unconfirmed, then reconfirmed in the meantime.
- let mut seen_txids = HashSet::new();
- let events = std::mem::take(&mut *confirmable.events.lock().unwrap());
- assert_eq!(events.len(), 5);
-
- match events[0] {
- TestConfirmableEvent::Unconfirmed(t) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.insert(t));
- },
- _ => panic!("Unexpected event"),
- }
-
- match events[1] {
- TestConfirmableEvent::Unconfirmed(t) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.insert(t));
- },
- _ => panic!("Unexpected event"),
- }
-
- match events[2] {
- TestConfirmableEvent::BestBlockUpdated(..) => {},
- _ => panic!("Unexpected event"),
- }
-
- match events[3] {
- TestConfirmableEvent::Confirmed(t, _, _) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.remove(&t));
- },
- _ => panic!("Unexpected event"),
- }
-
- match events[4] {
- TestConfirmableEvent::Confirmed(t, _, _) => {
- assert!(t == txid || t == second_txid);
- assert!(seen_txids.remove(&t));
- },
- _ => panic!("Unexpected event"),
- }
-
- assert_eq!(seen_txids.len(), 0);
+ test_syncing!(tx_sync, confirmable, bitcoind, electrsd);
}