Skip to content

Commit 4d1a9fd

Browse files
notmandatorynotmandatory
committed
test(esplora): add async_ext and blocking_ext integration tests
1 parent f95506b commit 4d1a9fd

File tree

4 files changed

+237
-2
lines changed

4 files changed

+237
-2
lines changed

‎crates/esplora/Cargo.toml‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,10 @@ futures = { version = "0.3.26", optional = true }
2121
bitcoin = { version = "0.30.0", optional = true, default-features = false }
2222
miniscript = { version = "10.0.0", optional = true, default-features = false }
2323

24+
[dev-dependencies]
25+
electrsd = { version= "0.25.0", features = ["bitcoind_25_0", "esplora_a33e97e1", "legacy"] }
26+
tokio = { version = "1", features = ["rt", "rt-multi-thread", "macros"] }
27+
2428
[features]
2529
default = ["std", "async-https", "blocking"]
2630
std = ["bdk_chain/std"]

‎crates/esplora/src/async_ext.rs‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ use crate::{anchor_from_status, ASSUME_FINAL_DEPTH};
2121
pub trait EsploraAsyncExt {
2222
/// Prepare an [`LocalChain`] update with blocks fetched from Esplora.
2323
///
24-
/// * `prev_tip` is the previous tip of [`LocalChain::tip`].
25-
/// * `get_heights` is the block heights that we are interested in fetching from Esplora.
24+
/// * `local_tip` is the previous tip of [`LocalChain::tip`].
25+
/// * `request_heights` is the block heights that we are interested in fetching from Esplora.
2626
///
2727
/// The result of this method can be applied to [`LocalChain::apply_update`].
2828
///

‎crates/esplora/tests/async_ext.rs‎

Lines changed: 117 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,117 @@
1+
use bdk_esplora::EsploraAsyncExt;
2+
use electrsd::bitcoind::bitcoincore_rpc::RpcApi;
3+
use electrsd::bitcoind::{self, anyhow, BitcoinD};
4+
use electrsd::{Conf, ElectrsD};
5+
use esplora_client::{self, AsyncClient, Builder};
6+
use std::str::FromStr;
7+
use std::thread::sleep;
8+
use std::time::Duration;
9+
10+
use bdk_chain::bitcoin::{Address, Amount, BlockHash, Txid};
11+
12+
struct TestEnv {
13+
bitcoind: BitcoinD,
14+
#[allow(dead_code)]
15+
electrsd: ElectrsD,
16+
client: AsyncClient,
17+
}
18+
19+
impl TestEnv {
20+
fn new() -> Result<Self, anyhow::Error> {
21+
let bitcoind_exe =
22+
bitcoind::downloaded_exe_path().expect("bitcoind version feature must be enabled");
23+
let bitcoind = BitcoinD::new(bitcoind_exe).unwrap();
24+
25+
let mut electrs_conf = Conf::default();
26+
electrs_conf.http_enabled = true;
27+
let electrs_exe =
28+
electrsd::downloaded_exe_path().expect("electrs version feature must be enabled");
29+
let electrsd = ElectrsD::with_conf(electrs_exe, &bitcoind, &electrs_conf)?;
30+
31+
let base_url = format!("http://{}", &electrsd.esplora_url.clone().unwrap());
32+
let client = Builder::new(base_url.as_str()).build_async()?;
33+
34+
Ok(Self {
35+
bitcoind,
36+
electrsd,
37+
client,
38+
})
39+
}
40+
41+
fn mine_blocks(
42+
&self,
43+
count: usize,
44+
address: Option<Address>,
45+
) -> anyhow::Result<Vec<BlockHash>> {
46+
let coinbase_address = match address {
47+
Some(address) => address,
48+
None => self
49+
.bitcoind
50+
.client
51+
.get_new_address(None, None)?
52+
.assume_checked(),
53+
};
54+
let block_hashes = self
55+
.bitcoind
56+
.client
57+
.generate_to_address(count as _, &coinbase_address)?;
58+
Ok(block_hashes)
59+
}
60+
}
61+
62+
#[tokio::test]
63+
pub async fn test_update_tx_graph_without_keychain() -> anyhow::Result<()> {
64+
let env = TestEnv::new()?;
65+
let receive_address0 =
66+
Address::from_str("bcrt1qc6fweuf4xjvz4x3gx3t9e0fh4hvqyu2qw4wvxm")?.assume_checked();
67+
let receive_address1 =
68+
Address::from_str("bcrt1qfjg5lv3dvc9az8patec8fjddrs4aqtauadnagr")?.assume_checked();
69+
70+
let misc_spks = [
71+
receive_address0.script_pubkey(),
72+
receive_address1.script_pubkey(),
73+
];
74+
75+
let _block_hashes = env.mine_blocks(101, None)?;
76+
let txid1 = env.bitcoind.client.send_to_address(
77+
&receive_address1,
78+
Amount::from_sat(10000),
79+
None,
80+
None,
81+
None,
82+
None,
83+
Some(1),
84+
None,
85+
)?;
86+
let txid2 = env.bitcoind.client.send_to_address(
87+
&receive_address0,
88+
Amount::from_sat(20000),
89+
None,
90+
None,
91+
None,
92+
None,
93+
Some(1),
94+
None,
95+
)?;
96+
let _block_hashes = env.mine_blocks(1, None)?;
97+
while env.client.get_height().await.unwrap() < 102 {
98+
sleep(Duration::from_millis(10))
99+
}
100+
101+
let graph_update = env
102+
.client
103+
.scan_txs(
104+
misc_spks.into_iter(),
105+
vec![].into_iter(),
106+
vec![].into_iter(),
107+
1,
108+
)
109+
.await?;
110+
111+
let mut graph_update_txids: Vec<Txid> = graph_update.full_txs().map(|tx| tx.txid).collect();
112+
graph_update_txids.sort();
113+
let mut expected_txids = vec![txid1, txid2];
114+
expected_txids.sort();
115+
assert_eq!(graph_update_txids, expected_txids);
116+
Ok(())
117+
}

‎crates/esplora/tests/blocking_ext.rs‎

Lines changed: 114 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,114 @@
1+
use bdk_esplora::EsploraExt;
2+
use electrsd::bitcoind::bitcoincore_rpc::RpcApi;
3+
use electrsd::bitcoind::{self, anyhow, BitcoinD};
4+
use electrsd::{Conf, ElectrsD};
5+
use esplora_client::{self, BlockingClient, Builder};
6+
use std::str::FromStr;
7+
use std::thread::sleep;
8+
use std::time::Duration;
9+
10+
use bdk_chain::bitcoin::{Address, Amount, BlockHash, Txid};
11+
12+
struct TestEnv {
13+
bitcoind: BitcoinD,
14+
#[allow(dead_code)]
15+
electrsd: ElectrsD,
16+
client: BlockingClient,
17+
}
18+
19+
impl TestEnv {
20+
fn new() -> Result<Self, anyhow::Error> {
21+
let bitcoind_exe =
22+
bitcoind::downloaded_exe_path().expect("bitcoind version feature must be enabled");
23+
let bitcoind = BitcoinD::new(bitcoind_exe).unwrap();
24+
25+
let mut electrs_conf = Conf::default();
26+
electrs_conf.http_enabled = true;
27+
let electrs_exe =
28+
electrsd::downloaded_exe_path().expect("electrs version feature must be enabled");
29+
let electrsd = ElectrsD::with_conf(electrs_exe, &bitcoind, &electrs_conf)?;
30+
31+
let base_url = format!("http://{}", &electrsd.esplora_url.clone().unwrap());
32+
let client = Builder::new(base_url.as_str()).build_blocking()?;
33+
34+
Ok(Self {
35+
bitcoind,
36+
electrsd,
37+
client,
38+
})
39+
}
40+
41+
fn mine_blocks(
42+
&self,
43+
count: usize,
44+
address: Option<Address>,
45+
) -> anyhow::Result<Vec<BlockHash>> {
46+
let coinbase_address = match address {
47+
Some(address) => address,
48+
None => self
49+
.bitcoind
50+
.client
51+
.get_new_address(None, None)?
52+
.assume_checked(),
53+
};
54+
let block_hashes = self
55+
.bitcoind
56+
.client
57+
.generate_to_address(count as _, &coinbase_address)?;
58+
Ok(block_hashes)
59+
}
60+
}
61+
62+
#[test]
63+
pub fn test_update_tx_graph_without_keychain() -> anyhow::Result<()> {
64+
let env = TestEnv::new()?;
65+
let receive_address0 =
66+
Address::from_str("bcrt1qc6fweuf4xjvz4x3gx3t9e0fh4hvqyu2qw4wvxm")?.assume_checked();
67+
let receive_address1 =
68+
Address::from_str("bcrt1qfjg5lv3dvc9az8patec8fjddrs4aqtauadnagr")?.assume_checked();
69+
70+
let misc_spks = [
71+
receive_address0.script_pubkey(),
72+
receive_address1.script_pubkey(),
73+
];
74+
75+
let _block_hashes = env.mine_blocks(101, None)?;
76+
let txid1 = env.bitcoind.client.send_to_address(
77+
&receive_address1,
78+
Amount::from_sat(10000),
79+
None,
80+
None,
81+
None,
82+
None,
83+
Some(1),
84+
None,
85+
)?;
86+
let txid2 = env.bitcoind.client.send_to_address(
87+
&receive_address0,
88+
Amount::from_sat(20000),
89+
None,
90+
None,
91+
None,
92+
None,
93+
Some(1),
94+
None,
95+
)?;
96+
let _block_hashes = env.mine_blocks(1, None)?;
97+
while env.client.get_height().unwrap() < 102 {
98+
sleep(Duration::from_millis(10))
99+
}
100+
101+
let graph_update = env.client.scan_txs(
102+
misc_spks.into_iter(),
103+
vec![].into_iter(),
104+
vec![].into_iter(),
105+
1,
106+
)?;
107+
108+
let mut graph_update_txids: Vec<Txid> = graph_update.full_txs().map(|tx| tx.txid).collect();
109+
graph_update_txids.sort();
110+
let mut expected_txids = vec![txid1, txid2];
111+
expected_txids.sort();
112+
assert_eq!(graph_update_txids, expected_txids);
113+
Ok(())
114+
}

0 commit comments

Comments
 (0)