jleony/nanoreth
1
0
Fork 0
mirror of https://github.com/hl-archive-node/nanoreth.git synced 2025-12-06 19:09:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(net): add tracing to shutdown test (#1310)

Browse Source
This commit is contained in:
Matthias Seitz
2023-02-13 14:34:18 +01:00
committed by GitHub
parent 211c7e3401
commit 6a2c61f4d9
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
crates/net/network/tests/it/connect.rs
View File

@ -511,6 +511,7 @@ async fn test_geth_disconnect() {
#[tokio::test(flavor = "multi_thread")]
async fn test_shutdown() {
reth_tracing::init_test_tracing();
let net = Testnet::create(3).await;
let mut handles = net.handles();
@ -532,18 +533,18 @@ async fn test_shutdown() {
// Before shutting down, we have two connected peers
let peer1 = listener0.next_session_established().await.unwrap();
let peer2 = listener0.next_session_established().await.unwrap();
assert_eq!(handle0.num_connected_peers(), 2);
assert!(expected_connections.contains(&peer1));
assert!(expected_connections.contains(&peer2));
assert_eq!(handle0.num_connected_peers(), 2);
handle0.shutdown().await.unwrap();
// All sessions get disconnected
let (peer1, _reason) = listener0.next_session_closed().await.unwrap();
let (peer2, _reason) = listener0.next_session_closed().await.unwrap();
assert_eq!(handle0.num_connected_peers(), 0);
assert!(expected_connections.remove(&peer1));
assert!(expected_connections.remove(&peer2));
assert_eq!(handle0.num_connected_peers(), 0);
// New connections are rejected
handle0.add_peer(*handle1.peer_id(), handle1.local_addr());