Fix minor bugs

limebell · Jun 24, 2021 · d438b1e · d438b1e
1 parent 64996fb
commit d438b1e
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/Libplanet.Tests/Net/Messages/MessageTest.cs b/Libplanet.Tests/Net/Messages/MessageTest.cs
@@ -190,7 +190,7 @@ public void UseInvalidSignature()
                     validAppProtocolVersion,
                     ImmutableHashSet<PublicKey>.Empty,
                     null,
-                    TimeSpan.FromSeconds(1));
+                    TimeSpan.FromSeconds(5));
             });
         }
 

diff --git a/Libplanet.Tests/Net/Transports/TransportTest.cs b/Libplanet.Tests/Net/Transports/TransportTest.cs
@@ -3,6 +3,7 @@
 using System.Collections.Immutable;
 using System.Linq;
 using System.Net;
+using System.Net.Sockets;
 using System.Threading;
 using System.Threading.Tasks;
 using Libplanet.Crypto;
@@ -343,11 +344,16 @@ public async Task SendMessageToInvalidPeerAsync()
             try
             {
                 await InitializeAsync(transport);
+                // Make sure the tcp port is invalid.
+                var l = new TcpListener(IPAddress.Loopback, 0);
+                l.Start();
+                int port = ((IPEndPoint)l.LocalEndpoint).Port;
+                l.Stop();
                 var peer = new BoundPeer(
                     new PrivateKey().PublicKey,
                     new DnsEndPoint(
                         "0.0.0.0",
-                        ((BoundPeer)transport.AsPeer).EndPoint.Port + 1));
+                        port));
                 Task task = transport.SendMessageWithReplyAsync(
                     peer,
                     new Ping(),

diff --git a/Libplanet/Net/Swarm.BlockSync.cs b/Libplanet/Net/Swarm.BlockSync.cs
@@ -252,7 +252,8 @@ CancellationToken cancellationToken
                         logSessionIds: (logSessionId, subSessionId),
                         cancellationToken: cancellationToken
                     );
-                    IEnumerable<Tuple<long, BlockHash>> hashes = await hashesAsync.ToArrayAsync();
+                    IEnumerable<Tuple<long, BlockHash>> hashes =
+                        await hashesAsync.ToArrayAsync(cancellationToken);
 
                     if (!hashes.Any())
                     {