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

Question

我有以下 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有大量带有相关测试套件的示例代码，但我在 Akka Cluster 文档中找不到与测试 receive 相关的任何内容方法。

有人有建议吗？

Answer 1

这是一个教科书示例，说明依赖注入(inject)有助于提高可测试性。

如果您在 Subscriber 构造函数中收到 DistPubSub 的中介，而不是直接请求它，您可以单独测试 Suscriber actor，而无需使用 DistPubSub 作为测试夹具的一部分。