pekko/akka-docs/src/test/scala/docs/stream/cookbook/RecipeSourceFromFunction.scala

26 lines
636 B
Scala

/**
* Copyright (C) 2016-2017 Lightbend Inc. <http://www.lightbend.com>
*/
package docs.stream.cookbook
import java.util.UUID
import akka.NotUsed
import akka.stream.scaladsl._
class RecipeSourceFromFunction extends RecipeSpec {
"A source that repeatedly evaluates a function" must {
"be a mapping of Source.repeat" in {
def builderFunction(): String = UUID.randomUUID.toString
//#source-from-function
val source = Source.repeat(NotUsed).map(_ builderFunction())
//#source-from-function
val f = source.take(2).runWith(Sink.seq)
f.futureValue.distinct.size should ===(2)
}
}
}