@@ -3570,15 +3570,19 @@ mod tests {
35703570 let cfgs = create_peermgr_cfgs ( 2 ) ;
35713571 let peers = create_network ( 2 , & cfgs) ;
35723572
3573+ let chan_handler = peers[ handler & 1 ] . message_handler . chan_handler ;
3574+ let route_handler = peers[ handler & 1 ] . message_handler . route_handler ;
3575+ let custom_message_handler = peers[ handler & 1 ] . message_handler . custom_message_handler ;
3576+
35733577 match handler & !1 {
35743578 0 => {
3575- peers [ handler & 1 ] . message_handler . chan_handler . conn_tracker . fail_connections . store ( true , Ordering :: Release ) ;
3579+ chan_handler. conn_tracker . fail_connections . store ( true , Ordering :: Release ) ;
35763580 }
35773581 2 => {
3578- peers [ handler & 1 ] . message_handler . route_handler . conn_tracker . fail_connections . store ( true , Ordering :: Release ) ;
3582+ route_handler. conn_tracker . fail_connections . store ( true , Ordering :: Release ) ;
35793583 }
35803584 4 => {
3581- peers [ handler & 1 ] . message_handler . custom_message_handler . conn_tracker . fail_connections . store ( true , Ordering :: Release ) ;
3585+ custom_message_handler. conn_tracker . fail_connections . store ( true , Ordering :: Release ) ;
35823586 }
35833587 _ => panic ! ( ) ,
35843588 }
@@ -3591,13 +3595,13 @@ mod tests {
35913595 assert ! ( peers[ 1 ] . list_peers( ) . is_empty( ) ) ;
35923596 }
35933597 // At least one message handler should have seen the connection.
3594- assert ! ( peers [ handler & 1 ] . message_handler . chan_handler. conn_tracker. had_peers. load( Ordering :: Acquire ) ||
3595- peers [ handler & 1 ] . message_handler . route_handler. conn_tracker. had_peers. load( Ordering :: Acquire ) ||
3596- peers [ handler & 1 ] . message_handler . custom_message_handler. conn_tracker. had_peers. load( Ordering :: Acquire ) ) ;
3598+ assert ! ( chan_handler. conn_tracker. had_peers. load( Ordering :: Acquire ) ||
3599+ route_handler. conn_tracker. had_peers. load( Ordering :: Acquire ) ||
3600+ custom_message_handler. conn_tracker. had_peers. load( Ordering :: Acquire ) ) ;
35973601 // And both message handlers doing tracking should see the disconnection
3598- assert ! ( peers [ handler & 1 ] . message_handler . chan_handler. conn_tracker. connected_peers. lock( ) . unwrap( ) . is_empty( ) ) ;
3599- assert ! ( peers [ handler & 1 ] . message_handler . route_handler. conn_tracker. connected_peers. lock( ) . unwrap( ) . is_empty( ) ) ;
3600- assert ! ( peers [ handler & 1 ] . message_handler . custom_message_handler. conn_tracker. connected_peers. lock( ) . unwrap( ) . is_empty( ) ) ;
3602+ assert ! ( chan_handler. conn_tracker. connected_peers. lock( ) . unwrap( ) . is_empty( ) ) ;
3603+ assert ! ( route_handler. conn_tracker. connected_peers. lock( ) . unwrap( ) . is_empty( ) ) ;
3604+ assert ! ( custom_message_handler. conn_tracker. connected_peers. lock( ) . unwrap( ) . is_empty( ) ) ;
36013605 }
36023606
36033607 #[ test]
0 commit comments