raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Make the router doc test actually verify that it works #27184

Browse source
This commit is contained in:
Johan Andrén 2019-07-09 22:53:12 +02:00
parent 5388a65679
commit a054ae52c5
1 changed files with 25 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

29
akka-actor-typed-tests/src/test/scala/docs/akka/typed/RouterSpec.scala
View file

@ -23,7 +23,7 @@ object RouterSpec {
sealed trait Command
case class DoLog(text: String) extends Command
val behavior: Behavior[Command] = Behaviors.setup { ctx =>
def apply(): Behavior[Command] = Behaviors.setup { ctx =>
ctx.log.info("Starting worker")
Behaviors.receiveMessage {
@ -39,17 +39,26 @@ object RouterSpec {
val serviceKey = ServiceKey[Worker.Command]("log-worker")
}
class RouterSpec extends ScalaTestWithActorTestKit with WordSpecLike {
class RouterSpec extends ScalaTestWithActorTestKit("akka.loglevel=warning") with WordSpecLike {
import RouterSpec._
"The routing sample" must {
"show pool routing" in {
// trixery to monitor worker but make the sample look like we use it directly
val probe = createTestProbe[RouterSpec.Worker.Command]()
object Worker {
def apply(): Behavior[RouterSpec.Worker.Command] =
Behaviors.monitor(probe.ref, RouterSpec.Worker())
def DoLog(text: String) = RouterSpec.Worker.DoLog(text)
}
spawn(Behaviors.setup[Unit] { ctx =>
// #pool
val pool = Routers.pool(poolSize = 4)(() =>
// make sure the workers are restarted if they fail
Behaviors.supervise(Worker.behavior).onFailure[Exception](SupervisorStrategy.restart))
Behaviors.supervise(Worker()).onFailure[Exception](SupervisorStrategy.restart))
val router = ctx.spawn(pool, "worker-pool")
(0 to 10).foreach { n =>
@ -63,15 +72,25 @@ class RouterSpec extends ScalaTestWithActorTestKit with WordSpecLike {
Behaviors.empty
})
probe.receiveMessages(10)
}
"show group routing" in {
// trixery to monitor worker but make the sample look like we use it directly
val probe = createTestProbe[RouterSpec.Worker.Command]()
object Worker {
def apply(): Behavior[RouterSpec.Worker.Command] =
Behaviors.monitor(probe.ref, RouterSpec.Worker())
def DoLog(text: String) = RouterSpec.Worker.DoLog(text)
}
spawn(Behaviors.setup[Unit] { ctx =>
// #group
// this would likely happen elsewhere - if we create it locally we
// can just as well use a pool
val worker = ctx.spawn(Worker.behavior, "worker")
val worker = ctx.spawn(Worker(), "worker")
ctx.system.receptionist ! Receptionist.Register(serviceKey, worker)
val group = Routers.group(serviceKey);
@ -88,6 +107,8 @@ class RouterSpec extends ScalaTestWithActorTestKit with WordSpecLike {
Behaviors.empty
})
probe.receiveMessages(10)
}
}
}