@@ -17543,7 +17543,6 @@ mod tests {
1754317543 }
1754417544
1754517545 #[test]
17546- #[rustfmt::skip]
1754717546 #[cfg(peer_storage)]
1754817547 fn test_peer_storage() {
1754917548 let chanmon_cfgs = create_chanmon_cfgs(2);
@@ -17564,8 +17563,10 @@ mod tests {
1756417563 connect_blocks(&nodes[0], 1);
1756517564 connect_blocks(&nodes[0], 1);
1756617565
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();
1756917570 assert_ne!(peer_storage_msg_events_node0.len(), 0);
1757017571 assert_ne!(peer_storage_msg_events_node1.len(), 0);
1757117572
@@ -17574,7 +17575,7 @@ mod tests {
1757417575 MessageSendEvent::SendPeerStorage { ref node_id, ref msg } => {
1757517576 assert_eq!(*node_id, nodes[1].node.get_our_node_id());
1757617577 nodes[1].node.handle_peer_storage(nodes[0].node.get_our_node_id(), msg.clone());
17577- }
17578+ },
1757817579 _ => panic!("Unexpected event"),
1757917580 }
1758017581 }
@@ -17584,7 +17585,7 @@ mod tests {
1758417585 MessageSendEvent::SendPeerStorage { ref node_id, ref msg } => {
1758517586 assert_eq!(*node_id, nodes[0].node.get_our_node_id());
1758617587 nodes[0].node.handle_peer_storage(nodes[1].node.get_our_node_id(), msg.clone());
17587- }
17588+ },
1758817589 _ => panic!("Unexpected event"),
1758917590 }
1759017591 }
@@ -17594,14 +17595,41 @@ mod tests {
1759417595
1759517596 // Reload Node!
1759617597 // 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+ );
1759817607
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();
1760517633
1760617634 let node_1_events = nodes[1].node.get_and_clear_pending_msg_events();
1760717635 assert_eq!(node_1_events.len(), 2);
@@ -17612,19 +17640,18 @@ mod tests {
1761217640
1761317641 match node_0_events[0] {
1761417642 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"),
1761917647 }
1762017648
1762117649 for msg in node_1_events {
1762217650 if let MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } = msg {
1762317651 nodes[0].node.handle_channel_reestablish(nodes[1].node.get_our_node_id(), msg);
1762417652 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+ {
1762817655 assert_eq!(*node_id, nodes[0].node.get_our_node_id());
1762917656 // Should Panic here!
1763017657 let res = std::panic::catch_unwind(|| {
0 commit comments