feat: add NetworkPrimitives to NetworkBuilder (#13169)

Co-authored-by: Arsenii Kulikov <klkvrr@gmail.com>
This commit is contained in:
Dan Cline
2024-12-10 15:56:32 -05:00
committed by GitHub
parent 73f1583455
commit 37f3933db2
26 changed files with 201 additions and 143 deletions

View File

@ -1,7 +1,7 @@
use futures_util::StreamExt;
use reth_network_api::{
events::PeerEvent, test_utils::PeersHandleProvider, NetworkEvent, NetworkEventListenerProvider,
PeersInfo,
PeerRequest, PeersInfo,
};
use reth_network_peers::{NodeRecord, PeerId};
use reth_tokio_util::EventStream;
@ -9,8 +9,8 @@ use reth_tracing::tracing::info;
/// Helper for network operations
#[derive(Debug)]
pub struct NetworkTestContext<Network> {
network_events: EventStream<NetworkEvent>,
pub struct NetworkTestContext<Network: NetworkEventListenerProvider> {
network_events: EventStream<NetworkEvent<PeerRequest<Network::Primitives>>>,
network: Network,
}