#19931 Track pending requests properly on expectRequest

2016-06-27 12:56:29 +02:00 · 2016-06-27 12:56:29 +02:00 · e54d2933bd
commit e54d2933bd
parent 1ca40730ef
2 changed files with 20 additions and 1 deletions
--- a/akka-stream-testkit/src/main/scala/akka/stream/testkit/StreamTestKit.scala
+++ b/akka-stream-testkit/src/main/scala/akka/stream/testkit/StreamTestKit.scala
@ -168,7 +168,11 @@ object TestPublisher {
      this
    }

-    def expectRequest(): Long = subscription.expectRequest()
+    def expectRequest(): Long = {
+      val requests = subscription.expectRequest()
+      pendingRequests += requests
+      requests
+    }

    def expectCancellation(): Self = {
      subscription.expectCancellation()