From 036fea735c39b19531e16b0ad0519d10c441f434 Mon Sep 17 00:00:00 2001
From: Anton <a.skuratovich@gmail.com>
Date: Sun, 8 Sep 2024 16:01:06 +0200
Subject: [PATCH] fix pubsub tests (#93)

---
 .../Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs      | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs b/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
index 2b1d99a3..caf4f49b 100644
--- a/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
+++ b/src/libp2p/Libp2p.Protocols.Pubsub.Tests/PubsubProtocolTests.cs
@@ -23,6 +23,7 @@ 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]);
@@ -30,7 +31,8 @@ public async Task Test_Peer_is_dialed_when_added_by_discovery()
         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();
     }
 }