Skip to content

Commit 2fae484

Browse files
author
Antoine Riard
committed
-f add ClosureReason tests
1 parent 84be23a commit 2fae484

File tree

7 files changed

+155
-151
lines changed

7 files changed

+155
-151
lines changed

lightning/src/ln/chanmon_update_fail_tests.rs

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ use ln::msgs::{ChannelMessageHandler, ErrorAction, RoutingMessageHandler};
2828
use routing::router::get_route;
2929
use util::config::UserConfig;
3030
use util::enforcing_trait_impls::EnforcingSigner;
31-
use util::events::{Event, MessageSendEvent, MessageSendEventsProvider, PaymentPurpose};
31+
use util::events::{Event, MessageSendEvent, MessageSendEventsProvider, PaymentPurpose, ClosureReason};
3232
use util::errors::APIError;
3333
use util::ser::{ReadableArgs, Writeable};
3434
use util::test_utils::TestBroadcaster;
@@ -81,7 +81,7 @@ fn do_test_simple_monitor_permanent_update_fail(persister_fail: bool) {
8181
// PaymentFailed event
8282

8383
assert_eq!(nodes[0].node.list_channels().len(), 0);
84-
check_closed_event!(nodes[0], 1);
84+
check_closed_event!(nodes[0], 1, ClosureReason::DisconnectedPeer);
8585
}
8686

8787
#[test]
@@ -270,7 +270,7 @@ fn do_test_simple_monitor_temporary_update_fail(disconnect: bool, persister_fail
270270
// PaymentFailed event
271271

272272
assert_eq!(nodes[0].node.list_channels().len(), 0);
273-
check_closed_event!(nodes[0], 1);
273+
check_closed_event!(nodes[0], 1, ClosureReason::DisconnectedPeer);
274274
}
275275

276276
#[test]
@@ -1986,8 +1986,8 @@ fn do_during_funding_monitor_fail(confirm_a_first: bool, restore_b_before_conf:
19861986

19871987
send_payment(&nodes[0], &[&nodes[1]], 8000000);
19881988
close_channel(&nodes[0], &nodes[1], &channel_id, funding_tx, true);
1989-
check_closed_event!(nodes[0], 1);
1990-
check_closed_event!(nodes[1], 1);
1989+
check_closed_event!(nodes[0], 1, ClosureReason::DisconnectedPeer);
1990+
check_closed_event!(nodes[1], 1, ClosureReason::DisconnectedPeer);
19911991
}
19921992

19931993
#[test]
@@ -2616,8 +2616,8 @@ fn test_temporary_error_during_shutdown() {
26162616
assert_eq!(txn_a, txn_b);
26172617
assert_eq!(txn_a.len(), 1);
26182618
check_spends!(txn_a[0], funding_tx);
2619-
check_closed_event!(nodes[1], 1);
2620-
check_closed_event!(nodes[0], 1);
2619+
check_closed_event!(nodes[1], 1, ClosureReason::DisconnectedPeer);
2620+
check_closed_event!(nodes[0], 1, ClosureReason::DisconnectedPeer);
26212621
}
26222622

26232623
#[test]
@@ -2638,7 +2638,7 @@ fn test_permanent_error_during_sending_shutdown() {
26382638
assert!(nodes[0].node.close_channel(&channel_id).is_ok());
26392639
check_closed_broadcast!(nodes[0], true);
26402640
check_added_monitors!(nodes[0], 2);
2641-
check_closed_event!(nodes[0], 1);
2641+
check_closed_event!(nodes[0], 1, ClosureReason::DisconnectedPeer);
26422642
}
26432643

26442644
#[test]
@@ -2661,7 +2661,7 @@ fn test_permanent_error_during_handling_shutdown() {
26612661
nodes[1].node.handle_shutdown(&nodes[0].node.get_our_node_id(), &InitFeatures::known(), &shutdown);
26622662
check_closed_broadcast!(nodes[1], true);
26632663
check_added_monitors!(nodes[1], 2);
2664-
check_closed_event!(nodes[1], 1);
2664+
check_closed_event!(nodes[1], 1, ClosureReason::DisconnectedPeer);
26652665
}
26662666

26672667
#[test]

lightning/src/ln/functional_test_utils.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -766,12 +766,12 @@ macro_rules! check_closed_broadcast {
766766
/// Check that a channel's closing channel event has been issued
767767
#[macro_export]
768768
macro_rules! check_closed_event {
769-
($node: expr, $events: expr) => {{
769+
($node: expr, $events: expr, $reason: expr) => {{
770770
let events = $node.node.get_and_clear_pending_events();
771771
assert_eq!(events.len(), $events);
772772
match events[0] {
773773
Event::ChannelClosed { ref reason, .. } => {
774-
//assert_eq!(reason, $reason);
774+
assert_eq!(*reason, $reason);
775775
},
776776
_ => panic!("Unexpected event"),
777777
}

0 commit comments

Comments
 (0)