@@ -1126,8 +1126,8 @@ fn test_monitor_update_fail_reestablish() {
11261126 nodes[ 0 ] . node . peer_connected ( & nodes[ 1 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
11271127 nodes[ 1 ] . node . peer_connected ( & nodes[ 0 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
11281128
1129- let as_reestablish = get_event_msg ! ( nodes[ 0 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 1 ] . node . get_our_node_id ( ) ) ;
1130- let bs_reestablish = get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 0 ] . node . get_our_node_id ( ) ) ;
1129+ let as_reestablish = get_chan_reestablish_msgs ! ( nodes[ 0 ] , nodes[ 1 ] ) . pop ( ) . unwrap ( ) ;
1130+ let bs_reestablish = get_chan_reestablish_msgs ! ( nodes[ 1 ] , nodes[ 0 ] ) . pop ( ) . unwrap ( ) ;
11311131
11321132 nodes[ 0 ] . node . handle_channel_reestablish ( & nodes[ 1 ] . node . get_our_node_id ( ) , & bs_reestablish) ;
11331133
@@ -1145,8 +1145,8 @@ fn test_monitor_update_fail_reestablish() {
11451145 nodes[ 0 ] . node . peer_connected ( & nodes[ 1 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
11461146 nodes[ 1 ] . node . peer_connected ( & nodes[ 0 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
11471147
1148- assert ! ( as_reestablish == get_event_msg !( nodes[ 0 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 1 ] . node . get_our_node_id ( ) ) ) ;
1149- assert ! ( bs_reestablish == get_event_msg !( nodes[ 1 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 0 ] . node . get_our_node_id ( ) ) ) ;
1148+ assert_eq ! ( get_chan_reestablish_msgs !( nodes[ 0 ] , nodes[ 1 ] ) . pop ( ) . unwrap ( ) , as_reestablish ) ;
1149+ assert_eq ! ( get_chan_reestablish_msgs !( nodes[ 1 ] , nodes[ 0 ] ) . pop ( ) . unwrap ( ) , bs_reestablish ) ;
11501150
11511151 nodes[ 0 ] . node . handle_channel_reestablish ( & nodes[ 1 ] . node . get_our_node_id ( ) , & bs_reestablish) ;
11521152 assert_eq ! (
@@ -1319,8 +1319,8 @@ fn claim_while_disconnected_monitor_update_fail() {
13191319 nodes[ 0 ] . node . peer_connected ( & nodes[ 1 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
13201320 nodes[ 1 ] . node . peer_connected ( & nodes[ 0 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
13211321
1322- let as_reconnect = get_event_msg ! ( nodes[ 0 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 1 ] . node . get_our_node_id ( ) ) ;
1323- let bs_reconnect = get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 0 ] . node . get_our_node_id ( ) ) ;
1322+ let as_reconnect = get_chan_reestablish_msgs ! ( nodes[ 0 ] , nodes[ 1 ] ) . pop ( ) . unwrap ( ) ;
1323+ let bs_reconnect = get_chan_reestablish_msgs ! ( nodes[ 1 ] , nodes[ 0 ] ) . pop ( ) . unwrap ( ) ;
13241324
13251325 nodes[ 0 ] . node . handle_channel_reestablish ( & nodes[ 1 ] . node . get_our_node_id ( ) , & bs_reconnect) ;
13261326 let _as_channel_update = get_event_msg ! ( nodes[ 0 ] , MessageSendEvent :: SendChannelUpdate , nodes[ 1 ] . node. get_our_node_id( ) ) ;
@@ -1451,8 +1451,8 @@ fn monitor_failed_no_reestablish_response() {
14511451 nodes[ 0 ] . node . peer_connected ( & nodes[ 1 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
14521452 nodes[ 1 ] . node . peer_connected ( & nodes[ 0 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: empty ( ) , remote_network_address : None } ) ;
14531453
1454- let as_reconnect = get_event_msg ! ( nodes[ 0 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 1 ] . node . get_our_node_id ( ) ) ;
1455- let bs_reconnect = get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 0 ] . node . get_our_node_id ( ) ) ;
1454+ let as_reconnect = get_chan_reestablish_msgs ! ( nodes[ 0 ] , nodes[ 1 ] ) . pop ( ) . unwrap ( ) ;
1455+ let bs_reconnect = get_chan_reestablish_msgs ! ( nodes[ 1 ] , nodes[ 0 ] ) . pop ( ) . unwrap ( ) ;
14561456
14571457 nodes[ 1 ] . node . handle_channel_reestablish ( & nodes[ 0 ] . node . get_our_node_id ( ) , & as_reconnect) ;
14581458 let _bs_channel_update = get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelUpdate , nodes[ 0 ] . node. get_our_node_id( ) ) ;
@@ -2032,9 +2032,9 @@ fn test_pending_update_fee_ack_on_reconnect() {
20322032 nodes[ 1 ] . node . peer_disconnected ( & nodes[ 0 ] . node . get_our_node_id ( ) , false ) ;
20332033
20342034 nodes[ 0 ] . node . peer_connected ( & nodes[ 1 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: known ( ) , remote_network_address : None } ) ;
2035- let as_connect_msg = get_event_msg ! ( nodes[ 0 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 1 ] . node . get_our_node_id ( ) ) ;
2035+ let as_connect_msg = get_chan_reestablish_msgs ! ( nodes[ 0 ] , nodes[ 1 ] ) . pop ( ) . unwrap ( ) ;
20362036 nodes[ 1 ] . node . peer_connected ( & nodes[ 0 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: known ( ) , remote_network_address : None } ) ;
2037- let bs_connect_msg = get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 0 ] . node . get_our_node_id ( ) ) ;
2037+ let bs_connect_msg = get_chan_reestablish_msgs ! ( nodes[ 1 ] , nodes[ 0 ] ) . pop ( ) . unwrap ( ) ;
20382038
20392039 nodes[ 1 ] . node . handle_channel_reestablish ( & nodes[ 0 ] . node . get_our_node_id ( ) , & as_connect_msg) ;
20402040 let bs_resend_msgs = nodes[ 1 ] . node . get_and_clear_pending_msg_events ( ) ;
@@ -2160,9 +2160,9 @@ fn do_update_fee_resend_test(deliver_update: bool, parallel_updates: bool) {
21602160 nodes[ 1 ] . node . peer_disconnected ( & nodes[ 0 ] . node . get_our_node_id ( ) , false ) ;
21612161
21622162 nodes[ 0 ] . node . peer_connected ( & nodes[ 1 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: known ( ) , remote_network_address : None } ) ;
2163- let as_connect_msg = get_event_msg ! ( nodes[ 0 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 1 ] . node . get_our_node_id ( ) ) ;
2163+ let as_connect_msg = get_chan_reestablish_msgs ! ( nodes[ 0 ] , nodes[ 1 ] ) . pop ( ) . unwrap ( ) ;
21642164 nodes[ 1 ] . node . peer_connected ( & nodes[ 0 ] . node . get_our_node_id ( ) , & msgs:: Init { features : InitFeatures :: known ( ) , remote_network_address : None } ) ;
2165- let bs_connect_msg = get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelReestablish , nodes[ 0 ] . node . get_our_node_id ( ) ) ;
2165+ let bs_connect_msg = get_chan_reestablish_msgs ! ( nodes[ 1 ] , nodes[ 0 ] ) . pop ( ) . unwrap ( ) ;
21662166
21672167 nodes[ 1 ] . node . handle_channel_reestablish ( & nodes[ 0 ] . node . get_our_node_id ( ) , & as_connect_msg) ;
21682168 get_event_msg ! ( nodes[ 1 ] , MessageSendEvent :: SendChannelUpdate , nodes[ 0 ] . node. get_our_node_id( ) ) ;
0 commit comments