pekko/akka-actor-tests/src/test/scala/akka/testkit/CallingThreadDispatcherModelSpec.scala

package akka.testkit

import akka.actor.dispatch.ActorModelSpec
import java.util.concurrent.CountDownLatch

class CallingThreadDispatcherModelSpec extends ActorModelSpec {
  import ActorModelSpec._
  def newInterceptedDispatcher = new CallingThreadDispatcher with MessageDispatcherInterceptor

  // A CallingThreadDispatcher can by design not process messages in parallel,
  // so disable this test
  override def dispatcherShouldProcessMessagesInParallel {}

  // This test needs to be adapted: CTD runs the flood completely sequentially
  // with start, invocation, stop, schedule shutdown, abort shutdown, repeat;
  // add "keeper" actor to lock down the dispatcher instance, since the
  // frequent attempted shutdown seems rather costly (random timing failures
  // without this fix)
  override def dispatcherShouldHandleWavesOfActors {
    implicit val dispatcher = newInterceptedDispatcher

    def flood(num: Int) {
      val cachedMessage = CountDownNStop(new CountDownLatch(num))
      val keeper = newTestActor.start()
      (1 to num) foreach {
        _ => newTestActor.start() ! cachedMessage
      }
      keeper.stop()
      assertCountDown(cachedMessage.latch,10000, "Should process " + num + " countdowns")
    }
    for(run <- 1 to 3) {
      flood(10000)
      await(dispatcher.stops.get == run)(withinMs = 10000)
      assertDispatcher(dispatcher)(starts = run, stops = run)
    }
  }

}

// vim: set ts=2 sw=2 et: