Skip to content

Commit 0fa2156

Browse files
committed
f Pull out more handler vars
1 parent 09ee8ea commit 0fa2156

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

lightning/src/ln/peer_handler.rs

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)