71 lines
1.7 KiB
Scala
71 lines
1.7 KiB
Scala
package akka.actor.dispatch
|
|
|
|
import java.util.concurrent.{ CountDownLatch, TimeUnit }
|
|
import org.scalatest.junit.JUnitSuite
|
|
import org.junit.Test
|
|
|
|
import akka.dispatch.Dispatchers
|
|
import akka.actor.Actor
|
|
import Actor._
|
|
|
|
object PinnedActorSpec {
|
|
class TestActor extends Actor {
|
|
self.dispatcher = Dispatchers.newPinnedDispatcher(self)
|
|
|
|
def receive = {
|
|
case "Hello" ⇒
|
|
self.reply("World")
|
|
case "Failure" ⇒
|
|
throw new RuntimeException("Expected exception; to test fault-tolerance")
|
|
}
|
|
}
|
|
}
|
|
|
|
class PinnedActorSpec extends JUnitSuite {
|
|
import PinnedActorSpec._
|
|
|
|
private val unit = TimeUnit.MILLISECONDS
|
|
|
|
@Test
|
|
def shouldSendOneWay {
|
|
var oneWay = new CountDownLatch(1)
|
|
val actor = actorOf(new Actor {
|
|
self.dispatcher = Dispatchers.newPinnedDispatcher(self)
|
|
def receive = {
|
|
case "OneWay" ⇒ oneWay.countDown()
|
|
}
|
|
}).start()
|
|
val result = actor ! "OneWay"
|
|
assert(oneWay.await(1, TimeUnit.SECONDS))
|
|
actor.stop()
|
|
}
|
|
|
|
@Test
|
|
def shouldSendReplySync = {
|
|
val actor = actorOf[TestActor].start()
|
|
val result = (actor ? ("Hello", 10000)).as[String]
|
|
assert("World" === result.get)
|
|
actor.stop()
|
|
}
|
|
|
|
@Test
|
|
def shouldSendReplyAsync = {
|
|
val actor = actorOf[TestActor].start()
|
|
val result = (actor ? "Hello").as[String]
|
|
assert("World" === result.get)
|
|
actor.stop()
|
|
}
|
|
|
|
@Test
|
|
def shouldSendReceiveException = {
|
|
val actor = actorOf[TestActor].start()
|
|
try {
|
|
(actor ? "Failure").get
|
|
fail("Should have thrown an exception")
|
|
} catch {
|
|
case e ⇒
|
|
assert("Expected exception; to test fault-tolerance" === e.getMessage())
|
|
}
|
|
actor.stop()
|
|
}
|
|
}
|