Small changes to see if it helps fixing #895

akka · Nov 20, 2019 · f050c56 · f050c56
1 parent acb2497
commit f050c56
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/src/test/scala/akka/kafka/internal/CommittingWithMockSpec.scala b/tests/src/test/scala/akka/kafka/internal/CommittingWithMockSpec.scala
@@ -211,7 +211,7 @@ class CommittingWithMockSpec(_system: ActorSystem)
     mock.enqueue(msgs.map(toRecord))
 
     probe.request(count.toLong)
-    val done = Future.sequence(probe.expectNextN(count.toLong).map(_.committableOffset.commitInternal()))
+    val allCommits = Future.sequence(probe.expectNextN(count.toLong).map(_.committableOffset.commitInternal()))
 
     withClue("the commits are aggregated to a low number of calls to commitAsync:") {
       awaitAssert {
@@ -226,8 +226,8 @@ class CommittingWithMockSpec(_system: ActorSystem)
       case (offsets, callback) => callback.onComplete(offsets.asJava, null)
     }
 
-    Await.result(done, remainingOrDefault)
-    Await.result(control.shutdown(), remainingOrDefault)
+    allCommits.futureValue should have size (count.toLong)
+    control.shutdown().futureValue shouldBe Done
   }
 
   it should "support commit batching" in assertAllStagesStopped {