70 lines
2.4 KiB
Scala
70 lines
2.4 KiB
Scala
/**
|
|
* Copyright (C) 2009-2016 Typesafe Inc. <http://www.typesafe.com>
|
|
*/
|
|
|
|
package akka.pattern
|
|
|
|
import language.postfixOps
|
|
|
|
import akka.testkit.{ TestLatch, AkkaSpec }
|
|
import akka.actor.{ Props, Actor }
|
|
import java.util.concurrent.TimeoutException
|
|
import scala.concurrent.{ Future, Promise, Await }
|
|
import scala.concurrent.duration._
|
|
|
|
object PatternSpec {
|
|
final case class Work(duration: Duration)
|
|
class TargetActor extends Actor {
|
|
def receive = {
|
|
case (testLatch: TestLatch, duration: FiniteDuration) ⇒
|
|
Await.ready(testLatch, duration)
|
|
}
|
|
}
|
|
}
|
|
|
|
@org.junit.runner.RunWith(classOf[org.scalatest.junit.JUnitRunner])
|
|
class PatternSpec extends AkkaSpec("akka.actor.serialize-messages = off") {
|
|
implicit val ec = system.dispatcher
|
|
import PatternSpec._
|
|
|
|
"pattern.gracefulStop" must {
|
|
|
|
"provide Future for stopping an actor" in {
|
|
val target = system.actorOf(Props[TargetActor])
|
|
val result = gracefulStop(target, 5 seconds)
|
|
Await.result(result, 6 seconds) should ===(true)
|
|
}
|
|
|
|
"complete Future when actor already terminated" in {
|
|
val target = system.actorOf(Props[TargetActor])
|
|
Await.ready(gracefulStop(target, 5 seconds), 6 seconds)
|
|
Await.ready(gracefulStop(target, 1 millis), 1 second)
|
|
}
|
|
|
|
"complete Future with AskTimeoutException when actor not terminated within timeout" in {
|
|
val target = system.actorOf(Props[TargetActor])
|
|
val latch = TestLatch()
|
|
target ! ((latch, remainingOrDefault))
|
|
intercept[AskTimeoutException] { Await.result(gracefulStop(target, 500 millis), remainingOrDefault) }
|
|
latch.open()
|
|
}
|
|
}
|
|
|
|
"pattern.after" must {
|
|
"be completed successfully eventually" in {
|
|
// TODO after is unfortunately shadowed by ScalaTest, fix as part of #3759
|
|
val f = akka.pattern.after(1 second, using = system.scheduler)(Promise.successful(5).future)
|
|
|
|
val r = Future.firstCompletedOf(Seq(Promise[Int]().future, f))
|
|
Await.result(r, remainingOrDefault) should ===(5)
|
|
}
|
|
|
|
"be completed abnormally eventually" in {
|
|
// TODO after is unfortunately shadowed by ScalaTest, fix as part of #3759
|
|
val f = akka.pattern.after(1 second, using = system.scheduler)(Promise.failed(new IllegalStateException("Mexico")).future)
|
|
|
|
val r = Future.firstCompletedOf(Seq(Promise[Int]().future, f))
|
|
intercept[IllegalStateException] { Await.result(r, remainingOrDefault) }.getMessage should ===("Mexico")
|
|
}
|
|
}
|
|
}
|