!peq #19324 update persistence-query to stream 2.0.1

akka-persistence-query/src/test/scala/akka/persistence/query/DummyReadJournal.scala

@@ -14,7 +14,7 @@ import scala.runtime.BoxedUnit
  */
 class DummyReadJournal extends scaladsl.ReadJournal with scaladsl.AllPersistenceIdsQuery {
   override def allPersistenceIds(): Source[String, Unit] =
-    Source(() ⇒ Iterator.from(0)).map(_.toString)
+    Source.fromIterator(() ⇒ Iterator.from(0)).map(_.toString)
 }
 
 object DummyReadJournal {

akka-persistence-query/src/test/scala/akka/persistence/query/journal/leveldb/AllPersistenceIdsSpec.scala

@@ -45,10 +45,12 @@ class AllPersistenceIdsSpec extends AkkaSpec(AllPersistenceIdsSpec.config)
       expectMsg("c1-done")
 
       val src = queries.currentPersistenceIds()
-      src.runWith(TestSink.probe[String])
-        .request(5)
-        .expectNextUnordered("a", "b", "c")
-        .expectComplete()
+      val probe = src.runWith(TestSink.probe[String])
+      probe.within(10.seconds) {
+        probe.request(5)
+          .expectNextUnordered("a", "b", "c")
+          .expectComplete()
+      }
     }
 
     "find new persistenceIds" in {
@@ -58,20 +60,22 @@ class AllPersistenceIdsSpec extends AkkaSpec(AllPersistenceIdsSpec.config)
 
       val src = queries.allPersistenceIds()
       val probe = src.runWith(TestSink.probe[String])
-        .request(5)
-        .expectNextUnorderedN(List("a", "b", "c", "d"))
+      probe.within(10.seconds) {
+        probe.request(5)
+          .expectNextUnorderedN(List("a", "b", "c", "d"))
 
-      system.actorOf(TestActor.props("e")) ! "e1"
-      probe.expectNext("e")
+        system.actorOf(TestActor.props("e")) ! "e1"
+        probe.expectNext("e")
 
-      val more = (1 to 100).map("f" + _)
-      more.foreach { p ⇒
-        system.actorOf(TestActor.props(p)) ! p
+        val more = (1 to 100).map("f" + _)
+        more.foreach { p ⇒
+          system.actorOf(TestActor.props(p)) ! p
+        }
+
+        probe.request(100)
+        probe.expectNextUnorderedN(more)
       }
 
-      probe.request(100)
-      probe.expectNextUnorderedN(more)
-
     }
   }