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

add watch/unwatch for testActor to TestKit

Browse source
This commit is contained in:
Roland 2011-12-13 15:41:00 +01:00
parent 8617f9293d
commit 040f3076bc
1 changed files with 25 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

26
akka-testkit/src/main/scala/akka/testkit/TestKit.scala
View file

@ -16,6 +16,8 @@ object TestActor {
type Ignore = Option[PartialFunction[AnyRef, Boolean]]
case class SetIgnore(i: Ignore)
case class Watch(ref: ActorRef)
case class UnWatch(ref: ActorRef)
trait Message {
def msg: AnyRef
@ -34,7 +36,9 @@ class TestActor(queue: BlockingDeque[TestActor.Message]) extends Actor {
var ignore: Ignore = None
def receive = {
case SetIgnore(ign) ⇒ ignore = ign
case SetIgnore(ign) ⇒ ignore = ign
case x @ Watch(ref) ⇒ context.watch(ref); queue.offerLast(RealMessage(x, self))
case x @ UnWatch(ref) ⇒ context.unwatch(ref); queue.offerLast(RealMessage(x, self))
case x: AnyRef ⇒
val observe = ignore map (ignoreFunc ⇒ if (ignoreFunc isDefinedAt x) !ignoreFunc(x) else true) getOrElse true
if (observe) queue.offerLast(RealMessage(x, sender))
@ -123,6 +127,26 @@ class TestKit(_system: ActorSystem) {
*/
def ignoreNoMsg() { testActor ! TestActor.SetIgnore(None) }
/**
* Have the testActor watch someone (i.e. `context.watch(...)`). Waits until
* the Watch message is received back using expectMsg.
*/
def watch(ref: ActorRef) {
val msg = TestActor.Watch(ref)
testActor ! msg
expectMsg(msg)
}
/**
* Have the testActor stop watching someone (i.e. `context.unwatch(...)`). Waits until
* the Watch message is received back using expectMsg.
*/
def unwatch(ref: ActorRef) {
val msg = TestActor.UnWatch(ref)
testActor ! msg
expectMsg(msg)
}
/**
* Obtain current time (`System.nanoTime`) as Duration.
*/