diff --git a/akka-stream-tests/src/test/java/akka/stream/javadsl/SinkTest.java b/akka-stream-tests/src/test/java/akka/stream/javadsl/SinkTest.java index cf97664aa4..a7027160b4 100644 --- a/akka-stream-tests/src/test/java/akka/stream/javadsl/SinkTest.java +++ b/akka-stream-tests/src/test/java/akka/stream/javadsl/SinkTest.java @@ -3,37 +3,20 @@ */ package akka.stream.javadsl; -import akka.actor.ActorRef; -import akka.actor.ActorSystem; -import akka.dispatch.Foreach; -import akka.dispatch.Futures; -import akka.dispatch.OnSuccess; -import akka.japi.Pair; -import akka.japi.Util; -import akka.stream.FlowMaterializer; -import akka.stream.MaterializerSettings; -import akka.stream.OverflowStrategy; -import akka.stream.Transformer; -import akka.stream.javadsl.japi.*; -import akka.stream.testkit.AkkaSpec; -import akka.testkit.JavaTestKit; +import java.util.ArrayList; +import java.util.List; + import org.junit.ClassRule; import org.junit.Test; import org.reactivestreams.Publisher; -import scala.Option; -import scala.collection.immutable.Seq; + import scala.concurrent.Await; import scala.concurrent.Future; import scala.concurrent.duration.Duration; -import scala.concurrent.duration.FiniteDuration; -import scala.runtime.BoxedUnit; -import scala.util.Try; - -import java.util.*; -import java.util.concurrent.Callable; -import java.util.concurrent.TimeUnit; - -import static org.junit.Assert.assertEquals; +import akka.actor.ActorSystem; +import akka.stream.FlowMaterializer; +import akka.stream.MaterializerSettings; +import akka.stream.testkit.AkkaSpec; public class SinkTest { @@ -48,8 +31,17 @@ public class SinkTest { @Test public void mustBeAbleToUseFanoutPublisher() throws Exception { - KeyedSink> pubSink = Sink.fanoutPublisher(2, 2); - Publisher publisher = Source.from(new ArrayList()).runWith(pubSink, materializer); + final KeyedSink> pubSink = Sink.fanoutPublisher(2, 2); + final Publisher publisher = Source.from(new ArrayList()).runWith(pubSink, materializer); + } + + @Test + public void mustBeAbleToUseFuture() throws Exception { + final KeyedSink> futSink = Sink.future(); + final List list = new ArrayList(); + list.add(1); + final Future future = Source.from(list).runWith(futSink, materializer); + assert Await.result(future, Duration.create("1 second")).equals(1); } }