fix pubsub tests (#93)

NethermindEth · Sep 8, 2024 · 036fea7 · 036fea7
1 parent e9ab7b5
commit 036fea7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs b/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
@@ -23,14 +23,16 @@ public async Task Test_Peer_is_dialed_when_added_by_discovery()
 
         TestDiscoveryProtocol discovery = new();
         CancellationToken token = default;
+        TaskCompletionSource taskCompletionSource = new();
 
         _ = router.RunAsync(peer, discovery, token: token);
         discovery.OnAddPeer!([discoveredPeer]);
 
         await Task.Delay(100);
         _ = peer.Received().DialAsync(discoveredPeer, Arg.Any<CancellationToken>());
 
-        router.OutboundConnection(discoveredPeer, PubsubRouter.FloodsubProtocolVersion, Task.CompletedTask, (rpc) => { });
+        router.OutboundConnection(discoveredPeer, PubsubRouter.FloodsubProtocolVersion, taskCompletionSource.Task, (rpc) => { });
         Assert.That(state.ConnectedPeers, Has.Member(peerId));
+        taskCompletionSource.SetResult();
     }
 }