Add test for multiple listeners on one subscription

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

diff --git a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
index 4e5eb84..c37b5e0 100644 (file)
--- a/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
+++ b/src/test/java/net/pterodactylus/fcp/quelaton/DefaultFcpClientTest.java
@@ -2296,13 +2296,25 @@ public class DefaultFcpClientTest {
                public void subscriptionWorks() throws InterruptedException, ExecutionException, IOException {
                        Future<Optional<UskSubscription>> uskSubscription = fcpClient.subscribeUsk().uri(URI).execute();
                        connectAndAssert(() -> matchesFcpMessage("SubscribeUSK", "URI=" + URI, "EndMessage"));
-                       fcpServer.writeLine(
-                               "SubscribedUSK",
-                               "Identifier=" + identifier,
-                               "URI=" + URI,
-                               "DontPoll=false",
-                               "EndMessage"
-                       );
+                       replyWithSubscribed();
+                       assertThat(uskSubscription.get().get().getUri(), is(URI));
+                       AtomicInteger edition = new AtomicInteger();
+                       CountDownLatch updated = new CountDownLatch(2);
+                       uskSubscription.get().get().onUpdate(e -> {
+                               edition.set(e);
+                               updated.countDown();
+                       });
+                       sendUpdateNotification(23);
+                       sendUpdateNotification(24);
+                       assertThat("updated in time", updated.await(5, TimeUnit.SECONDS), is(true));
+                       assertThat(edition.get(), is(24));
+               }
+
+               @Test
+               public void subscriptionUpdatesMultipleTimes() throws InterruptedException, ExecutionException, IOException {
+                       Future<Optional<UskSubscription>> uskSubscription = fcpClient.subscribeUsk().uri(URI).execute();
+                       connectAndAssert(() -> matchesFcpMessage("SubscribeUSK", "URI=" + URI, "EndMessage"));
+                       replyWithSubscribed();
                        assertThat(uskSubscription.get().get().getUri(), is(URI));
                        AtomicInteger edition = new AtomicInteger();
                        CountDownLatch updated = new CountDownLatch(2);
@@ -2310,22 +2322,31 @@ public class DefaultFcpClientTest {
                                edition.set(e);
                                updated.countDown();
                        });
+                       uskSubscription.get().get().onUpdate(e -> updated.countDown());
+                       sendUpdateNotification(23);
+                       assertThat("updated in time", updated.await(5, TimeUnit.SECONDS), is(true));
+                       assertThat(edition.get(), is(23));
+               }
+
+               private void replyWithSubscribed() throws IOException {
                        fcpServer.writeLine(
-                               "SubscribedUSKUpdate",
+                               "SubscribedUSK",
                                "Identifier=" + identifier,
                                "URI=" + URI,
-                               "Edition=23",
+                               "DontPoll=false",
                                "EndMessage"
                        );
+               }
+
+               private void sendUpdateNotification(int edition, String... additionalLines) throws IOException {
                        fcpServer.writeLine(
                                "SubscribedUSKUpdate",
                                "Identifier=" + identifier,
                                "URI=" + URI,
-                               "Edition=24",
-                               "EndMessage"
+                               "Edition=" + edition
                        );
-                       assertThat("updated in time", updated.await(5, TimeUnit.SECONDS), is(true));
-                       assertThat(edition.get(), is(24));
+                       fcpServer.writeLine(additionalLines);
+                       fcpServer.writeLine("EndMessage");
                }
 
        }