BAEL-679 code for reactive streams (#1970)

* BAEL-679 code for reactive streams * BAEL-679 typo
2017-06-02 18:48:17 +02:00
parent d614fc2fa7
commit 9184693a2b
3 changed files with 150 additions and 0 deletions
@@ -0,0 +1,69 @@
+package com.baeldung.java9.streams.reactive;
+
+
+import org.junit.Test;
+
+import java.util.List;
+import java.util.concurrent.SubmissionPublisher;
+
+import static org.assertj.core.api.Java6Assertions.assertThat;
+
+public class ReactiveStreamsTest {
+
+    @Test
+    public void givenPublisher_whenSubscribeToIt_thenShouldConsumeAllElements() throws InterruptedException {
+        //given
+        SubmissionPublisher<String> publisher = new SubmissionPublisher<>();
+        EndSubscriber<String> subscriber = new EndSubscriber<>(6);
+        publisher.subscribe(subscriber);
+        List<String> items = List.of("1", "x", "2", "x", "3", "x");
+
+        //when
+        assertThat(publisher.getNumberOfSubscribers()).isEqualTo(1);
+        items.forEach(publisher::submit);
+        publisher.close();
+
+        //then
+        Thread.sleep(1000);
+        assertThat(subscriber.consumedElements).containsExactlyElementsOf(items);
+    }
+
+    @Test
+    public void givenPublisher_whenSubscribeAndTransformElements_thenShouldConsumeAllElements() throws InterruptedException {
+        //given
+        SubmissionPublisher<String> publisher = new SubmissionPublisher<>();
+        TransformProcessor<String, Integer> transformProcessor = new TransformProcessor<>(Integer::parseInt);
+        EndSubscriber<Integer> subscriber = new EndSubscriber<>(3);
+        List<String> items = List.of("1", "2", "3");
+        List<Integer> expectedResult = List.of(1, 2, 3);
+
+        //when
+        publisher.subscribe(transformProcessor);
+        transformProcessor.subscribe(subscriber);
+        items.forEach(publisher::submit);
+        publisher.close();
+
+        //then
+        Thread.sleep(1000);
+        assertThat(subscriber.consumedElements).containsExactlyElementsOf(expectedResult);
+    }
+
+    @Test
+    public void givenPublisher_whenRequestForOnlyOneElement_thenShouldConsumeOnlyThatOne() throws InterruptedException {
+        //given
+        SubmissionPublisher<String> publisher = new SubmissionPublisher<>();
+        EndSubscriber<String> subscriber = new EndSubscriber<>(1);
+        publisher.subscribe(subscriber);
+        List<String> items = List.of("1", "x", "2", "x", "3", "x");
+        List<String> expected = List.of("1");
+
+        //when
+        assertThat(publisher.getNumberOfSubscribers()).isEqualTo(1);
+        items.forEach(publisher::submit);
+        publisher.close();
+
+        //then
+        Thread.sleep(1000);
+        assertThat(subscriber.consumedElements).containsExactlyElementsOf(expected);
+    }
+}