0

我想测试以下场景：

假设我有一个父演员，它创建了两个像这样的子演员。

class A extends Actor {
  def getActorOf(props: Props) = {
    context.actorOf(props, props.clazz.getTypeName)
  }

  def receive: Receive = {
    case "ping" => {
      val bChild = getActorOf(Props[B])
      val cChild = getActorOf(Props[C])
      Seq(bChild, cChild)
        .foreach(child => child ! "ping forwarded")
    }
  }
}

我想测试一下，如果父母得到消息，他会向自己的两个孩子'ping'发送 '消息。ping forwarded'

是否可以使用 TestKit 做到这一点？

4

1 回答 1

2

大概是这样的吧？

 class TestMe extends A {
   val (probeB, probeC) = (TestProbe(), TestProbe())
   override def getActorOf(props: Props) = props match {
     case Props(_, classOf[B], _) => probeB.ref
     case Props(_, classOf[C], _) => probeC.ref
   }
 }

 val fixture = TestActorRef[TestMe](Props[TestMe])
 fixture ! "ping" 
 fixture.underlyingActor.probeB.expectMsg("ping forwarded")
 fixture.underlyingActor.probeB.expectMsg("ping forwarded")

就个人而言，我更喜欢尽可能“传统”的方法：

 trait Forwarder {
   def startAndForward[T : ClassTag](message: Any)(implicit context: ActorContext) = {
     val actor = context.actorOf(Props[T])
     actor ! message
     actor
  }
}
object Forwarder extends Forwarder

 class A(f: Forwarder = Forwarder) extends Actor {

   def receive: Receive = {
     case m@"ping" => 
       f.startAndForward[B]("ping forwarded")     
       f.startAndForward[C]("ping forwarded")
       sender ! "pong"
   }
 }

现在，您可以直接运行测试：

 val fwd = mock[Forwarder]
 val fixture = context.actorOf(Props(new A(fwd)))
 fixture.ask("ping").futureValue shouldBe "pong"
 verify(fwd).startAndForward[B](ArgumentMatchers.eq("ping forwarded"))(any, any)
 verify(fwd).startAndForward[C](ArgumentMatchers.eq("ping forwarded"))(any, any)
于 2019-02-03T19:04:36.713 回答