Skip to content

Commit ba639f6

Browse files
authored
chore: Add the missing EmptySource case to TraversalBuilder (#1743)
1 parent 864777e commit ba639f6

File tree

2 files changed

+18
-1
lines changed

2 files changed

+18
-1
lines changed

stream-tests/src/test/scala/org/apache/pekko/stream/impl/TraversalBuilderSpec.scala

+17-1
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ import org.apache.pekko
1717
import pekko.NotUsed
1818
import pekko.stream._
1919
import pekko.stream.impl.TraversalTestUtils._
20-
import pekko.stream.scaladsl.Keep
20+
import pekko.stream.scaladsl.{ Keep, Source }
2121
import pekko.testkit.PekkoSpec
2222

2323
class TraversalBuilderSpec extends PekkoSpec {
@@ -447,4 +447,20 @@ class TraversalBuilderSpec extends PekkoSpec {
447447
}
448448
}
449449

450+
"find Source.empty via TraversalBuilder with isEmptySource" in {
451+
val emptySource = EmptySource
452+
TraversalBuilder.isEmptySource(emptySource) should be(true)
453+
}
454+
455+
"find javadsl Source.empty via TraversalBuilder with isEmptySource" in {
456+
import pekko.stream.javadsl.Source
457+
val emptySource = Source.empty()
458+
TraversalBuilder.isEmptySource(emptySource) should be(true)
459+
}
460+
461+
"find scaldsl Source.empty via TraversalBuilder with isEmptySource" in {
462+
val emptySource = Source.empty
463+
TraversalBuilder.isEmptySource(emptySource) should be(true)
464+
}
465+
450466
}

stream/src/main/scala/org/apache/pekko/stream/impl/TraversalBuilder.scala

+1
Original file line numberDiff line numberDiff line change
@@ -386,6 +386,7 @@ import pekko.util.unused
386386
def isEmptySource(graph: Graph[SourceShape[_], _]): Boolean = graph match {
387387
case source: scaladsl.Source[_, _] if source eq scaladsl.Source.empty => true
388388
case source: javadsl.Source[_, _] if source eq javadsl.Source.empty() => true
389+
case EmptySource => true
389390
case _ => false
390391
}
391392

0 commit comments

Comments
 (0)