111 lines
3.8 KiB
Java
111 lines
3.8 KiB
Java
/**
|
|
* Copyright (C) 2014-2016 Typesafe Inc. <http://www.typesafe.com>
|
|
*/
|
|
package akka.stream.javadsl;
|
|
|
|
import java.util.Arrays;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.concurrent.CompletionStage;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
import akka.NotUsed;
|
|
import akka.japi.function.Function;
|
|
import akka.stream.*;
|
|
import org.junit.ClassRule;
|
|
import org.junit.Test;
|
|
import org.reactivestreams.Publisher;
|
|
import scala.concurrent.Await;
|
|
import scala.concurrent.Future;
|
|
import scala.concurrent.duration.Duration;
|
|
import akka.japi.function.Function2;
|
|
import akka.stream.testkit.AkkaSpec;
|
|
import akka.testkit.JavaTestKit;
|
|
|
|
import static org.junit.Assert.*;
|
|
|
|
public class SinkTest extends StreamTest {
|
|
public SinkTest() {
|
|
super(actorSystemResource);
|
|
}
|
|
|
|
@ClassRule
|
|
public static AkkaJUnitActorSystemResource actorSystemResource = new AkkaJUnitActorSystemResource("FlowTest",
|
|
AkkaSpec.testConf());
|
|
|
|
@Test
|
|
public void mustBeAbleToUseFanoutPublisher() throws Exception {
|
|
final Sink<Object, Publisher<Object>> pubSink = Sink.asPublisher(AsPublisher.WITH_FANOUT);
|
|
@SuppressWarnings("unused")
|
|
final Publisher<Object> publisher = Source.from(new ArrayList<Object>()).runWith(pubSink, materializer);
|
|
}
|
|
|
|
@Test
|
|
public void mustBeAbleToUseFuture() throws Exception {
|
|
final Sink<Integer, CompletionStage<Integer>> futSink = Sink.head();
|
|
final List<Integer> list = Collections.singletonList(1);
|
|
final CompletionStage<Integer> future = Source.from(list).runWith(futSink, materializer);
|
|
assert future.toCompletableFuture().get(1, TimeUnit.SECONDS).equals(1);
|
|
}
|
|
|
|
@Test
|
|
public void mustBeAbleToUseFold() throws Exception {
|
|
Sink<Integer, CompletionStage<Integer>> foldSink = Sink.fold(0, (arg1, arg2) -> arg1 + arg2);
|
|
@SuppressWarnings("unused")
|
|
CompletionStage<Integer> integerFuture = Source.from(new ArrayList<Integer>()).runWith(foldSink, materializer);
|
|
}
|
|
|
|
@Test
|
|
public void mustBeAbleToUseActorRefSink() throws Exception {
|
|
final JavaTestKit probe = new JavaTestKit(system);
|
|
final Sink<Integer, ?> actorRefSink = Sink.actorRef(probe.getRef(), "done");
|
|
Source.from(Arrays.asList(1, 2, 3)).runWith(actorRefSink, materializer);
|
|
probe.expectMsgEquals(1);
|
|
probe.expectMsgEquals(2);
|
|
probe.expectMsgEquals(3);
|
|
probe.expectMsgEquals("done");
|
|
}
|
|
|
|
@Test
|
|
public void mustBeAbleToCombine() throws Exception {
|
|
final JavaTestKit probe1 = new JavaTestKit(system);
|
|
final JavaTestKit probe2 = new JavaTestKit(system);
|
|
|
|
final Sink<Integer, ?> sink1 = Sink.actorRef(probe1.getRef(), "done1");
|
|
final Sink<Integer, ?> sink2 = Sink.actorRef(probe2.getRef(), "done2");
|
|
|
|
final Sink<Integer, ?> sink = Sink.combine(sink1, sink2, new ArrayList<Sink<Integer, ?>>(),
|
|
new Function<Integer, Graph<UniformFanOutShape<Integer, Integer>, NotUsed>>() {
|
|
public Graph<UniformFanOutShape<Integer, Integer>, NotUsed> apply(Integer elem) {
|
|
return Broadcast.create(elem);
|
|
}
|
|
}
|
|
);
|
|
|
|
Source.from(Arrays.asList(0, 1)).runWith(sink, materializer);
|
|
|
|
probe1.expectMsgEquals(0);
|
|
probe2.expectMsgEquals(0);
|
|
probe1.expectMsgEquals(1);
|
|
probe2.expectMsgEquals(1);
|
|
|
|
probe1.expectMsgEquals("done1");
|
|
probe2.expectMsgEquals("done2");
|
|
}
|
|
|
|
@Test
|
|
public void mustBeAbleToUseContramap() throws Exception {
|
|
List<Integer> out = Source.range(0, 2).toMat(Sink.<Integer>seq().contramap(x -> x + 1), Keep.right())
|
|
.run(materializer).toCompletableFuture().get(3, TimeUnit.SECONDS);
|
|
|
|
assertEquals(Arrays.asList(1, 2, 3), out);
|
|
}
|
|
|
|
public void mustSuitablyOverrideAttributeHandlingMethods() {
|
|
@SuppressWarnings("unused")
|
|
final Sink<Integer, CompletionStage<Integer>> s =
|
|
Sink.<Integer> head().withAttributes(Attributes.name("")).addAttributes(Attributes.asyncBoundary()).named("");
|
|
}
|
|
}
|