X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FDefaultFcpClientTest.java;h=abe32d5cd5f765cc4032d661e23964b2ec8d8a0b;hb=80ee63c3eea721f3b0f536d5d812e34549332264;hp=f423e9763dc204ddf4a82635b734576f90db8e49;hpb=d27cfa7be4f40d51a2af83e0810c2759b55f3f7f;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 f423e97..abe32d5 100644
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -110,22 +110,6 @@ public class DefaultFcpClientTest {
 		keyPairFuture.get();
 	}
 
-	@Test
-	public void defaultFcpClientCanGenerateKeypair() throws ExecutionException, InterruptedException, IOException {
-		Future<FcpKeyPair> keyPairFuture = fcpClient.generateKeypair().execute();
-		connectNode();
-		List<String> lines = fcpServer.collectUntil(is("EndMessage"));
-		String identifier = extractIdentifier(lines);
-		fcpServer.writeLine("SSKKeypair",
-			"InsertURI=" + INSERT_URI + "",
-			"RequestURI=" + REQUEST_URI + "",
-			"Identifier=" + identifier,
-			"EndMessage");
-		FcpKeyPair keyPair = keyPairFuture.get();
-		assertThat(keyPair.getPublicKey(), is(REQUEST_URI));
-		assertThat(keyPair.getPrivateKey(), is(INSERT_URI));
-	}
-
 	private void connectNode() throws InterruptedException, ExecutionException, IOException {
 		fcpServer.connect().get();
 		fcpServer.collectUntil(is("EndMessage"));
@@ -1070,6 +1054,28 @@ public class DefaultFcpClientTest {
 		assertThat(lines, requestMatcher.get());
 	}
 
+	public class GenerateKeyPair {
+
+		@Test
+		public void defaultFcpClientCanGenerateKeypair() throws ExecutionException, InterruptedException, IOException {
+			Future<FcpKeyPair> keyPairFuture = fcpClient.generateKeypair().execute();
+			connectAndAssert(() ->matchesFcpMessage("GenerateSSK", "EndMessage"));
+			replyWithKeyPair();
+			FcpKeyPair keyPair = keyPairFuture.get();
+			assertThat(keyPair.getPublicKey(), is(REQUEST_URI));
+			assertThat(keyPair.getPrivateKey(), is(INSERT_URI));
+		}
+
+		private void replyWithKeyPair() throws IOException {
+			fcpServer.writeLine("SSKKeypair",
+				"InsertURI=" + INSERT_URI + "",
+				"RequestURI=" + REQUEST_URI + "",
+				"Identifier=" + identifier,
+				"EndMessage");
+		}
+
+	}
+
 	public class Peers {
 
 		public class PeerCommands {