Handle connection failures in the reply sequence

[jFCPlib.git] / src / test / java / net / pterodactylus / fcp / quelaton / FcpReplySequenceTest.java

diff --git a/src/test/java/net/pterodactylus/fcp/quelaton/FcpReplySequenceTest.java b/src/test/java/net/pterodactylus/fcp/quelaton/FcpReplySequenceTest.java
index f81201e..9300ee8 100644 (file)
--- a/src/test/java/net/pterodactylus/fcp/quelaton/FcpReplySequenceTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/FcpReplySequenceTest.java
@@ -127,9 +127,13 @@ public class FcpReplySequenceTest {
                waitForASpecificMessage(replySequence::receivedNodeHello, NodeHello.class, NodeHello::new);
        }
 
-       @Test
+       @Test(expected = ExecutionException.class)
        public void waitingForConnectionClosedDuplicateClientNameWorks() throws IOException, ExecutionException, InterruptedException {
-               waitForASpecificMessage( replySequence::receivedCloseConnectionDuplicateClientName, CloseConnectionDuplicateClientName.class, CloseConnectionDuplicateClientName::new);
+               replySequence.setExpectedMessage("");
+               Future<Boolean> result = replySequence.send(fcpMessage);
+               replySequence.receivedCloseConnectionDuplicateClientName(fcpConnection,
+                       new CloseConnectionDuplicateClientName(new FcpMessage("CloseConnectionDuplicateClientName")));
+               result.get();
        }
 
        @Test
@@ -407,12 +411,6 @@ public class FcpReplySequenceTest {
                }
 
                @Override
-               protected void consumeCloseConnectionDuplicateClientName(
-                       CloseConnectionDuplicateClientName closeConnectionDuplicateClientName) {
-                       gotMessage.set(closeConnectionDuplicateClientName.getName());
-               }
-
-               @Override
                protected void consumeSSKKeypair(SSKKeypair sskKeypair) {
                        gotMessage.set(sskKeypair.getName());
                }