feat(Swarm): raise PeerNotReachable when connect fails

richardschneider · Aug 10, 2019 · 285739c · 285739c
1 parent 46588bc
commit 285739c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/src/Swarm.cs b/src/Swarm.cs
@@ -88,6 +88,11 @@ public class Swarm : IService, IPolicy<MultiAddress>
         /// </remarks>
         public event EventHandler<Peer> PeerRemoved;
 
+        /// <summary>
+        ///   Raised when a peer cannot be connected to.
+        /// </summary>
+        public event EventHandler<Peer> PeerNotReachable;
+
         /// <summary>
         ///  The local peer.
         /// </summary>
@@ -497,6 +502,11 @@ public async Task StopAsync()
                         .ConfigureAwait(false);
                 }
             }
+            catch (Exception)
+            {
+                PeerNotReachable?.Invoke(this, peer);
+                throw;
+            }
             finally
             {
                 pendingConnections.TryRemove(peer, out AsyncLazy<PeerConnection> _);

diff --git a/test/SwarmTest.cs b/test/SwarmTest.cs
@@ -513,6 +513,25 @@ public void Connect_Refused()
             });
         }
 
+        [TestMethod]
+        public void Connect_Failure_Event()
+        {
+            var remoteId = "QmXFX2P5ammdmXQgfqGkfswtEVFsZUJ5KeHRXQYCTdiTAb";
+            MultiAddress remoteAddress = $"/ip4/127.0.0.1/tcp/4040/ipfs/{remoteId}";
+            var swarm = new Swarm { LocalPeer = self };
+            Peer unreachable = null;
+            swarm.PeerNotReachable += (s, e) =>
+            {
+                unreachable = e;
+            };
+            ExceptionAssert.Throws<Exception>(() =>
+            {
+                var _ = swarm.ConnectAsync(remoteAddress).Result;
+            });
+            Assert.IsNotNull(unreachable);
+            Assert.AreEqual(remoteId, unreachable.Id.ToBase58());
+        }
+
         [TestMethod]
         public void Connect_Not_Peer()
         {