@@ -17543,7 +17543,6 @@ mod tests {
17543
17543
}
17544
17544
17545
17545
#[test]
17546
- #[rustfmt::skip]
17547
17546
#[cfg(peer_storage)]
17548
17547
fn test_peer_storage() {
17549
17548
let chanmon_cfgs = create_chanmon_cfgs(2);
@@ -17564,8 +17563,10 @@ mod tests {
17564
17563
connect_blocks(&nodes[0], 1);
17565
17564
connect_blocks(&nodes[0], 1);
17566
17565
17567
- let peer_storage_msg_events_node0 = nodes[0].chain_monitor.chain_monitor.get_and_clear_pending_msg_events();
17568
- let peer_storage_msg_events_node1 = nodes[1].chain_monitor.chain_monitor.get_and_clear_pending_msg_events();
17566
+ let peer_storage_msg_events_node0 =
17567
+ nodes[0].chain_monitor.chain_monitor.get_and_clear_pending_msg_events();
17568
+ let peer_storage_msg_events_node1 =
17569
+ nodes[1].chain_monitor.chain_monitor.get_and_clear_pending_msg_events();
17569
17570
assert_ne!(peer_storage_msg_events_node0.len(), 0);
17570
17571
assert_ne!(peer_storage_msg_events_node1.len(), 0);
17571
17572
@@ -17574,7 +17575,7 @@ mod tests {
17574
17575
MessageSendEvent::SendPeerStorage { ref node_id, ref msg } => {
17575
17576
assert_eq!(*node_id, nodes[1].node.get_our_node_id());
17576
17577
nodes[1].node.handle_peer_storage(nodes[0].node.get_our_node_id(), msg.clone());
17577
- }
17578
+ },
17578
17579
_ => panic!("Unexpected event"),
17579
17580
}
17580
17581
}
@@ -17584,7 +17585,7 @@ mod tests {
17584
17585
MessageSendEvent::SendPeerStorage { ref node_id, ref msg } => {
17585
17586
assert_eq!(*node_id, nodes[0].node.get_our_node_id());
17586
17587
nodes[0].node.handle_peer_storage(nodes[1].node.get_our_node_id(), msg.clone());
17587
- }
17588
+ },
17588
17589
_ => panic!("Unexpected event"),
17589
17590
}
17590
17591
}
@@ -17594,14 +17595,41 @@ mod tests {
17594
17595
17595
17596
// Reload Node!
17596
17597
// TODO: Handle the case where we've completely forgotten about an active channel.
17597
- reload_node!(nodes[0], test_default_channel_config(), &nodes_0_serialized, &[&old_state_monitor[..]], persister, chain_monitor, nodes_0_deserialized);
17598
+ reload_node!(
17599
+ nodes[0],
17600
+ test_default_channel_config(),
17601
+ &nodes_0_serialized,
17602
+ &[&old_state_monitor[..]],
17603
+ persister,
17604
+ chain_monitor,
17605
+ nodes_0_deserialized
17606
+ );
17598
17607
17599
- nodes[0].node.peer_connected(nodes[1].node.get_our_node_id(), &msgs::Init {
17600
- features: nodes[1].node.init_features(), networks: None, remote_network_address: None
17601
- }, true).unwrap();
17602
- nodes[1].node.peer_connected(nodes[0].node.get_our_node_id(), &msgs::Init {
17603
- features: nodes[0].node.init_features(), networks: None, remote_network_address: None
17604
- }, false).unwrap();
17608
+ nodes[0]
17609
+ .node
17610
+ .peer_connected(
17611
+ nodes[1].node.get_our_node_id(),
17612
+ &msgs::Init {
17613
+ features: nodes[1].node.init_features(),
17614
+ networks: None,
17615
+ remote_network_address: None,
17616
+ },
17617
+ true,
17618
+ )
17619
+ .unwrap();
17620
+
17621
+ nodes[1]
17622
+ .node
17623
+ .peer_connected(
17624
+ nodes[0].node.get_our_node_id(),
17625
+ &msgs::Init {
17626
+ features: nodes[0].node.init_features(),
17627
+ networks: None,
17628
+ remote_network_address: None,
17629
+ },
17630
+ false,
17631
+ )
17632
+ .unwrap();
17605
17633
17606
17634
let node_1_events = nodes[1].node.get_and_clear_pending_msg_events();
17607
17635
assert_eq!(node_1_events.len(), 2);
@@ -17612,19 +17640,18 @@ mod tests {
17612
17640
17613
17641
match node_0_events[0] {
17614
17642
MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
17615
- assert_eq!(*node_id, nodes[1].node.get_our_node_id());
17616
- // nodes[0] would send a bogus channel reestablish, so there's no need to handle this.
17617
- }
17618
- _ => panic!("Unexpected event"),
17643
+ assert_eq!(*node_id, nodes[1].node.get_our_node_id());
17644
+ // nodes[0] would send a bogus channel reestablish, so there's no need to handle this.
17645
+ },
17646
+ _ => panic!("Unexpected event"),
17619
17647
}
17620
17648
17621
17649
for msg in node_1_events {
17622
17650
if let MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } = msg {
17623
17651
nodes[0].node.handle_channel_reestablish(nodes[1].node.get_our_node_id(), msg);
17624
17652
assert_eq!(*node_id, nodes[0].node.get_our_node_id());
17625
- } else if let MessageSendEvent::SendPeerStorageRetrieval { ref node_id, ref msg } = msg {
17626
- // Should Panic here!
17627
- nodes[0].node.handle_peer_storage_retrieval(nodes[1].node.get_our_node_id(), msg.clone());
17653
+ } else if let MessageSendEvent::SendPeerStorageRetrieval { ref node_id, ref msg } = msg
17654
+ {
17628
17655
assert_eq!(*node_id, nodes[0].node.get_our_node_id());
17629
17656
// Should Panic here!
17630
17657
let res = std::panic::catch_unwind(|| {
0 commit comments