Skip to content

Commit 434bad1

Browse files
committed
f Pull all peers.len() into scoped variables
1 parent f793181 commit 434bad1

File tree

1 file changed

+56
-14
lines changed

1 file changed

+56
-14
lines changed

lightning/src/ln/peer_handler.rs

Lines changed: 56 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3456,7 +3456,10 @@ mod tests {
34563456
let cfgs = create_peermgr_cfgs(2);
34573457
let peers = create_network(2, &cfgs);
34583458
establish_connection(&peers[0], &peers[1]);
3459-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3459+
{
3460+
let peers_len = peers[0].peers.read().unwrap().len();
3461+
assert_eq!(peers_len, 1);
3462+
}
34603463

34613464
let their_id = peers[1].node_signer.get_node_id(Recipient::Node).unwrap();
34623465
cfgs[0].chan_handler.pending_events.lock().unwrap().push(MessageSendEvent::HandleError {
@@ -3465,7 +3468,10 @@ mod tests {
34653468
});
34663469

34673470
peers[0].process_events();
3468-
assert_eq!(peers[0].peers.read().unwrap().len(), 0);
3471+
{
3472+
let peers_len = peers[0].peers.read().unwrap().len();
3473+
assert_eq!(peers_len, 0);
3474+
}
34693475
}
34703476

34713477
#[test]
@@ -3477,7 +3483,10 @@ mod tests {
34773483
let b_chan_handler = test_utils::TestChannelMessageHandler::new(ChainHash::using_genesis_block(Network::Testnet));
34783484
let mut peers = create_network(2, &cfgs);
34793485
let (fd_a, mut fd_b) = establish_connection(&peers[0], &peers[1]);
3480-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3486+
{
3487+
let peers_len = peers[0].peers.read().unwrap().len();
3488+
assert_eq!(peers_len, 1);
3489+
}
34813490

34823491
let their_id = peers[1].node_signer.get_node_id(Recipient::Node).unwrap();
34833492

@@ -3532,10 +3541,16 @@ mod tests {
35323541
let cfgs = create_peermgr_cfgs(2);
35333542
let peers = create_network(2, &cfgs);
35343543
establish_connection(&peers[0], &peers[1]);
3535-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3544+
{
3545+
let peers_len = peers[0].peers.read().unwrap().len();
3546+
assert_eq!(peers_len, 1);
3547+
}
35363548

35373549
peers[0].disconnect_all_peers();
3538-
assert_eq!(peers[0].peers.read().unwrap().len(), 0);
3550+
{
3551+
let peers_len = peers[0].peers.read().unwrap().len();
3552+
assert_eq!(peers_len, 0);
3553+
}
35393554
}
35403555

35413556
#[test]
@@ -3544,17 +3559,26 @@ mod tests {
35443559
let cfgs = create_peermgr_cfgs(2);
35453560
let peers = create_network(2, &cfgs);
35463561
establish_connection(&peers[0], &peers[1]);
3547-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3562+
{
3563+
let peers_len = peers[0].peers.read().unwrap().len();
3564+
assert_eq!(peers_len, 1);
3565+
}
35483566

35493567
// peers[0] awaiting_pong is set to true, but the Peer is still connected
35503568
peers[0].timer_tick_occurred();
35513569
peers[0].process_events();
3552-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3570+
{
3571+
let peers_len = peers[0].peers.read().unwrap().len();
3572+
assert_eq!(peers_len, 1);
3573+
}
35533574

35543575
// Since timer_tick_occurred() is called again when awaiting_pong is true, all Peers are disconnected
35553576
peers[0].timer_tick_occurred();
35563577
peers[0].process_events();
3557-
assert_eq!(peers[0].peers.read().unwrap().len(), 0);
3578+
{
3579+
let peers_len = peers[0].peers.read().unwrap().len();
3580+
assert_eq!(peers_len, 0);
3581+
}
35583582
}
35593583

35603584
fn do_test_peer_connected_error_disconnects(handler: usize) {
@@ -3665,9 +3689,15 @@ mod tests {
36653689
peers[0].new_inbound_connection(fd_a.clone(), None).unwrap();
36663690

36673691
// If we get a single timer tick before completion, that's fine
3668-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3692+
{
3693+
let peers_len = peers[0].peers.read().unwrap().len();
3694+
assert_eq!(peers_len, 1);
3695+
}
36693696
peers[0].timer_tick_occurred();
3670-
assert_eq!(peers[0].peers.read().unwrap().len(), 1);
3697+
{
3698+
let peers_len = peers[0].peers.read().unwrap().len();
3699+
assert_eq!(peers_len, 1);
3700+
}
36713701

36723702
assert_eq!(peers[0].read_event(&mut fd_a, &initial_data).unwrap(), false);
36733703
peers[0].process_events();
@@ -3677,7 +3707,10 @@ mod tests {
36773707

36783708
// ...but if we get a second timer tick, we should disconnect the peer
36793709
peers[0].timer_tick_occurred();
3680-
assert_eq!(peers[0].peers.read().unwrap().len(), 0);
3710+
{
3711+
let peers_len = peers[0].peers.read().unwrap().len();
3712+
assert_eq!(peers_len, 0);
3713+
}
36813714

36823715
let b_data = fd_b.outbound_data.lock().unwrap().split_off(0);
36833716
assert!(peers[0].read_event(&mut fd_a, &b_data).is_err());
@@ -3740,7 +3773,10 @@ mod tests {
37403773
assert!(peer_b.peers.read().unwrap().get(&fd_b).unwrap().lock().unwrap().handshake_complete());
37413774

37423775
// Make sure we're still connected.
3743-
assert_eq!(peer_b.peers.read().unwrap().len(), 1);
3776+
{
3777+
let peers_len = peer_b.peers.read().unwrap().len();
3778+
assert_eq!(peers_len, 1);
3779+
}
37443780

37453781
// B should send a ping on the first timer tick after `handshake_complete`.
37463782
assert!(fd_b.outbound_data.lock().unwrap().split_off(0).is_empty());
@@ -3771,11 +3807,17 @@ mod tests {
37713807
peer_b.timer_tick_occurred();
37723808
send_warning();
37733809
}
3774-
assert_eq!(peer_b.peers.read().unwrap().len(), 1);
3810+
{
3811+
let peers_len = peer_b.peers.read().unwrap().len();
3812+
assert_eq!(peers_len, 1);
3813+
}
37753814

37763815
// One more tick should enforce the pong timeout.
37773816
peer_b.timer_tick_occurred();
3778-
assert_eq!(peer_b.peers.read().unwrap().len(), 0);
3817+
{
3818+
let peers_len = peer_b.peers.read().unwrap().len();
3819+
assert_eq!(peers_len, 0);
3820+
}
37793821
}
37803822

37813823
#[test]

0 commit comments

Comments
 (0)