Skip to content

Commit e07716d

Browse files
committed
test: clean up MockLogger and LogWriter setup
- Revert MockLogger configuration and assertions - Remove clutter from setting up LogWriter across all tests
1 parent 9f87227 commit e07716d

File tree

2 files changed

+22
-72
lines changed

2 files changed

+22
-72
lines changed

tests/common/mod.rs

Lines changed: 3 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -274,10 +274,6 @@ impl MockLogger {
274274
}
275275
}
276276

277-
/// [`MockLogger`] as `log` logger - destination for [`Writer::LogFacadeWriter`]
278-
/// to write logs to.
279-
///
280-
/// [`Writer::LogFacadeWriter`]: ldk_node::logger::Writer::LogFacadeWriter
281277
impl Log for MockLogger {
282278
fn log(&self, record: &log::Record) {
283279
let message = format!(
@@ -333,11 +329,11 @@ pub(crate) use setup_builder;
333329

334330
pub(crate) fn setup_two_nodes(
335331
chain_source: &TestChainSource, allow_0conf: bool, anchor_channels: bool,
336-
anchors_trusted_no_reserve: bool, log_writer: TestLogWriter,
332+
anchors_trusted_no_reserve: bool,
337333
) -> (TestNode, TestNode) {
338334
println!("== Node A ==");
339335
let config_a = random_config(anchor_channels);
340-
let node_a = setup_node(chain_source, config_a, None, log_writer.clone());
336+
let node_a = setup_node(chain_source, config_a, None);
341337

342338
println!("\n== Node B ==");
343339
let mut config_b = random_config(anchor_channels);
@@ -352,13 +348,12 @@ pub(crate) fn setup_two_nodes(
352348
.trusted_peers_no_reserve
353349
.push(node_a.node_id());
354350
}
355-
let node_b = setup_node(chain_source, config_b, None, log_writer);
351+
let node_b = setup_node(chain_source, config_b, None);
356352
(node_a, node_b)
357353
}
358354

359355
pub(crate) fn setup_node(
360356
chain_source: &TestChainSource, config: Config, seed_bytes: Option<Vec<u8>>,
361-
log_writer: TestLogWriter,
362357
) -> TestNode {
363358
setup_builder!(builder, config);
364359
match chain_source {
@@ -379,18 +374,6 @@ pub(crate) fn setup_node(
379374
},
380375
}
381376

382-
match log_writer {
383-
TestLogWriter::File(fs_config) => {
384-
builder.set_filesystem_logger(fs_config);
385-
},
386-
TestLogWriter::LogFacade(log_level) => {
387-
builder.set_log_facade_logger(log_level);
388-
},
389-
TestLogWriter::Custom(log_writer) => {
390-
builder.set_custom_logger(log_writer);
391-
},
392-
}
393-
394377
if let Some(seed) = seed_bytes {
395378
builder.set_entropy_seed_bytes(seed).unwrap();
396379
}

tests/integration_tests_rust.rs

Lines changed: 19 additions & 52 deletions
Original file line numberDiff line numberDiff line change
@@ -9,14 +9,12 @@ mod common;
99

1010
use common::{
1111
do_channel_full_cycle, expect_channel_ready_event, expect_event, expect_payment_received_event,
12-
expect_payment_successful_event, generate_blocks_and_wait, init_custom_logger, init_log_logger,
13-
open_channel, premine_and_distribute_funds, random_config, setup_bitcoind_and_electrsd,
14-
setup_builder, setup_node, setup_two_nodes, wait_for_tx, TestChainSource, TestLogWriter,
15-
TestSyncStore,
12+
expect_payment_successful_event, generate_blocks_and_wait, open_channel,
13+
premine_and_distribute_funds, random_config, setup_bitcoind_and_electrsd, setup_builder,
14+
setup_node, setup_two_nodes, wait_for_tx, TestChainSource, TestSyncStore,
1615
};
1716

1817
use ldk_node::config::{EsploraSyncConfig, FilesystemLoggerConfig};
19-
use ldk_node::logger::LogLevel;
2018
use ldk_node::payment::{PaymentKind, QrPaymentResult, SendingParameters};
2119
use ldk_node::{Builder, Event, NodeError};
2220

@@ -34,62 +32,55 @@ use std::sync::Arc;
3432
fn channel_full_cycle() {
3533
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
3634
let chain_source = TestChainSource::Esplora(&electrsd);
37-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
38-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
35+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
3936
do_channel_full_cycle(node_a, node_b, &bitcoind.client, &electrsd.client, false, true, false);
4037
}
4138

4239
#[test]
4340
fn channel_full_cycle_bitcoind() {
4441
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
4542
let chain_source = TestChainSource::BitcoindRpc(&bitcoind);
46-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
47-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
43+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
4844
do_channel_full_cycle(node_a, node_b, &bitcoind.client, &electrsd.client, false, true, false);
4945
}
5046

5147
#[test]
5248
fn channel_full_cycle_force_close() {
5349
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
5450
let chain_source = TestChainSource::Esplora(&electrsd);
55-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
56-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
51+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
5752
do_channel_full_cycle(node_a, node_b, &bitcoind.client, &electrsd.client, false, true, true);
5853
}
5954

6055
#[test]
6156
fn channel_full_cycle_force_close_trusted_no_reserve() {
6257
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
6358
let chain_source = TestChainSource::Esplora(&electrsd);
64-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
65-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, true, log_writer);
59+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, true);
6660
do_channel_full_cycle(node_a, node_b, &bitcoind.client, &electrsd.client, false, true, true);
6761
}
6862

6963
#[test]
7064
fn channel_full_cycle_0conf() {
7165
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
7266
let chain_source = TestChainSource::Esplora(&electrsd);
73-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
74-
let (node_a, node_b) = setup_two_nodes(&chain_source, true, true, false, log_writer);
67+
let (node_a, node_b) = setup_two_nodes(&chain_source, true, true, false);
7568
do_channel_full_cycle(node_a, node_b, &bitcoind.client, &electrsd.client, true, true, false)
7669
}
7770

7871
#[test]
7972
fn channel_full_cycle_legacy_staticremotekey() {
8073
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
8174
let chain_source = TestChainSource::Esplora(&electrsd);
82-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
83-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, false, false, log_writer);
75+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, false, false);
8476
do_channel_full_cycle(node_a, node_b, &bitcoind.client, &electrsd.client, false, false, false);
8577
}
8678

