scala - 如何模拟 Akka Actor 来对课程进行单元测试？

Question

我有一个控制器类，它控制发送到注入控制器的 Akka 演员的请求。

控制器代码：

class Controller(actor: ActorRef) {
  def control(msg: String): Future[String] = {
    actor.ask(msg)(Timeout(2 seconds)).mapTo[String]
  }
}

我的演员的代码是：

class ActorA extends Actor {
  override def receive: Receive = {
    case msg: String => sender ! msg
    case msg: Int => sender ! msg.toString
    case _ => "Invalid command!"
}

现在我需要模拟 ActorA 的行为来对 Controller 进行单元测试。有没有办法通过 Akka TestKit 做到这一点？

Answer 1

使用TestProbe. 从测试文档：

val probe = TestProbe()
val future = probe.ref ? "hello"
probe.expectMsg(0 millis, "hello") 
probe.reply("world")
assert(future.isCompleted && future.value == Some(Success("world")))