如何在Akka Cluster中测试订阅者的接收方法?

时间:2017-01-01 14:41:43

标签: scala unit-testing akka akka-cluster distributedpubsub

我有以下Subscriber抽象基类:

abstract class Subscriber(topics: Seq[String]) extends Actor with ActorLogging {
  import DistributedPubSubMediator.{ Subscribe, SubscribeAck }

  val mediator = DistributedPubSub(context.system).mediator

  // subscribe to each topic
  topics.foreach{mediator ! Subscribe(_, self)}

  def receive = {
    case SubscribeAck(Subscribe(name, None, `self`)) ⇒
      log.info(s"Subscribing to $name")
  }
}

我想测试它是否接收发布到订阅子类的主题的消息。一些简单的伪代码证明了以下内容:

val topic = "foo"

class FooSubscriber extends Subscriber(Seq(topic))

val fooSubActor = system.actorOf(Props[FooSubscriber])    
val mediator = DistributedPubSub(system).mediator
val msg = "This is a string"

// Publish the msg to the "foo" topic.
mediator ! Publish(topic, msg)

fooSubActor.expectMsg(msg)

我知道对特定演员收到的消息进行断言的唯一方法是TestProbe,但我不知道如何让TestProbe扩展我的课程。

通常,Akka docs包含大量带有关联测试套件的示例代码,但我无法在与测试receive方法相关的Akka Cluster文档中找到任何内容。

有人有建议吗?

1 个答案:

答案 0 :(得分:0)

这是一个教科书示例,其中依赖注入有助于提高可测试性。

如果您在Subscriber构造函数中收到DistPubSub的介体而不是直接询问它,您可以单独测试Suscriber actor,而无需将DistPubSub用作测试夹具的一部分。