8779
#[test]
8880
fn channel_open_fails_when_funds_insufficient() {
8981
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
9082
let chain_source = TestChainSource::Esplora(&electrsd);
91-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
92-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
83+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
9384

9485
let addr_a = node_a.onchain_payment().new_address().unwrap();
9586
let addr_b = node_b.onchain_payment().new_address().unwrap();
@@ -289,8 +280,7 @@ fn start_stop_reinit() {
289280
fn onchain_spend_receive() {
290281
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
291282
let chain_source = TestChainSource::Esplora(&electrsd);
292-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
293-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
283+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
294284

295285
let addr_a = node_a.onchain_payment().new_address().unwrap();
296286
let addr_b = node_b.onchain_payment().new_address().unwrap();
@@ -391,9 +381,7 @@ fn onchain_wallet_recovery() {
391381
let seed_bytes = vec![42u8; 64];
392382

393383
let original_config = random_config(true);
394-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
395-
let original_node =
396-
setup_node(&chain_source, original_config, Some(seed_bytes.clone()), log_writer);
384+
let original_node = setup_node(&chain_source, original_config, Some(seed_bytes.clone()));
397385

398386
let premine_amount_sat = 100_000;
399387

@@ -438,8 +426,7 @@ fn onchain_wallet_recovery() {
438426

439427
// Now we start from scratch, only the seed remains the same.
440428
let recovered_config = random_config(true);
441-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
442-
let recovered_node = setup_node(&chain_source, recovered_config, Some(seed_bytes), log_writer);
429+
let recovered_node = setup_node(&chain_source, recovered_config, Some(seed_bytes));
443430

444431
recovered_node.sync_wallets().unwrap();
445432
assert_eq!(
@@ -482,8 +469,7 @@ fn sign_verify_msg() {
482469
let (_bitcoind, electrsd) = setup_bitcoind_and_electrsd();
483470
let config = random_config(true);
484471
let chain_source = TestChainSource::Esplora(&electrsd);
485-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
486-
let node = setup_node(&chain_source, config, None, log_writer);
472+
let node = setup_node(&chain_source, config, None);
487473

488474
// Tests arbitrary message signing and later verification
489475
let msg = "OK computer".as_bytes();
@@ -501,8 +487,7 @@ fn connection_restart_behavior() {
501487
fn do_connection_restart_behavior(persist: bool) {
502488
let (_bitcoind, electrsd) = setup_bitcoind_and_electrsd();
503489
let chain_source = TestChainSource::Esplora(&electrsd);
504-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
505-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, false, false, log_writer);
490+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, false, false);
506491

507492
let node_id_a = node_a.node_id();
508493
let node_id_b = node_b.node_id();
@@ -554,8 +539,7 @@ fn do_connection_restart_behavior(persist: bool) {
554539
fn concurrent_connections_succeed() {
555540
let (_bitcoind, electrsd) = setup_bitcoind_and_electrsd();
556541
let chain_source = TestChainSource::Esplora(&electrsd);
557-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
558-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
542+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
559543

560544
let node_a = Arc::new(node_a);
561545
let node_b = Arc::new(node_b);
@@ -586,8 +570,7 @@ fn concurrent_connections_succeed() {
586570
fn simple_bolt12_send_receive() {
587571
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
588572
let chain_source = TestChainSource::Esplora(&electrsd);
589-
let log_writer = TestLogWriter::File(FilesystemLoggerConfig::default());
590-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
573+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
591574

592575
let address_a = node_a.onchain_payment().new_address().unwrap();
593576
let premine_amount_sat = 5_000_000;
@@ -796,10 +779,7 @@ fn generate_bip21_uri() {
796779
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
797780
let chain_source = TestChainSource::Esplora(&electrsd);
798781

799-
// Setup custom logger.
800-
let mock_logger = init_custom_logger();
801-
let log_writer = TestLogWriter::Custom(mock_logger.clone());
802-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
782+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
803783

804784
let address_a = node_a.onchain_payment().new_address().unwrap();
805785
let premined_sats = 5_000_000;
@@ -835,21 +815,14 @@ fn generate_bip21_uri() {
835815
},
836816
Err(e) => panic!("Failed to generate URI: {:?}", e),
837817
}
838-
839-
let logs = mock_logger.retrieve_logs();
840-
let last_log_entry = logs.last().unwrap();
841-
assert!(last_log_entry.contains("[INFO] Invoice created:"));
842818
}
843819

844820
#[test]
845821
fn unified_qr_send_receive() {
846822
let (bitcoind, electrsd) = setup_bitcoind_and_electrsd();
847823
let chain_source = TestChainSource::Esplora(&electrsd);
848824

849-
// Setup `log` facade logger.
850-
let mock_logger = init_log_logger(log::LevelFilter::Trace);
851-
let log_writer = TestLogWriter::LogFacade(LogLevel::Trace);
852-
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false, log_writer);
825+
let (node_a, node_b) = setup_two_nodes(&chain_source, false, true, false);
853826

854827
let address_a = node_a.onchain_payment().new_address().unwrap();
855828
let premined_sats = 5_000_000;
@@ -955,10 +928,4 @@ fn unified_qr_send_receive() {
955928

956929
assert_eq!(node_b.list_balances().total_onchain_balance_sats, 800_000);
957930
assert_eq!(node_b.list_balances().total_lightning_balance_sats, 200_000);
958-
959-
assert!(mock_logger
960-
.retrieve_logs()
961-
.last()
962-
.unwrap()
963-
.contains("Incremental sync of on-chain wallet finished"),);
964931
}

0 commit comments

Comments
 (0)