X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Ffcp%2Fquelaton%2FDefaultFcpClientTest.java;h=77573c05673faf06bdea9498c853b7fb544f4e7d;hb=002ccd34749862f5e43376d8c8518f4dc0c3f0bc;hp=c8a834a017ca17d910598ce444cfc943060df9dc;hpb=e601f2e7ccb7e1db6b09b3236bc8bedd5fdca255;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 c8a834a..77573c0 100644
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -1605,8 +1605,7 @@ public class DefaultFcpClientTest {
 					.uri("KSK@foo.txt")
 					.execute();
 				connectNode();
-				List<String> lines = fcpServer.collectUntil(is("Hello"));
-				String identifier = extractIdentifier(lines);
+				readMessage("Hello", this::matchesDirectClientPut);
 				replyWithGeneratedUri();
 				replyWithPutSuccessful(identifier);
 				assertThat(key.get().get().getKey(), is("KSK@foo.txt"));
@@ -1699,8 +1698,8 @@ public class DefaultFcpClientTest {
 				connectAndAssert(() -> matchesClientPutDiskDir("Verbosity=1"));
 				replyWithSimpleProgress(1, 2, 3, 4, 5, 6, true, 8);
 				replyWithSimpleProgress(11, 12, 13, 14, 15, 16, false, 18);
-				fcpServer.writeLine("PutSuccessful", "Identifier=" + identifier, "URI=CHK@abc", "EndMessage");
-				assertThat(key.get().get().getKey(), is("CHK@abc"));
+				replyWithPutSuccessful(identifier);
+				assertThat(key.get().get().getKey(), is("KSK@foo.txt"));
 				assertThat(requestProgress, contains(
 					isRequestProgress(1, 2, 3, 4, 5, 6, true, 8),
 					isRequestProgress(11, 12, 13, 14, 15, 16, false, 18)