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

feat: Add PeerEvent to NetworkEvent enum (#574)

Browse Source 
* Add peerAdded and peerRemoved events to NetworkEvent enum

* Remove unnecessary todo macros
This commit is contained in:
Sanket Shanbhag
2022-12-24 14:55:09 +05:30
committed by GitHub
parent 7c80dde0ec
commit 43008b7b07
4 changed files with 14 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,6 +65,12 @@ async fn test_establish_connections() {
NetworkEvent::SessionEstablished { peer_id, .. } => {
assert!(expected_connections.remove(&peer_id))
}
NetworkEvent::PeerAdded(peer_id) => {
assert!(!expected_connections.contains(&peer_id))
}
NetworkEvent::PeerRemoved(_) => {
panic!("unexpected event")
}
}
}
assert!(expected_connections.is_empty());

4
crates/net/network/tests/it/testnet.rs
View File

@ -308,7 +308,7 @@ impl NetworkEventStream {
while let Some(ev) = self.inner.next().await {
match ev {
NetworkEvent::SessionClosed { peer_id, reason } => return Some((peer_id, reason)),
NetworkEvent::SessionEstablished { .. } => continue,
_ => continue,
}
}
None
@ -317,8 +317,8 @@ impl NetworkEventStream {
pub async fn next_session_established(&mut self) -> Option<PeerId> {
while let Some(ev) = self.inner.next().await {
match ev {
NetworkEvent::SessionClosed { .. } => continue,
NetworkEvent::SessionEstablished { peer_id, .. } => return Some(peer_id),
_ => continue,
}
}
None