From 10b87473ca4af20eee5c2b55bd6313945a666f97 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@freenetproject.org>
Date: Mon, 24 Aug 2015 06:29:59 +0200
Subject: [PATCH] Refactor connection failure tests

---
 .../fcp/quelaton/DefaultFcpClientTest.java         | 46 +++++++++++-----------
 1 file changed, 23 insertions(+), 23 deletions(-)

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
-- 
2.7.4