pekko/akka-docs/common/code/SchedulerDocSpec.scala

package akka.scheduler.actor

//#imports1
import akka.actor.Actor
import akka.actor.Props
import akka.util.duration._

//#imports1

import org.scalatest.{ BeforeAndAfterAll, WordSpec }
import org.scalatest.matchers.MustMatchers
import akka.testkit._
import akka.util.duration._

class SchedulerDocSpec extends AkkaSpec(Map("akka.loglevel" -> "INFO")) {
  "schedule a one-off task" in {
    //#schedule-one-off-message
    //Schedules to send the "foo"-message to the testActor after 50ms
    system.scheduler.scheduleOnce(50 milliseconds, testActor, "foo")
    //#schedule-one-off-message

    expectMsg(1 second, "foo")

    //#schedule-one-off-thunk
    //Schedules to send the "foo"-message to the testActor after 50ms
    system.scheduler.scheduleOnce(50 milliseconds) {
      testActor ! "foo"
    }
    //#schedule-one-off-thunk

    expectMsg(1 second, "foo")

    //#schedule-one-off-runnable
    //Schedules to send the "foo"-message to the testActor after 50ms
    system.scheduler.scheduleOnce(
      50 milliseconds,
      new Runnable {
        def run = testActor ! "foo"
      })

    //#schedule-one-off-runnable

    expectMsg(1 second, "foo")
  }

  "schedule a recurring task" in {
    //#schedule-recurring
    val Tick = "tick"
    val tickActor = system.actorOf(Props(new Actor {
      def receive = {
        case Tick ⇒ //Do something
      }
    }))
    //This will schedule to send the Tick-message
    //to the tickActor after 0ms repeating every 50ms
    val cancellable =
      system.scheduler.schedule(0 milliseconds,
        50 milliseconds,
        tickActor,
        Tick)

    //This cancels further Ticks to be sent
    cancellable.cancel()
    //#schedule-recurring
    tickActor.stop()
  }
}