chore: Fix flaky test in unsafeOptionalDataVia

apache · Dec 17, 2024 · 42d3a5d · 42d3a5d
1 parent 53761e5
commit 42d3a5d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/stream-tests/src/test/scala/org/apache/pekko/stream/scaladsl/SourceWithContextSpec.scala b/stream-tests/src/test/scala/org/apache/pekko/stream/scaladsl/SourceWithContextSpec.scala
@@ -248,7 +248,7 @@ class SourceWithContextSpec extends StreamSpec {
       )(Keep.none)
         .runWith(TestSink.probe[(Option[Int], Int)])
         .request(4)
-        .expectNext((Some(1), 1), (None, 2), (None, 3), (Some(4), 4))
+        .expectNextN(List((Some(1), 1), (None, 2), (None, 3), (Some(4), 4)))
         .expectComplete()
     }
   }

diff --git a/stream/src/main/scala/org/apache/pekko/stream/scaladsl/SourceWithContext.scala b/stream/src/main/scala/org/apache/pekko/stream/scaladsl/SourceWithContext.scala
@@ -68,7 +68,7 @@ object SourceWithContext {
         }
 
         val filterUnavailable = Flow[(Option[SOut], Ctx)].collect {
-          case (None, ctx) => (Option.empty[FOut], ctx)
+          case x @ (None, _) => x.asInstanceOf[(Option[FOut], Ctx)]
         }
 
         val mapIntoOption = Flow[(FOut, Ctx)].map {