Added shutdown

Author: jamie-allen

chapter04/src/test/scala/com/reactivedesignpatterns/chapter4/FailureParentSpec.scala

@@ -6,15 +6,20 @@ import akka.actor.SupervisorStrategy.Stop
 import akka.testkit.TestProbe
 
 class FailureParentSpec extends WordSpec with Matchers with BeforeAndAfterAll {
-	"Using a FailureParent" must {
+  implicit val system = ActorSystem()
+
+  "Using a FailureParent" must {
 	  "Result in failures being collected and returned" in {
-		  implicit val system = ActorSystem()
 		  val failures = TestProbe()
 			val failureParent = system.actorOf(Props(new FailureParent(failures.ref)))
 			failureParent ! TestFailureParentMessage
 			failures.expectMsgType[NullPointerException]
 	  }
 	}
+
+  override def afterAll(): Unit = {
+    system.shutdown()
+  }
 }
 
 case object TestFailureParentMessage 

chapter04/src/test/scala/com/reactivedesignpatterns/chapter4/StepParentSpec.scala

@@ -6,18 +6,21 @@ import akka.actor.SupervisorStrategy.Restart
 import akka.testkit.TestProbe
 
 class StepParentSpec extends WordSpec with Matchers with BeforeAndAfterAll {
+  implicit val system = ActorSystem()
+
   "An actor that throws an exception" must {
     "Result in the supervisor returning a reference to that actor" in {
-      implicit val system = ActorSystem()
       val testProbe = TestProbe()
 			val parent = system.actorOf(Props[StepParent], "stepParent")
-			parent.tell(Props[MyActor], testProbe.ref)
+			parent.tell(Props[MyStepParentActor], testProbe.ref)
 			val child = testProbe.expectMsgType[ActorRef]
     }
   }
-}
 
-case object TestMessage
+  override def afterAll(): Unit = {
+    system.shutdown()
+  }
+}
 
 class StepParent extends Actor {
    override val supervisorStrategy = OneForOneStrategy() {
@@ -29,7 +32,7 @@ class StepParent extends Actor {
    }
 }
 
-class MyActor extends Actor {
+class MyStepParentActor extends Actor {
   def receive = {
     case _ => throw new NullPointerException
   }