Skip to content

Commit 10db851

Browse files
committed
Move async_payment_role to TestConfig
It's weird to have a special intermediary `setup_node` method if we have `TestConfig` for exactly that reason by now. So we move `async_payment_role` over.
1 parent c99d09d commit 10db851

File tree

2 files changed

+11
-26
lines changed

2 files changed

+11
-26
lines changed

tests/common/mod.rs

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -292,6 +292,7 @@ pub(crate) struct TestConfig {
292292
pub log_writer: TestLogWriter,
293293
pub store_type: TestStoreType,
294294
pub node_entropy: NodeEntropy,
295+
pub async_payments_role: Option<AsyncPaymentsRole>,
295296
}
296297

297298
impl Default for TestConfig {
@@ -302,7 +303,8 @@ impl Default for TestConfig {
302303

303304
let mnemonic = generate_entropy_mnemonic(None);
304305
let node_entropy = NodeEntropy::from_bip39_mnemonic(mnemonic, None);
305-
TestConfig { node_config, log_writer, store_type, node_entropy }
306+
let async_payments_role = None;
307+
TestConfig { node_config, log_writer, store_type, node_entropy, async_payments_role }
306308
}
307309
}
308310

@@ -359,13 +361,6 @@ pub(crate) fn setup_two_nodes_with_store(
359361
}
360362

361363
pub(crate) fn setup_node(chain_source: &TestChainSource, config: TestConfig) -> TestNode {
362-
setup_node_for_async_payments(chain_source, config, None)
363-
}
364-
365-
pub(crate) fn setup_node_for_async_payments(
366-
chain_source: &TestChainSource, config: TestConfig,
367-
async_payments_role: Option<AsyncPaymentsRole>,
368-
) -> TestNode {
369364
setup_builder!(builder, config.node_config);
370365
match chain_source {
371366
TestChainSource::Esplora(electrsd) => {
@@ -417,7 +412,7 @@ pub(crate) fn setup_node_for_async_payments(
417412
},
418413
}
419414

420-
builder.set_async_payments_role(async_payments_role).unwrap();
415+
builder.set_async_payments_role(config.async_payments_role).unwrap();
421416

422417
let node = match config.store_type {
423418
TestStoreType::TestSyncStore => {

tests/integration_tests_rust.rs

Lines changed: 7 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,7 @@ use common::{
2323
expect_splice_pending_event, generate_blocks_and_wait, open_channel, open_channel_push_amt,
2424
premine_and_distribute_funds, premine_blocks, prepare_rbf, random_config,
2525
random_listening_addresses, setup_bitcoind_and_electrsd, setup_builder, setup_node,
26-
setup_node_for_async_payments, setup_two_nodes, wait_for_tx, TestChainSource, TestStoreType,
27-
TestSyncStore,
26+
setup_two_nodes, wait_for_tx, TestChainSource, TestStoreType, TestSyncStore,
2827
};
2928
use ldk_node::config::{AsyncPaymentsRole, EsploraSyncConfig};
3029
use ldk_node::entropy::NodeEntropy;
@@ -1315,30 +1314,21 @@ async fn async_payment() {
13151314
config_sender.node_config.node_alias = None;
13161315
config_sender.log_writer =
13171316
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("sender ".to_string())));
1318-
let node_sender = setup_node_for_async_payments(
1319-
&chain_source,
1320-
config_sender,
1321-
Some(AsyncPaymentsRole::Client),
1322-
);
1317+
config_sender.async_payments_role = Some(AsyncPaymentsRole::Client);
1318+
let node_sender = setup_node(&chain_source, config_sender);
13231319

13241320
let mut config_sender_lsp = random_config(true);
13251321
config_sender_lsp.log_writer =
13261322
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("sender_lsp ".to_string())));
1327-
let node_sender_lsp = setup_node_for_async_payments(
1328-
&chain_source,
1329-
config_sender_lsp,
1330-
Some(AsyncPaymentsRole::Server),
1331-
);
1323+
config_sender_lsp.async_payments_role = Some(AsyncPaymentsRole::Server);
1324+
let node_sender_lsp = setup_node(&chain_source, config_sender_lsp);
13321325

13331326
let mut config_receiver_lsp = random_config(true);
13341327
config_receiver_lsp.log_writer =
13351328
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("receiver_lsp".to_string())));
1329+
config_receiver_lsp.async_payments_role = Some(AsyncPaymentsRole::Server);
13361330

1337-
let node_receiver_lsp = setup_node_for_async_payments(
1338-
&chain_source,
1339-
config_receiver_lsp,
1340-
Some(AsyncPaymentsRole::Server),
1341-
);
1331+
let node_receiver_lsp = setup_node(&chain_source, config_receiver_lsp);
13421332

13431333
let mut config_receiver = random_config(true);
13441334
config_receiver.node_config.listening_addresses = None;

0 commit comments

Comments
 (0)