Skip to content

Commit b476e05

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 3363c9b commit b476e05

2 files changed

Lines changed: 11 additions & 26 deletions

File tree

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) => {
@@ -419,7 +414,7 @@ pub(crate) fn setup_node_for_async_payments(
419414
},
420415
}
421416

422-
builder.set_async_payments_role(async_payments_role).unwrap();
417+
builder.set_async_payments_role(config.async_payments_role).unwrap();
423418

424419
let node = match config.store_type {
425420
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;
@@ -1317,30 +1316,21 @@ async fn async_payment() {
13171316
config_sender.node_config.node_alias = None;
13181317
config_sender.log_writer =
13191318
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("sender ".to_string())));
1320-
let node_sender = setup_node_for_async_payments(
1321-
&chain_source,
1322-
config_sender,
1323-
Some(AsyncPaymentsRole::Client),
1324-
);
1319+
config_sender.async_payments_role = Some(AsyncPaymentsRole::Client);
1320+
let node_sender = setup_node(&chain_source, config_sender);
13251321

13261322
let mut config_sender_lsp = random_config(true);
13271323
config_sender_lsp.log_writer =
13281324
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("sender_lsp ".to_string())));
1329-
let node_sender_lsp = setup_node_for_async_payments(
1330-
&chain_source,
1331-
config_sender_lsp,
1332-
Some(AsyncPaymentsRole::Server),
1333-
);
1325+
config_sender_lsp.async_payments_role = Some(AsyncPaymentsRole::Server);
1326+
let node_sender_lsp = setup_node(&chain_source, config_sender_lsp);
13341327

13351328
let mut config_receiver_lsp = random_config(true);
13361329
config_receiver_lsp.log_writer =
13371330
TestLogWriter::Custom(Arc::new(MultiNodeLogger::new("receiver_lsp".to_string())));
1331+
config_receiver_lsp.async_payments_role = Some(AsyncPaymentsRole::Server);
13381332

1339-
let node_receiver_lsp = setup_node_for_async_payments(
1340-
&chain_source,
1341-
config_receiver_lsp,
1342-
Some(AsyncPaymentsRole::Server),
1343-
);
1333+
let node_receiver_lsp = setup_node(&chain_source, config_receiver_lsp);
13441334

13451335
let mut config_receiver = random_config(true);
13461336
config_receiver.node_config.listening_addresses = None;

0 commit comments

Comments
 (0)