#24097 Try a few times to get an exception

2017-12-06 20:52:47 +07:00 · 2017-12-06 20:52:47 +07:00 · 2fd9bb736b
commit 2fd9bb736b
parent f5d28a8e07
1 changed files with 4 additions and 1 deletions
--- a/akka-stream-tests/src/test/scala/akka/stream/io/OutputStreamSourceSpec.scala
+++ b/akka-stream-tests/src/test/scala/akka/stream/io/OutputStreamSourceSpec.scala
@ -160,7 +160,10 @@ class OutputStreamSourceSpec extends StreamSpec(UnboundedMailboxConfig) {

      s.cancel()
      sourceProbe.expectMsg(GraphStageMessages.DownstreamFinish)
-      the[Exception] thrownBy outputStream.write(bytesArray) shouldBe a[IOException]
+
+      awaitAssert {
+        the[Exception] thrownBy outputStream.write(bytesArray) shouldBe a[IOException]
+      }
    }

    "fail to materialize with zero sized input buffer" in {