如何使用Akka TestKit响应失败的Ask模式?

时间:2015-09-19 14:49:08

标签: scala akka akka-testkit

我有一个Akka Actor,它使用Ask模式从子actor中检索Future,并对成功和失败起作用。我无法弄清楚如何模仿儿童演员并以失败回应。

以下是代码:

import java.util.concurrent.TimeUnit
import akka.actor.{Actor, ActorRef, ActorSystem, Props}
import akka.pattern.ask
import akka.testkit.{ImplicitSender, TestKitBase, TestProbe}
import akka.util.Timeout
import org.specs2.mutable.Specification
import org.specs2.specification.Scope
import scala.util.{Failure, Success}

class Parent(child: ActorRef) extends Actor {
  implicit val timeout = Timeout(5, TimeUnit.SECONDS)
  import context.dispatcher

  override def receive: Receive = {
    case "go" => {
      val senderRef = sender()
      (child ? "question").mapTo[String] onComplete {
        case Success("answer") =>
          senderRef ! "child responded with a successful answer"
        case Failure(throwable) =>
          senderRef ! "child responded with a failure"
      }
    }
  }
}

class SimplifiedProblemSpec extends Specification {

  "The Parent Actor" should {
    "act on success" in new TestScope {
      parent ! "go"
      childProbe.expectMsg("question")
      childProbe.reply("answer")
      expectMsg("child responded with a successful answer")
    }
    "act on failure" in new TestScope {
      parent ! "go"
      childProbe.expectMsg("question")
      // How to reply with failure?
      expectMsg("child responded with a failure")
    }
  }

}

abstract class TestScope extends Scope with TestKitBase with ImplicitSender {
  implicit lazy val system: ActorSystem = ActorSystem()

  val childProbe = TestProbe()
  val parent = system.actorOf(Props(classOf[Parent], childProbe.ref), "Parent")

}

非常感谢任何帮助。

谢谢

2 个答案:

答案 0 :(得分:4)

您可以使用Status.Failure包中的akka.actor案例类。来自docs:

  

此类/消息类型优选用于指示执行某些操作的失败。例如,它用于在使用AskSupport时发出故障信号(询问/?)。

答案 1 :(得分:0)

老实说,我并非100%确定这是你要求的,但万一你可以尝试这样的事情:

case class FailureMessage(message: String) extends Exception(message)

object MyActor {
    def props = Props(classOf[MyActor])
}

class MyActor extends Actor {
    override def receive = {
        case "cool" => sender ! Status.Success("cool")
        case boom => sender ! Status.Failure(FailureMessage("boom"))
   }
}

class MyActorSpec extends TestKit(ActorSystem("system")) with ImplicitSender with WordSpecLike with Matchers {

  implicit val timeout = Timeout(1 second)

  override def afterAll = TestKit.shutdownActorSystem(system)

  "system" must {
      "return Failure if boom" in {
          val actor = system.actorOf(MyActor.props)
          actor ! "boom"
          expectMsg(Status.Failure(FailureMessage("boom")))
      }
  }