Skip to content

Commit 4eb8745

Browse files
nit: Refactor node setup in tests to use new_node function for improved clarity
1 parent 7cef820 commit 4eb8745

File tree

3 files changed

+18
-27
lines changed

3 files changed

+18
-27
lines changed

tests/common/mod.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -301,6 +301,11 @@ pub(crate) fn setup_two_nodes(
301301
(node_a, node_b)
302302
}
303303

304+
pub(crate) fn new_node(chain_source: &TestChainSource, anchor_channels: bool) -> TestNode {
305+
let config = random_config(anchor_channels);
306+
setup_node(chain_source, config, None)
307+
}
308+
304309
pub(crate) fn setup_node(
305310
chain_source: &TestChainSource, config: TestConfig, seed_bytes: Option<Vec<u8>>,
306311
) -> TestNode {

tests/integration_tests_rust.rs

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,10 @@ use common::{
2020
bump_fee_and_broadcast, distribute_funds_unconfirmed, do_channel_full_cycle,
2121
expect_channel_pending_event, expect_channel_ready_event, expect_event,
2222
expect_payment_claimable_event, expect_payment_received_event, expect_payment_successful_event,
23-
generate_blocks_and_wait, open_channel, open_channel_push_amt, premine_and_distribute_funds,
24-
premine_blocks, prepare_rbf, random_config, random_listening_addresses,
25-
setup_bitcoind_and_electrsd, setup_builder, setup_node, setup_node_for_async_payments,
26-
setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
23+
generate_blocks_and_wait, new_node, open_channel, open_channel_push_amt,
24+
premine_and_distribute_funds, premine_blocks, prepare_rbf, random_config,
25+
random_listening_addresses, setup_bitcoind_and_electrsd, setup_builder, setup_node,
26+
setup_node_for_async_payments, setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
2727
};
2828
use ldk_node::config::{AsyncPaymentsRole, EsploraSyncConfig};
2929
use ldk_node::liquidity::LSPS2ServiceConfig;
@@ -687,18 +687,11 @@ fn run_rbf_test(is_insert_block: bool) {
687687
let chain_source_electrsd = TestChainSource::Electrum(&electrsd);
688688
let chain_source_esplora = TestChainSource::Esplora(&electrsd);
689689

690-
macro_rules! config_node {
691-
($chain_source: expr, $anchor_channels: expr) => {{
692-
let config_a = random_config($anchor_channels);
693-
let node = setup_node(&$chain_source, config_a, None);
694-
node
695-
}};
696-
}
697690
let anchor_channels = false;
698691
let nodes = vec![
699-
config_node!(chain_source_electrsd, anchor_channels),
700-
config_node!(chain_source_bitcoind, anchor_channels),
701-
config_node!(chain_source_esplora, anchor_channels),
692+
new_node(&chain_source_bitcoind, anchor_channels),
693+
new_node(&chain_source_electrsd, anchor_channels),
694+
new_node(&chain_source_esplora, anchor_channels),
702695
];
703696

704697
let (bitcoind, electrs) = (&bitcoind.client, &electrsd.client);

tests/reorg_test.rs

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@ use proptest::prelude::prop;
88
use proptest::proptest;
99

1010
use crate::common::{
11-
expect_event, generate_blocks_and_wait, invalidate_blocks, open_channel,
12-
premine_and_distribute_funds, random_config, setup_bitcoind_and_electrsd, setup_node,
13-
wait_for_outpoint_spend, TestChainSource,
11+
expect_event, generate_blocks_and_wait, invalidate_blocks, new_node, open_channel,
12+
premine_and_distribute_funds, setup_bitcoind_and_electrsd, wait_for_outpoint_spend,
13+
TestChainSource,
1414
};
1515

1616
proptest! {
@@ -23,18 +23,11 @@ proptest! {
2323
let chain_source_electrsd = TestChainSource::Electrum(&electrsd);
2424
let chain_source_esplora = TestChainSource::Esplora(&electrsd);
2525

26-
macro_rules! config_node {
27-
($chain_source: expr, $anchor_channels: expr) => {{
28-
let config_a = random_config($anchor_channels);
29-
let node = setup_node(&$chain_source, config_a, None);
30-
node
31-
}};
32-
}
3326
let anchor_channels = true;
3427
let nodes = vec![
35-
config_node!(chain_source_electrsd, anchor_channels),
36-
config_node!(chain_source_bitcoind, anchor_channels),
37-
config_node!(chain_source_esplora, anchor_channels),
28+
new_node(&chain_source_electrsd, anchor_channels),
29+
new_node(&chain_source_bitcoind, anchor_channels),
30+
new_node(&chain_source_esplora, anchor_channels),
3831
];
3932

4033
let (bitcoind, electrs) = (&bitcoind.client, &electrsd.client);

0 commit comments

Comments
 (0)