fix(net): add already connected check (#521)

* fix(net): add already connected check

* fix: get events before adding peer
This commit is contained in:
Matthias Seitz
2022-12-19 18:03:10 +01:00
committed by GitHub
parent 5758c297b3
commit aac078a132
5 changed files with 71 additions and 4 deletions

View File

@ -154,6 +154,8 @@ async fn test_incoming_node_id_blacklist() {
let network = NetworkManager::new(config).await.unwrap();
let handle = network.handle().clone();
let events = handle.event_listener();
tokio::task::spawn(network);
// make geth connect to us
@ -162,12 +164,12 @@ async fn test_incoming_node_id_blacklist() {
provider.add_peer(our_enode).await.unwrap();
let events = handle.event_listener();
let mut event_stream = NetworkEventStream::new(events);
// check for session to be opened
let incoming_peer_id = event_stream.next_session_established().await.unwrap();
assert_eq!(incoming_peer_id, geth_peer_id);
// check to see that the session was closed
let incoming_peer_id = event_stream.next_session_closed().await.unwrap();
assert_eq!(incoming_peer_id, geth_peer_id);