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: replace unused_tcp_udp with port 0 (#4805)

Browse Source
This commit is contained in:
Matthias Seitz
2023-09-26 18:57:16 +02:00
committed by GitHub
parent 29a52010b6
commit f7c3bbcc99
1 changed files with 9 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -11,9 +11,7 @@ use reth_interfaces::{
};
use reth_net_common::ban_list::BanList;
use reth_network::{
test_utils::{
enr_to_peer_id, unused_tcp_udp, NetworkEventStream, PeerConfig, Testnet, GETH_TIMEOUT,
},
test_utils::{enr_to_peer_id, NetworkEventStream, PeerConfig, Testnet, GETH_TIMEOUT},
NetworkConfigBuilder, NetworkEvent, NetworkManager, PeersConfig,
};
use reth_network_api::{NetworkInfo, Peers, PeersInfo};
@ -328,10 +326,9 @@ async fn test_incoming_node_id_blacklist() {
let ban_list = BanList::new(vec![geth_peer_id], HashSet::new());
let peer_config = PeersConfig::default().with_ban_list(ban_list);
let (reth_p2p, reth_disc) = unused_tcp_udp();
let config = NetworkConfigBuilder::new(secret_key)
.listener_addr(reth_p2p)
.discovery_addr(reth_disc)
.listener_port(0)
.disable_discovery()
.peer_config(peer_config)
.build(NoopProvider::default());
@ -343,7 +340,7 @@ async fn test_incoming_node_id_blacklist() {
tokio::task::spawn(network);
// make geth connect to us
let our_enode = NodeRecord::new(reth_p2p, *handle.peer_id());
let our_enode = NodeRecord::new(handle.local_addr(), *handle.peer_id());
provider.add_peer(our_enode.to_string()).await.unwrap();
@ -378,10 +375,9 @@ async fn test_incoming_connect_with_single_geth() {
// get the peer id we should be expecting
let geth_peer_id = enr_to_peer_id(provider.node_info().await.unwrap().enr);
let (reth_p2p, reth_disc) = unused_tcp_udp();
let config = NetworkConfigBuilder::new(secret_key)
.listener_addr(reth_p2p)
.discovery_addr(reth_disc)
.listener_port(0)
.disable_discovery()
.build(NoopProvider::default());
let network = NetworkManager::new(config).await.unwrap();
@ -393,7 +389,7 @@ async fn test_incoming_connect_with_single_geth() {
let mut event_stream = NetworkEventStream::new(events);
// make geth connect to us
let our_enode = NodeRecord::new(reth_p2p, *handle.peer_id());
let our_enode = NodeRecord::new(handle.local_addr(), *handle.peer_id());
provider.add_peer(our_enode.to_string()).await.unwrap();
@ -413,10 +409,9 @@ async fn test_outgoing_connect_with_single_geth() {
tokio::time::timeout(GETH_TIMEOUT, async move {
let secret_key = SecretKey::new(&mut rand::thread_rng());
let (reth_p2p, reth_disc) = unused_tcp_udp();
let config = NetworkConfigBuilder::new(secret_key)
.listener_addr(reth_p2p)
.discovery_addr(reth_disc)
.listener_port(0)
.disable_discovery()
.build(NoopProvider::default());
let network = NetworkManager::new(config).await.unwrap();