在单元测试中验证 akka-stream 源答案

【问题标题】：Validating akka-stream Source in unit test在单元测试中验证 akka-stream 源
【发布时间】：2022-01-11 15:25:24
【问题描述】：

在单元测试中验证 akka-stream 源的惯用方法是什么？我是这样做的：

f.service.fetchData(id).flatMap {
          case Right(source) => {
            // TODO: I need to validate here that source contains "Test value"
          }
          case Left(_) => fail("Wrongly responded with error")
        }

来源基本上是：

Source.single(ByteString("Test value"))

我尝试通过将 Source 连接到 Sink 并检查发出的值来执行，但断言似乎不起作用。

【问题讨论】：

得到它的工作： source .fold("") { case (x, y) => x + y.decodeString("UTF-8") } .runWith(Sink.head) 。等待

标签： scala unit-testing akka-stream

【解决方案1】：

测试 Sources、Flows 和 Sinks 的正确方法是使用test probes。

鉴于以下来源，您要测试的女巫逻辑

val source = Source(Seq("TestValue")).collect {
  case s @ "TestValue" => Right[String, String](s)
  case _               => Left[String, String]("error")
}

（我知道Left 永远不会在这里触发，但这只是一个例子）

现在，您可以定义一个连接到给定源的TestSink。生成的图表按以下方式执行

"assert correct value" in {
  implicit val system: ActorSystem = ???
  val probe = source.toMat(TestSink.probe)(Keep.right).run()
  probe.request(1)
  probe.expectNext(Right("TestValue"))
}

TestSink.probe 是一个实现为TestSubscriber.Probe[T] 的接收器，它提供对流的控制。由于它是一个接收器，它需要发出对元素的需求。它通过request(1) 或请求一个元素来完成。然后是断言sink.expectNext(Right("TestValue"))，它检查是否收到了正确的值。

还有一个名为 TestSource.probe 的对应项允许测试 Sink。

通过结合它们，您可以测试Flow。

【讨论】：