Merge branch 'tmp'

akka-docs/rst/scala/code/docs/actor/FaultHandlingDocSpec.scala

@@ -6,8 +6,10 @@ package docs.actor
 import language.postfixOps
 
 //#testkit
-import akka.testkit.{ AkkaSpec, ImplicitSender, EventFilter }
-import akka.actor.{ ActorRef, Props, Terminated }
+import akka.testkit.{ TestKit, ImplicitSender, EventFilter }
+import akka.actor.{ ActorRef, Props, ActorSystem, Terminated }
+import org.scalatest._
+import com.typesafe.config.{ Config, ConfigFactory }
 
 //#testkit
 object FaultHandlingDocSpec {
@@ -16,8 +18,6 @@ object FaultHandlingDocSpec {
   import akka.actor.Actor
 
   //#child
-  //#supervisor
-  //#supervisor
   class Supervisor extends Actor {
     //#strategy
     import akka.actor.OneForOneStrategy
@@ -79,7 +79,6 @@ object FaultHandlingDocSpec {
 
     def receive = Actor.emptyBehavior
   }
-  //#supervisor
 
   //#child
   class Child extends Actor {
@@ -91,83 +90,87 @@ object FaultHandlingDocSpec {
     }
   }
   //#child
+
+  val testConf: Config = ConfigFactory.parseString("""
+      akka {
+        loggers = ["akka.testkit.TestEventListener"]
+      }
+  """)
 }
 
+import FaultHandlingDocSpec._
 //#testkit
-class FaultHandlingDocSpec extends AkkaSpec with ImplicitSender {
-  //#testkit
+class FaultHandlingDocSpec extends TestKit(ActorSystem("FaultHandlingDocSpec", testConf))
+  with FlatSpecLike with BeforeAndAfterAll with ImplicitSender {
 
-  import FaultHandlingDocSpec._
-  //#testkit
+  override def afterAll() {
+    system.terminate()
+  }
 
-  "A supervisor" must {
+  "A supervisor" must "apply the chosen strategy for its child" in {
+	  //#testkit
 
-    "apply the chosen strategy for its child" in {
-      //#testkit
+    //#create
+    val supervisor = system.actorOf(Props[Supervisor], "supervisor")
 
-      //#create
-      val supervisor = system.actorOf(Props[Supervisor], "supervisor")
+    supervisor ! Props[Child]
+    val child = expectMsgType[ActorRef] // retrieve answer from TestKit’s testActor
+    //#create
+    EventFilter.warning(occurrences = 1) intercept {
+      //#resume
+      child ! 42 // set state to 42
+      child ! "get"
+      expectMsg(42)
 
-      supervisor ! Props[Child]
-      val child = expectMsgType[ActorRef] // retrieve answer from TestKit’s testActor
-      //#create
-      EventFilter.warning(occurrences = 1) intercept {
-        //#resume
-        child ! 42 // set state to 42
-        child ! "get"
-        expectMsg(42)
-
-        child ! new ArithmeticException // crash it
-        child ! "get"
-        expectMsg(42)
-        //#resume
-      }
-      EventFilter[NullPointerException](occurrences = 1) intercept {
-        //#restart
-        child ! new NullPointerException // crash it harder
-        child ! "get"
-        expectMsg(0)
-        //#restart
-      }
-      EventFilter[IllegalArgumentException](occurrences = 1) intercept {
-        //#stop
-        watch(child) // have testActor watch “child”
-        child ! new IllegalArgumentException // break it
-        expectMsgPF() { case Terminated(`child`) => () }
-        //#stop
-      }
-      EventFilter[Exception]("CRASH", occurrences = 2) intercept {
-        //#escalate-kill
-        supervisor ! Props[Child] // create new child
-        val child2 = expectMsgType[ActorRef]
-
-        watch(child2)
-        child2 ! "get" // verify it is alive
-        expectMsg(0)
-
-        child2 ! new Exception("CRASH") // escalate failure
-        expectMsgPF() {
-          case t @ Terminated(`child2`) if t.existenceConfirmed => ()
-        }
-        //#escalate-kill
-        //#escalate-restart
-        val supervisor2 = system.actorOf(Props[Supervisor2], "supervisor2")
-
-        supervisor2 ! Props[Child]
-        val child3 = expectMsgType[ActorRef]
-
-        child3 ! 23
-        child3 ! "get"
-        expectMsg(23)
-
-        child3 ! new Exception("CRASH")
-        child3 ! "get"
-        expectMsg(0)
-        //#escalate-restart
-      }
-      //#testkit
-      // code here
+      child ! new ArithmeticException // crash it
+      child ! "get"
+      expectMsg(42)
+      //#resume
     }
+    EventFilter[NullPointerException](occurrences = 1) intercept {
+      //#restart
+      child ! new NullPointerException // crash it harder
+      child ! "get"
+      expectMsg(0)
+      //#restart
+    }
+    EventFilter[IllegalArgumentException](occurrences = 1) intercept {
+      //#stop
+      watch(child) // have testActor watch “child”
+      child ! new IllegalArgumentException // break it
+      expectMsgPF() { case Terminated(`child`) => () }
+      //#stop
+    }
+    EventFilter[Exception]("CRASH", occurrences = 2) intercept {
+      //#escalate-kill
+      supervisor ! Props[Child] // create new child
+      val child2 = expectMsgType[ActorRef]
+      watch(child2)
+      child2 ! "get" // verify it is alive
+      expectMsg(0)
+
+      child2 ! new Exception("CRASH") // escalate failure
+      expectMsgPF() {
+        case t @ Terminated(`child2`) if t.existenceConfirmed => ()
+      }
+      //#escalate-kill
+      //#escalate-restart
+      val supervisor2 = system.actorOf(Props[Supervisor2], "supervisor2")
+
+      supervisor2 ! Props[Child]
+      val child3 = expectMsgType[ActorRef]
+
+      child3 ! 23
+      child3 ! "get"
+      expectMsg(23)
+
+      child3 ! new Exception("CRASH")
+      child3 ! "get"
+      expectMsg(0)
+      //#escalate-restart
+    }
+    //#testkit
+    // code here
   }
 }
 //#testkit