X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FDefaultFcpClientTest.java;h=e454798712dc2eaf8ebb9d8d2208c1872acac55b;hb=10b87473ca4af20eee5c2b55bd6313945a666f97;hp=3374556d456cafd1b09315681ccb03bb13914c4e;hpb=236cdec10f72a0fb989092c55b7ca5a5f5142779;p=jFCPlib.git

diff --git a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
index 3374556..e454798 100644
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -88,29 +88,6 @@ public class DefaultFcpClientTest {
 		threadPool.shutdown();
 	}
 
-	@Test(expected = ExecutionException.class)
-	public void defaultFcpClientThrowsExceptionIfItCanNotConnect()
-	throws IOException, ExecutionException, InterruptedException {
-		Future<FcpKeyPair> keyPairFuture = fcpClient.generateKeypair().execute();
-		fcpServer.connect().get();
-		fcpServer.collectUntil(is("EndMessage"));
-		fcpServer.writeLine(
-			"CloseConnectionDuplicateClientName",
-			"EndMessage"
-		);
-		keyPairFuture.get();
-	}
-
-	@Test(expected = ExecutionException.class)
-	public void defaultFcpClientThrowsExceptionIfConnectionIsClosed()
-	throws IOException, ExecutionException, InterruptedException {
-		Future<FcpKeyPair> keyPairFuture = fcpClient.generateKeypair().execute();
-		fcpServer.connect().get();
-		fcpServer.collectUntil(is("EndMessage"));
-		fcpServer.close();
-		keyPairFuture.get();
-	}
-
 	private void connectNode() throws InterruptedException, ExecutionException, IOException {
 		fcpServer.connect().get();
 		fcpServer.collectUntil(is("EndMessage"));
@@ -961,6 +938,29 @@ public class DefaultFcpClientTest {
 		assertThat(lines, requestMatcher.get());
 	}
 
+	public class Connections {
+
+		@Test(expected = ExecutionException.class)
+		public void throwsExceptionOnFailure() throws IOException, ExecutionException, InterruptedException {
+			Future<FcpKeyPair> keyPairFuture = fcpClient.generateKeypair().execute();
+			connectAndAssert(() -> matchesFcpMessage("GenerateSSK"));
+			fcpServer.writeLine(
+				"CloseConnectionDuplicateClientName",
+				"EndMessage"
+			);
+			keyPairFuture.get();
+		}
+
+		@Test(expected = ExecutionException.class)
+		public void throwsExceptionIfConnectionIsClosed() throws IOException, ExecutionException, InterruptedException {
+			Future<FcpKeyPair> keyPairFuture = fcpClient.generateKeypair().execute();
+			connectAndAssert(() -> matchesFcpMessage("GenerateSSK"));
+			fcpServer.close();
+			keyPairFuture.get();
+		}
+
+	}
+
 	public class GenerateKeyPair {
 
 		@Test