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

Rewriting the fix to PatternSpec

Browse source
This commit is contained in:
Viktor Klang 2012-12-18 14:31:18 +01:00
parent 6afed38d5d
commit cdcc098e01
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
akka-actor-tests/src/test/scala/akka/pattern/PatternSpec.scala
View file

@ -6,7 +6,7 @@ package akka.pattern
import language.postfixOps import language.postfixOps
import akka.testkit.AkkaSpec import akka.testkit.{ TestLatch, AkkaSpec }
import akka.actor.{ Props, Actor } import akka.actor.{ Props, Actor }
import java.util.concurrent.TimeoutException import java.util.concurrent.TimeoutException
import scala.concurrent.{ Future, Promise, Await } import scala.concurrent.{ Future, Promise, Await }
@ -16,7 +16,8 @@ object PatternSpec {
case class Work(duration: Duration) case class Work(duration: Duration)
class TargetActor extends Actor { class TargetActor extends Actor {
def receive = { def receive = {
case Work(duration) ⇒ Thread.sleep(duration.toMillis) case (testLatch: TestLatch, duration: FiniteDuration) ⇒
Await.ready(testLatch, duration)
} }
} }
} }
@ -40,10 +41,12 @@ class PatternSpec extends AkkaSpec {
Await.ready(gracefulStop(target, 1 millis), 1 second) Await.ready(gracefulStop(target, 1 millis), 1 second)
} }
"complete Future with TimeoutException when actor not terminated within timeout" in { "complete Future with AskTimeoutException when actor not terminated within timeout" in {
val target = system.actorOf(Props[TargetActor]) val target = system.actorOf(Props[TargetActor])
target ! Work(250 millis) val latch = TestLatch()
intercept[TimeoutException] { Await.result(gracefulStop(target, 10 millis), 200 millis) } target ! (latch, remaining)
intercept[AskTimeoutException] { Await.result(gracefulStop(target, 500 millis), remaining) }
latch.open()
} }
} }