Skip to content

Commit

Permalink
rustfmt: lightning-transaction-sync/tests/integration_tests.rs
Browse files Browse the repository at this point in the history
  • Loading branch information
tnull committed Jun 7, 2024
1 parent 85a8204 commit 21fee20
Show file tree
Hide file tree
Showing 2 changed files with 63 additions and 26 deletions.
88 changes: 63 additions & 25 deletions lightning-transaction-sync/tests/integration_tests.rs
Original file line number Diff line number Diff line change
@@ -1,26 +1,29 @@
#![cfg(all(not(target_os = "windows"), any(feature = "esplora-blocking", feature = "esplora-async", feature = "electrum")))]
#![cfg(all(
not(target_os = "windows"),
any(feature = "esplora-blocking", feature = "esplora-async", feature = "electrum")
))]

#[cfg(any(feature = "esplora-blocking", feature = "esplora-async"))]
use lightning_transaction_sync::EsploraSyncClient;
#[cfg(feature = "electrum")]
use lightning_transaction_sync::ElectrumSyncClient;
use lightning::chain::{Confirm, Filter, WatchedOutput};
use lightning::chain::transaction::{OutPoint, TransactionData};
use lightning::chain::{Confirm, Filter, WatchedOutput};
use lightning::util::test_utils::TestLogger;
#[cfg(feature = "electrum")]
use lightning_transaction_sync::ElectrumSyncClient;
#[cfg(any(feature = "esplora-blocking", feature = "esplora-async"))]
use lightning_transaction_sync::EsploraSyncClient;

use electrsd::{bitcoind, bitcoind::BitcoinD, ElectrsD};
use bitcoin::{Amount, Txid, BlockHash};
use bdk_macros::maybe_await;
use bitcoin::blockdata::block::Header;
use bitcoin::blockdata::constants::genesis_block;
use bitcoin::network::Network;
use electrsd::bitcoind::bitcoincore_rpc::bitcoincore_rpc_json::AddressType;
use bitcoin::{Amount, BlockHash, Txid};
use bitcoind::bitcoincore_rpc::RpcApi;
use bdk_macros::maybe_await;
use electrsd::bitcoind::bitcoincore_rpc::bitcoincore_rpc_json::AddressType;
use electrsd::{bitcoind, bitcoind::BitcoinD, ElectrsD};

use std::collections::{HashMap, HashSet};
use std::env;
use std::sync::Mutex;
use std::time::Duration;
use std::collections::{HashMap, HashSet};

pub fn setup_bitcoind_and_electrsd() -> (BitcoinD, ElectrsD) {
let bitcoind_exe =
Expand Down Expand Up @@ -63,8 +66,11 @@ pub fn wait_for_block(electrsd: &ElectrsD, min_height: usize) {
// it didn't. Since we can't proceed without subscribing, we try again after a delay
// and panic if it still fails.
std::thread::sleep(Duration::from_secs(1));
electrsd.client.block_headers_subscribe_raw().expect("failed to subscribe to block headers")
}
electrsd
.client
.block_headers_subscribe_raw()
.expect("failed to subscribe to block headers")
},
};
loop {
if header.height >= min_height {
Expand All @@ -90,7 +96,7 @@ where
None if delay.as_millis() < 512 => {
delay = delay.mul_f32(2.0);
tries += 1;
}
},
None if tries == 10 => panic!("Exceeded our maximum wait time."),
None => tries += 1,
}
Expand Down Expand Up @@ -132,7 +138,8 @@ impl Confirm for TestConfirmable {
let block_hash = header.block_hash();
self.confirmed_txs.lock().unwrap().insert(txid, (block_hash, height));
self.unconfirmed_txs.lock().unwrap().remove(&txid);
self.events.lock().unwrap().push(TestConfirmableEvent::Confirmed(txid, block_hash, height));
let event = TestConfirmableEvent::Confirmed(txid, block_hash, height);
self.events.lock().unwrap().push(event);
}
}

Expand All @@ -145,11 +152,13 @@ impl Confirm for TestConfirmable {
fn best_block_updated(&self, header: &Header, height: u32) {
let block_hash = header.block_hash();
*self.best_block.lock().unwrap() = (block_hash, height);
self.events.lock().unwrap().push(TestConfirmableEvent::BestBlockUpdated(block_hash, height));
let event = TestConfirmableEvent::BestBlockUpdated(block_hash, height);
self.events.lock().unwrap().push(event);
}

fn get_relevant_txids(&self) -> Vec<(Txid, u32, Option<BlockHash>)> {
self.confirmed_txs.lock().unwrap().iter().map(|(&txid, (hash, height))| (txid, *height, Some(*hash))).collect::<Vec<_>>()
let lock = self.confirmed_txs.lock().unwrap();
lock.iter().map(|(&txid, (hash, height))| (txid, *height, Some(*hash))).collect()
}
}

Expand All @@ -165,12 +174,37 @@ macro_rules! test_syncing {
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();
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.payload().script_pubkey());

maybe_await!($tx_sync.sync(vec![&$confirmable])).unwrap();
Expand All @@ -193,13 +227,17 @@ macro_rules! test_syncing {
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()
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
script_pubkey: prev_script_pubkey,
};

$tx_sync.register_output(output);
Expand Down
1 change: 0 additions & 1 deletion rustfmt_excluded_files
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,6 @@
./lightning-rapid-gossip-sync/src/error.rs
./lightning-rapid-gossip-sync/src/lib.rs
./lightning-rapid-gossip-sync/src/processing.rs
./lightning-transaction-sync/tests/integration_tests.rs
./lightning/src/blinded_path/message.rs
./lightning/src/blinded_path/mod.rs
./lightning/src/blinded_path/payment.rs
Expand Down

0 comments on commit 21fee20

Please sign in to comment.