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

test(net): speedup network tests (#1337)

Browse Source
This commit is contained in:
Matthias Seitz
2023-02-14 15:53:05 +01:00
committed by GitHub
parent 5361d5f834
commit 0aaf51d462
2 changed files with 4 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
crates/net/network/src/test_utils/testnet.rs
View File

@ -158,9 +158,8 @@ impl Testnet<NoopProvider> {
/// Creates a new [`Testnet`] with the given number of peers /// Creates a new [`Testnet`] with the given number of peers
pub async fn try_create(num_peers: usize) -> Result<Self, NetworkError> { pub async fn try_create(num_peers: usize) -> Result<Self, NetworkError> {
let mut this = Testnet::default(); let mut this = Testnet::default();
for _ in 0..num_peers {
this.add_peer().await?; this.extend_peer_with_config((0..num_peers).map(|_| Default::default())).await?;
}
Ok(this) Ok(this)
} }

6
crates/net/network/tests/it/session.rs
View File

@ -15,7 +15,6 @@ async fn test_session_established_with_highest_version() {
reth_tracing::init_test_tracing(); reth_tracing::init_test_tracing();
let net = Testnet::create(2).await; let net = Testnet::create(2).await;
net.for_each(|peer| assert_eq!(0, peer.num_peers())); net.for_each(|peer| assert_eq!(0, peer.num_peers()));
let mut handles = net.handles(); let mut handles = net.handles();
@ -37,12 +36,11 @@ async fn test_session_established_with_highest_version() {
assert_eq!(handle1.peer_id(), &peer_id); assert_eq!(handle1.peer_id(), &peer_id);
assert_eq!(status.version, EthVersion::Eth67 as u8); assert_eq!(status.version, EthVersion::Eth67 as u8);
} }
_ => { ev => {
panic!("unexpected event") panic!("unexpected event {ev:?}")
} }
} }
} }
handle.terminate().await; handle.terminate().await;
} }