Skip to content

Commit cc9f907

Browse files
nit: introduce setup_node and rename setup_node to setup_node_from_config
Introduce setup_node for clearer node creation. Rename old setup_node to setup_node_from_config for custom configs.
1 parent e4bb615 commit cc9f907

File tree

3 files changed

+26
-34
lines changed

3 files changed

+26
-34
lines changed

tests/common/mod.rs

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,7 @@ pub(crate) fn setup_two_nodes(
285285
) -> (TestNode, TestNode) {
286286
println!("== Node A ==");
287287
let config_a = random_config(anchor_channels);
288-
let node_a = setup_node(chain_source, config_a, None);
288+
let node_a = setup_node_from_config(chain_source, config_a, None);
289289

290290
println!("\n== Node B ==");
291291
let mut config_b = random_config(anchor_channels);
@@ -301,11 +301,16 @@ pub(crate) fn setup_two_nodes(
301301
.trusted_peers_no_reserve
302302
.push(node_a.node_id());
303303
}
304-
let node_b = setup_node(chain_source, config_b, None);
304+
let node_b = setup_node_from_config(chain_source, config_b, None);
305305
(node_a, node_b)
306306
}
307307

308-
pub(crate) fn setup_node(
308+
pub(crate) fn setup_node(chain_source: &TestChainSource, anchor_channels: bool) -> TestNode {
309+
let config = random_config(anchor_channels);
310+
setup_node_from_config(chain_source, config, None)
311+
}
312+
313+
pub(crate) fn setup_node_from_config(
309314
chain_source: &TestChainSource, config: TestConfig, seed_bytes: Option<Vec<u8>>,
310315
) -> TestNode {
311316
setup_node_for_async_payments(chain_source, config, seed_bytes, None)

tests/integration_tests_rust.rs

Lines changed: 13 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ use common::{
2323
generate_blocks_and_wait, open_channel, open_channel_push_amt, premine_and_distribute_funds,
2424
premine_blocks, prepare_rbf, random_config, random_listening_addresses,
2525
setup_bitcoind_and_electrsd, setup_builder, setup_node, setup_node_for_async_payments,
26-
setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
26+
setup_node_from_config, setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
2727
};
2828
use ldk_node::config::{AsyncPaymentsRole, EsploraSyncConfig};
2929
use ldk_node::liquidity::LSPS2ServiceConfig;
@@ -596,7 +596,8 @@ fn onchain_wallet_recovery() {
596596
let seed_bytes = vec![42u8; 64];
597597

598598
let original_config = random_config(true);
599-
let original_node = setup_node(&chain_source, original_config, Some(seed_bytes.clone()));
599+
let original_node =
600+
setup_node_from_config(&chain_source, original_config, Some(seed_bytes.clone()));
600601

601602
let premine_amount_sat = 100_000;
602603

@@ -635,7 +636,7 @@ fn onchain_wallet_recovery() {
635636

636637
// Now we start from scratch, only the seed remains the same.
637638
let recovered_config = random_config(true);
638-
let recovered_node = setup_node(&chain_source, recovered_config, Some(seed_bytes));
639+
let recovered_node = setup_node_from_config(&chain_source, recovered_config, Some(seed_bytes));
639640

640641
recovered_node.sync_wallets().unwrap();
641642
assert_eq!(
@@ -686,18 +687,11 @@ fn run_rbf_test(is_insert_block: bool) {
686687
let chain_source_electrsd = TestChainSource::Electrum(&electrsd);
687688
let chain_source_esplora = TestChainSource::Esplora(&electrsd);
688689

689-
macro_rules! config_node {
690-
($chain_source: expr, $anchor_channels: expr) => {{
691-
let config_a = random_config($anchor_channels);
692-
let node = setup_node(&$chain_source, config_a, None);
693-
node
694-
}};
695-
}
696690
let anchor_channels = false;
697691
let nodes = vec![
698-
config_node!(chain_source_electrsd, anchor_channels),
699-
config_node!(chain_source_bitcoind, anchor_channels),
700-
config_node!(chain_source_esplora, anchor_channels),
692+
setup_node(&chain_source_bitcoind, anchor_channels),
693+
setup_node(&chain_source_electrsd, anchor_channels),
694+
setup_node(&chain_source_esplora, anchor_channels),
701695
];
702696

703697
let (bitcoind, electrs) = (&bitcoind.client, &electrsd.client);
@@ -807,7 +801,7 @@ fn sign_verify_msg() {
807801
let (_bitcoind, electrsd) = setup_bitcoind_and_electrsd();
808802
let config = random_config(true);
809803
let chain_source = TestChainSource::Esplora(&electrsd);
810-
let node = setup_node(&chain_source, config, None);
804+
let node = setup_node_from_config(&chain_source, config, None);
811805

812806
// Tests arbitrary message signing and later verification
813807
let msg = "OK computer".as_bytes();
@@ -1174,7 +1168,7 @@ fn async_payment() {
11741168
config_receiver.node_config.node_alias = None;
11751169
config_receiver.log_writer =
11761170
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("receiver ".to_string())));
1177-
let node_receiver = setup_node(&chain_source, config_receiver, None);
1171+
let node_receiver = setup_node_from_config(&chain_source, config_receiver, None);
11781172

11791173
let address_sender = node_sender.onchain_payment().new_address().unwrap();
11801174
let address_sender_lsp = node_sender_lsp.onchain_payment().new_address().unwrap();
@@ -1296,8 +1290,8 @@ fn test_node_announcement_propagation() {
12961290
config_b.node_config.listening_addresses = Some(node_b_listening_addresses.clone());
12971291
config_b.node_config.announcement_addresses = None;
12981292

1299-
let node_a = setup_node(&chain_source, config_a, None);
1300-
let node_b = setup_node(&chain_source, config_b, None);
1293+
let node_a = setup_node_from_config(&chain_source, config_a, None);
1294+
let node_b = setup_node_from_config(&chain_source, config_b, None);
13011295

13021296
let address_a = node_a.onchain_payment().new_address().unwrap();
13031297
let premine_amount_sat = 5_000_000;
@@ -1753,7 +1747,7 @@ fn facade_logging() {
17531747
config.log_writer = TestLogWriter::LogFacade;
17541748

17551749
println!("== Facade logging starts ==");
1756-
let _node = setup_node(&chain_source, config, None);
1750+
let _node = setup_node_from_config(&chain_source, config, None);
17571751

17581752
assert!(!logger.retrieve_logs().is_empty());
17591753
for (_, entry) in logger.retrieve_logs().iter().enumerate() {
@@ -1834,6 +1828,6 @@ async fn drop_in_async_context() {
18341828
let seed_bytes = vec![42u8; 64];
18351829

18361830
let config = random_config(true);
1837-
let node = setup_node(&chain_source, config, Some(seed_bytes));
1831+
let node = setup_node_from_config(&chain_source, config, Some(seed_bytes));
18381832
node.stop().unwrap();
18391833
}

tests/reorg_test.rs

Lines changed: 5 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,8 @@ use proptest::proptest;
99

1010
use crate::common::{
1111
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,
12+
premine_and_distribute_funds, setup_bitcoind_and_electrsd, setup_node, 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+
setup_node(&chain_source_electrsd, anchor_channels),
29+
setup_node(&chain_source_bitcoind, anchor_channels),
30+
setup_node(&chain_source_esplora, anchor_channels),
3831
];
3932

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

0 commit comments

Comments
 (0)