04ab4ebb9c
* -str #21423 remove deprecated Stage references * include mima filters for removed classes
373 lines
11 KiB
Scala
373 lines
11 KiB
Scala
/**
|
|
* Copyright (C) 2015-2016 Lightbend Inc. <http://www.lightbend.com>
|
|
*/
|
|
package akka.stream.impl.fusing
|
|
|
|
import akka.NotUsed
|
|
import akka.stream.testkit.StreamSpec
|
|
import akka.stream.{ OverflowStrategy, Attributes }
|
|
import akka.stream.scaladsl.{ Merge, Broadcast, Balance, Zip }
|
|
import GraphInterpreter._
|
|
|
|
class GraphInterpreterSpec extends StreamSpec with GraphInterpreterSpecKit {
|
|
import GraphStages._
|
|
|
|
"GraphInterpreter" must {
|
|
|
|
// Reusable components
|
|
val identity = GraphStages.identity[Int]
|
|
val detach = detacher[Int]
|
|
val zip = Zip[Int, String]
|
|
val bcast = Broadcast[Int](2)
|
|
val merge = Merge[Int](2)
|
|
val balance = Balance[Int](2)
|
|
|
|
"implement identity" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink = new DownstreamProbe[Int]("sink")
|
|
|
|
builder(identity)
|
|
.connect(source, identity.in)
|
|
.connect(identity.out, sink)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, 1)))
|
|
}
|
|
|
|
"implement chained identity" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink = new DownstreamProbe[Int]("sink")
|
|
|
|
// Constructing an assembly by hand and resolving ambiguities
|
|
val assembly = new GraphAssembly(
|
|
stages = Array(identity, identity),
|
|
originalAttributes = Array(Attributes.none, Attributes.none),
|
|
ins = Array(identity.in, identity.in, null),
|
|
inOwners = Array(0, 1, -1),
|
|
outs = Array(null, identity.out, identity.out),
|
|
outOwners = Array(-1, 0, 1))
|
|
|
|
manualInit(assembly)
|
|
interpreter.attachDownstreamBoundary(2, sink)
|
|
interpreter.attachUpstreamBoundary(0, source)
|
|
interpreter.init(null)
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, 1)))
|
|
}
|
|
|
|
"implement detacher stage" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink = new DownstreamProbe[Int]("sink")
|
|
|
|
builder(detach)
|
|
.connect(source, detach.shape.in)
|
|
.connect(detach.shape.out, sink)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, 1), RequestOne(source)))
|
|
|
|
// Source waits
|
|
source.onNext(2)
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
// "pushAndPull"
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(OnNext(sink, 2), RequestOne(source)))
|
|
|
|
// Sink waits
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
// "pushAndPull"
|
|
source.onNext(3)
|
|
lastEvents() should ===(Set(OnNext(sink, 3), RequestOne(source)))
|
|
}
|
|
|
|
"implement Zip" in new TestSetup {
|
|
val source1 = new UpstreamProbe[Int]("source1")
|
|
val source2 = new UpstreamProbe[String]("source2")
|
|
val sink = new DownstreamProbe[(Int, String)]("sink")
|
|
|
|
builder(zip)
|
|
.connect(source1, zip.in0)
|
|
.connect(source2, zip.in1)
|
|
.connect(zip.out, sink)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source1), RequestOne(source2)))
|
|
|
|
source1.onNext(42)
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
source2.onNext("Meaning of life")
|
|
lastEvents() should ===(Set(OnNext(sink, (42, "Meaning of life")), RequestOne(source1), RequestOne(source2)))
|
|
}
|
|
|
|
"implement Broadcast" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink1 = new DownstreamProbe[Int]("sink1")
|
|
val sink2 = new DownstreamProbe[Int]("sink2")
|
|
|
|
builder(bcast)
|
|
.connect(source, bcast.in)
|
|
.connect(bcast.out(0), sink1)
|
|
.connect(bcast.out(1), sink2)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink1.requestOne()
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink2.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink1, 1), OnNext(sink2, 1)))
|
|
|
|
}
|
|
|
|
"implement broadcast-zip" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink = new DownstreamProbe[(Int, Int)]("sink")
|
|
val zip = new Zip[Int, Int]
|
|
|
|
builder(zip, bcast)
|
|
.connect(source, bcast.in)
|
|
.connect(bcast.out(0), zip.in0)
|
|
.connect(bcast.out(1), zip.in1)
|
|
.connect(zip.out, sink)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, (1, 1)), RequestOne(source)))
|
|
|
|
sink.requestOne()
|
|
source.onNext(2)
|
|
lastEvents() should ===(Set(OnNext(sink, (2, 2)), RequestOne(source)))
|
|
|
|
}
|
|
|
|
"implement zip-broadcast" in new TestSetup {
|
|
val source1 = new UpstreamProbe[Int]("source1")
|
|
val source2 = new UpstreamProbe[Int]("source2")
|
|
val sink1 = new DownstreamProbe[(Int, Int)]("sink")
|
|
val sink2 = new DownstreamProbe[(Int, Int)]("sink2")
|
|
val zip = new Zip[Int, Int]
|
|
val bcast = Broadcast[(Int, Int)](2)
|
|
|
|
builder(bcast, zip)
|
|
.connect(source1, zip.in0)
|
|
.connect(source2, zip.in1)
|
|
.connect(zip.out, bcast.in)
|
|
.connect(bcast.out(0), sink1)
|
|
.connect(bcast.out(1), sink2)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink1.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source1), RequestOne(source2)))
|
|
|
|
sink2.requestOne()
|
|
|
|
source1.onNext(1)
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
source2.onNext(2)
|
|
lastEvents() should ===(Set(OnNext(sink1, (1, 2)), OnNext(sink2, (1, 2)), RequestOne(source1), RequestOne(source2)))
|
|
|
|
}
|
|
|
|
"implement merge" in new TestSetup {
|
|
val source1 = new UpstreamProbe[Int]("source1")
|
|
val source2 = new UpstreamProbe[Int]("source2")
|
|
val sink = new DownstreamProbe[Int]("sink")
|
|
|
|
builder(merge)
|
|
.connect(source1, merge.in(0))
|
|
.connect(source2, merge.in(1))
|
|
.connect(merge.out, sink)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source1), RequestOne(source2)))
|
|
|
|
source1.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, 1), RequestOne(source1)))
|
|
|
|
source2.onNext(2)
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(OnNext(sink, 2), RequestOne(source2)))
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
source2.onNext(3)
|
|
lastEvents() should ===(Set(OnNext(sink, 3), RequestOne(source2)))
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
source1.onNext(4)
|
|
lastEvents() should ===(Set(OnNext(sink, 4), RequestOne(source1)))
|
|
|
|
}
|
|
|
|
"implement balance" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink1 = new DownstreamProbe[Int]("sink1")
|
|
val sink2 = new DownstreamProbe[Int]("sink2")
|
|
|
|
builder(balance)
|
|
.connect(source, balance.in)
|
|
.connect(balance.out(0), sink1)
|
|
.connect(balance.out(1), sink2)
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink1.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
sink2.requestOne()
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink1, 1), RequestOne(source)))
|
|
|
|
source.onNext(2)
|
|
lastEvents() should ===(Set(OnNext(sink2, 2)))
|
|
}
|
|
|
|
"implement non-divergent cycle" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink = new DownstreamProbe[Int]("sink")
|
|
|
|
builder(merge, balance)
|
|
.connect(source, merge.in(0))
|
|
.connect(merge.out, balance.in)
|
|
.connect(balance.out(0), sink)
|
|
.connect(balance.out(1), merge.in(1))
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, 1), RequestOne(source)))
|
|
|
|
// Token enters merge-balance cycle and gets stuck
|
|
source.onNext(2)
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
// Unstuck it
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(OnNext(sink, 2)))
|
|
|
|
}
|
|
|
|
"implement divergent cycle" in new TestSetup {
|
|
val source = new UpstreamProbe[Int]("source")
|
|
val sink = new DownstreamProbe[Int]("sink")
|
|
|
|
builder(detach, balance, merge)
|
|
.connect(source, merge.in(0))
|
|
.connect(merge.out, balance.in)
|
|
.connect(balance.out(0), sink)
|
|
.connect(balance.out(1), detach.shape.in)
|
|
.connect(detach.shape.out, merge.in(1))
|
|
.init()
|
|
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext(1)
|
|
lastEvents() should ===(Set(OnNext(sink, 1), RequestOne(source)))
|
|
|
|
// Token enters merge-balance cycle and spins until event limit
|
|
// Without the limit this would spin forever (where forever = Int.MaxValue iterations)
|
|
source.onNext(2, eventLimit = 1000)
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
// The cycle is still alive and kicking, just suspended due to the event limit
|
|
interpreter.isSuspended should be(true)
|
|
|
|
// Do to the fairness properties of both the interpreter event queue and the balance stage
|
|
// the element will eventually leave the cycle and reaches the sink.
|
|
// This should not hang even though we do not have an event limit set
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(OnNext(sink, 2)))
|
|
|
|
// The cycle is now empty
|
|
interpreter.isSuspended should be(false)
|
|
}
|
|
|
|
"implement buffer" in new TestSetup {
|
|
val source = new UpstreamProbe[String]("source")
|
|
val sink = new DownstreamProbe[String]("sink")
|
|
val buffer = Buffer[String](2, OverflowStrategy.backpressure)
|
|
|
|
builder(buffer)
|
|
.connect(source, buffer.in)
|
|
.connect(buffer.out, sink)
|
|
.init()
|
|
|
|
stepAll()
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set.empty)
|
|
|
|
source.onNext("A")
|
|
lastEvents() should ===(Set(RequestOne(source), OnNext(sink, "A")))
|
|
|
|
source.onNext("B")
|
|
lastEvents() should ===(Set(RequestOne(source)))
|
|
|
|
source.onNext("C", eventLimit = 0)
|
|
sink.requestOne()
|
|
lastEvents() should ===(Set(OnNext(sink, "B"), RequestOne(source)))
|
|
|
|
sink.requestOne(eventLimit = 0)
|
|
source.onComplete(eventLimit = 3)
|
|
// OnComplete arrives early due to push chasing
|
|
lastEvents() should ===(Set(OnNext(sink, "C"), OnComplete(sink)))
|
|
}
|
|
}
|
|
|
|
}
|