Skip to content

Commit 431ac3b

Browse files
nit: Refactor node setup in tests to use new_node function for improved clarity
1 parent 8bd7f4d commit 431ac3b

File tree

3 files changed

+17
-26
lines changed

3 files changed

+17
-26
lines changed

tests/common/mod.rs

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -308,6 +308,11 @@ pub(crate) fn setup_two_nodes(
308308
(node_a, node_b)
309309
}
310310

311+
pub(crate) fn new_node(chain_source: &TestChainSource, anchor_channels: bool) -> TestNode {
312+
let config = random_config(anchor_channels);
313+
setup_node(chain_source, config, None)
314+
}
315+
311316
pub(crate) fn setup_node(
312317
chain_source: &TestChainSource, config: TestConfig, seed_bytes: Option<Vec<u8>>,
313318
) -> TestNode {

tests/integration_tests_rust.rs

Lines changed: 6 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ use common::{
1313
expect_payment_claimable_event, expect_payment_received_event, expect_payment_successful_event,
1414
generate_blocks_and_wait,
1515
logging::{init_log_logger, validate_log_entry, TestLogWriter},
16-
open_channel, premine_and_distribute_funds, premine_blocks, prepare_rbf, random_config,
17-
random_listening_addresses, setup_bitcoind_and_electrsd, setup_builder, setup_node,
18-
setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
16+
new_node, open_channel, premine_and_distribute_funds, premine_blocks, prepare_rbf,
17+
random_config, random_listening_addresses, setup_bitcoind_and_electrsd, setup_builder,
18+
setup_node, setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
1919
};
2020

2121
use ldk_node::config::EsploraSyncConfig;
@@ -690,18 +690,11 @@ fn run_rbf_test(is_insert_block: bool) {
690690
let chain_source_electrsd = TestChainSource::Electrum(&electrsd);
691691
let chain_source_esplora = TestChainSource::Esplora(&electrsd);
692692

693-
macro_rules! config_node {
694-
($chain_source: expr, $anchor_channels: expr) => {{
695-
let config_a = random_config($anchor_channels);
696-
let node = setup_node(&$chain_source, config_a, None);
697-
node
698-
}};
699-
}
700693
let anchor_channels = false;
701694
let nodes = vec![
702-
config_node!(chain_source_electrsd, anchor_channels),
703-
config_node!(chain_source_bitcoind, anchor_channels),
704-
config_node!(chain_source_esplora, anchor_channels),
695+
new_node(&chain_source_bitcoind, anchor_channels),
696+
new_node(&chain_source_electrsd, anchor_channels),
697+
new_node(&chain_source_esplora, anchor_channels),
705698
];
706699

707700
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
@@ -6,9 +6,9 @@ use proptest::{prelude::prop, proptest};
66
use std::collections::HashMap;
77

88
use crate::common::{
9-
expect_event, generate_blocks_and_wait, invalidate_blocks, open_channel,
10-
premine_and_distribute_funds, random_config, setup_bitcoind_and_electrsd, setup_node,
11-
wait_for_outpoint_spend, TestChainSource,
9+
expect_event, generate_blocks_and_wait, invalidate_blocks, new_node, open_channel,
10+
premine_and_distribute_funds, setup_bitcoind_and_electrsd, wait_for_outpoint_spend,
11+
TestChainSource,
1212
};
1313

1414
proptest! {
@@ -21,18 +21,11 @@ proptest! {
2121
let chain_source_electrsd = TestChainSource::Electrum(&electrsd);
2222
let chain_source_esplora = TestChainSource::Esplora(&electrsd);
2323

24-
macro_rules! config_node {
25-
($chain_source: expr, $anchor_channels: expr) => {{
26-
let config_a = random_config($anchor_channels);
27-
let node = setup_node(&$chain_source, config_a, None);
28-
node
29-
}};
30-
}
3124
let anchor_channels = true;
3225
let nodes = vec![
33-
config_node!(chain_source_electrsd, anchor_channels),
34-
config_node!(chain_source_bitcoind, anchor_channels),
35-
config_node!(chain_source_esplora, anchor_channels),
26+
new_node(&chain_source_electrsd, anchor_channels),
27+
new_node(&chain_source_bitcoind, anchor_channels),
28+
new_node(&chain_source_esplora, anchor_channels),
3629
];
3730

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

0 commit comments

Comments
 (0)