From 3ffbda4c7678be3e6990f45dca936c996987c705 Mon Sep 17 00:00:00 2001
From: Oaroomy <walnuty87@gmail.com>
Date: Sat, 14 Dec 2019 15:39:15 +0900
Subject: [PATCH] Fix test

---
 Libplanet.Tests/Net/SwarmTest.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Libplanet.Tests/Net/SwarmTest.cs b/Libplanet.Tests/Net/SwarmTest.cs
index 92e23ee036b..01703ad6a1a 100644
--- a/Libplanet.Tests/Net/SwarmTest.cs
+++ b/Libplanet.Tests/Net/SwarmTest.cs
@@ -15,6 +15,7 @@
 using Libplanet.Crypto;
 using Libplanet.Net;
 using Libplanet.Net.Messages;
+using Libplanet.Net.Protocols;
 using Libplanet.Store;
 using Libplanet.Tests.Blockchain;
 using Libplanet.Tests.Common.Action;
@@ -334,7 +335,7 @@ public async Task BootstrapException()
 
             try
             {
-                await Assert.ThrowsAsync<SwarmException>(
+                await Assert.ThrowsAsync<PeerDiscoveryException>(
                     () => swarmB.BootstrapAsync(new[] { swarmA.AsPeer }, 3000, 3000));
 
                 await StartAsync(swarmA);
@@ -530,7 +531,7 @@ void GameHandler(object sender, DifferentProtocolVersionEventArgs e)
             {
                 await StartAsync(b);
 
-                await Assert.ThrowsAsync<SwarmException>(() => BootstrapAsync(a, b.AsPeer));
+                await Assert.ThrowsAsync<PeerDiscoveryException>(() => BootstrapAsync(a, b.AsPeer));
 
                 Assert.True(isCalled);
             }