raboof/pekko
1
0
Fork 0
Code Issues Pull requests Projects Releases 6 Packages Wiki Activity Actions 10

Execute tasks from Actors constructor on creation of BehaviorTestKit #24703

Browse source
This commit is contained in:
Pritam Kadam 2018-03-13 21:45:05 +05:30 • committed by Johan Andrén
parent 563c7fbcf0
commit 278ef5fb8c
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
akka-testkit-typed/src/main/scala/akka/testkit/typed/internal/BehaviorTestKitImpl.scala
View file

@ -31,6 +31,9 @@ private[akka] final class BehaviorTestKitImpl[T](_name: String, _initialBehavior
private var currentUncanonical = _initialBehavior
private var current = Behavior.validateAsInitial(Behavior.start(_initialBehavior, ctx))
// execute any future tasks scheduled in Actor's constructor
runAllTasks()
override def retrieveEffect(): Effect = ctx.effectQueue.poll() match {
case null ⇒ NoEffects
case x ⇒ x
@ -77,14 +80,18 @@ private[akka] final class BehaviorTestKitImpl[T](_name: String, _initialBehavior
throw e
}
private def runAllTasks(): Unit = {
ctx.executionContext match {
case controlled: ControlledExecutor ⇒ controlled.runAll()
case _ ⇒
}
}
override def run(msg: T): Unit = {
try {
currentUncanonical = Behavior.interpretMessage(current, ctx, msg)
current = Behavior.canonicalize(currentUncanonical, current, ctx)
ctx.executionContext match {
case controlled: ControlledExecutor ⇒ controlled.runAll()
case _ ⇒
}
runAllTasks()
} catch handleException
}