From ea8e3ace334c553998eb317e2e4c523ee79cc729 Mon Sep 17 00:00:00 2001
From: Christopher Batey <christopher.batey@gmail.com>
Date: Thu, 21 Feb 2019 15:42:24 +0000
Subject: [PATCH] Harden RateTransformationDocSpec: Use test patience rather
 than hadcode #26262

---
 .../test/scala/docs/stream/RateTransformationDocSpec.scala | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/akka-docs/src/test/scala/docs/stream/RateTransformationDocSpec.scala b/akka-docs/src/test/scala/docs/stream/RateTransformationDocSpec.scala
index c910e96651..919649a379 100644
--- a/akka-docs/src/test/scala/docs/stream/RateTransformationDocSpec.scala
+++ b/akka-docs/src/test/scala/docs/stream/RateTransformationDocSpec.scala
@@ -6,15 +6,16 @@ package docs.stream
 
 import akka.stream._
 import akka.stream.scaladsl._
-import akka.stream.testkit._
 import akka.stream.testkit.scaladsl._
+
 import scala.util.Random
 import scala.math._
-import scala.concurrent.Await
 import scala.concurrent.duration._
 import scala.collection.immutable
 import akka.testkit.{ AkkaSpec, TestLatch }
 
+import scala.concurrent.Await
+
 class RateTransformationDocSpec extends AkkaSpec {
 
   implicit val materializer = ActorMaterializer()
@@ -88,7 +89,7 @@ class RateTransformationDocSpec extends AkkaSpec {
       .grouped(10)
       .runWith(Sink.head)
 
-    val extrapolated = Await.result(fut, 100.millis)
+    val extrapolated = fut.futureValue
     extrapolated.size shouldBe 10
     extrapolated.sum shouldBe 10 * initial
   }