Fix test

planetarium · Dec 14, 2019 · ec85e57 · ec85e57
1 parent 095adca
commit ec85e57
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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);
             }