pekko/akka-actor-tests/src/test/scala/akka/event/EventStreamSpec.scala

/**
 *  Copyright (C) 2009-2013 Typesafe Inc. <http://www.typesafe.com>
 */
package akka.event

import language.postfixOps

import scala.concurrent.duration._
import akka.actor.{ Actor, ActorRef, ActorSystemImpl, ActorSystem, Props, UnhandledMessage }
import com.typesafe.config.ConfigFactory
import scala.collection.JavaConverters._
import akka.event.Logging.InitializeLogger
import akka.pattern.gracefulStop
import akka.testkit.{ EventFilter, TestEvent, TestProbe, AkkaSpec }

object EventStreamSpec {

  val config = ConfigFactory.parseString("""
      akka {
        actor.serialize-messages = off
        stdout-loglevel = WARNING
        loglevel = INFO
        loggers = ["akka.event.EventStreamSpec$MyLog", "%s"]
      }
      """.format(Logging.StandardOutLogger.getClass.getName))

  val configUnhandled = ConfigFactory.parseString("""
      akka {
        actor.serialize-messages = off
        stdout-loglevel = WARNING
        loglevel = DEBUG
        actor.debug.unhandled = on
      }
      """)

  case class M(i: Int)

  case class SetTarget(ref: ActorRef)

  class MyLog extends Actor {
    var dst: ActorRef = context.system.deadLetters
    def receive = {
      case Logging.InitializeLogger(bus) ⇒
        bus.subscribe(context.self, classOf[SetTarget])
        bus.subscribe(context.self, classOf[UnhandledMessage])
        sender ! Logging.LoggerInitialized
      case SetTarget(ref)      ⇒ { dst = ref; dst ! "OK" }
      case e: Logging.LogEvent ⇒ dst ! e
      case u: UnhandledMessage ⇒ dst ! u
    }
  }

  // class hierarchy for subchannel test
  class A
  class B1 extends A
  class B2 extends A
  class C extends B1

  trait T
  trait AT extends T
  trait ATT extends AT
  trait BT extends T
  trait BTT extends BT
  class CC
  class CCATBT extends CC with ATT with BTT
}

@org.junit.runner.RunWith(classOf[org.scalatest.junit.JUnitRunner])
class EventStreamSpec extends AkkaSpec(EventStreamSpec.config) {

  import EventStreamSpec._

  val impl = system.asInstanceOf[ActorSystemImpl]

  "An EventStream" must {

    "manage subscriptions" in {
      val bus = new EventStream(true)
      bus.subscribe(testActor, classOf[M])
      bus.publish(M(42))
      within(1 second) {
        expectMsg(M(42))
        bus.unsubscribe(testActor)
        bus.publish(M(13))
        expectNoMsg
      }
    }

    "not allow null as subscriber" in {
      val bus = new EventStream(true)
      intercept[IllegalArgumentException] { bus.subscribe(null, classOf[M]) }.getMessage should be("subscriber is null")
    }

    "not allow null as unsubscriber" in {
      val bus = new EventStream(true)
      intercept[IllegalArgumentException] { bus.unsubscribe(null, classOf[M]) }.getMessage should be("subscriber is null")
      intercept[IllegalArgumentException] { bus.unsubscribe(null) }.getMessage should be("subscriber is null")
    }

    "be able to log unhandled messages" in {
      val sys = ActorSystem("EventStreamSpecUnhandled", configUnhandled)
      try {
        sys.eventStream.subscribe(testActor, classOf[AnyRef])
        val m = UnhandledMessage(42, sys.deadLetters, sys.deadLetters)
        sys.eventStream.publish(m)
        expectMsgAllOf(m, Logging.Debug(sys.deadLetters.path.toString, sys.deadLetters.getClass, "unhandled message from " + sys.deadLetters + ": 42"))
        sys.eventStream.unsubscribe(testActor)
      } finally {
        shutdown(sys)
      }
    }

    "manage log levels" in {
      val bus = new EventStream(false)
      bus.startDefaultLoggers(impl)
      bus.publish(SetTarget(testActor))
      expectMsg("OK")
      within(2 seconds) {
        import Logging._
        verifyLevel(bus, InfoLevel)
        bus.setLogLevel(WarningLevel)
        verifyLevel(bus, WarningLevel)
        bus.setLogLevel(DebugLevel)
        verifyLevel(bus, DebugLevel)
        bus.setLogLevel(ErrorLevel)
        verifyLevel(bus, ErrorLevel)
      }
    }

    "manage sub-channels using classes" in {
      val a = new A
      val b1 = new B1
      val b2 = new B2
      val c = new C
      val bus = new EventStream(false)
      within(2 seconds) {
        bus.subscribe(testActor, classOf[B2]) should equal(true)
        bus.publish(c)
        bus.publish(b2)
        expectMsg(b2)
        bus.subscribe(testActor, classOf[A]) should equal(true)
        bus.publish(c)
        expectMsg(c)
        bus.publish(b1)
        expectMsg(b1)
        bus.unsubscribe(testActor, classOf[B1]) should equal(true)
        bus.publish(c)
        bus.publish(b2)
        bus.publish(a)
        expectMsg(b2)
        expectMsg(a)
        expectNoMsg
      }
    }

    "manage sub-channels using classes and traits (update on subscribe)" in {
      val es = new EventStream(false)
      val tm1 = new CC
      val tm2 = new CCATBT
      val a1, a2, a3, a4 = TestProbe()

      es.subscribe(a1.ref, classOf[AT]) should equal(true)
      es.subscribe(a2.ref, classOf[BT]) should equal(true)
      es.subscribe(a3.ref, classOf[CC]) should equal(true)
      es.subscribe(a4.ref, classOf[CCATBT]) should equal(true)
      es.publish(tm1)
      es.publish(tm2)
      a1.expectMsgType[AT] should equal(tm2)
      a2.expectMsgType[BT] should equal(tm2)
      a3.expectMsgType[CC] should equal(tm1)
      a3.expectMsgType[CC] should equal(tm2)
      a4.expectMsgType[CCATBT] should equal(tm2)
      es.unsubscribe(a1.ref, classOf[AT]) should equal(true)
      es.unsubscribe(a2.ref, classOf[BT]) should equal(true)
      es.unsubscribe(a3.ref, classOf[CC]) should equal(true)
      es.unsubscribe(a4.ref, classOf[CCATBT]) should equal(true)
    }

    "manage sub-channels using classes and traits (update on unsubscribe)" in {
      val es = new EventStream(false)
      val tm1 = new CC
      val tm2 = new CCATBT
      val a1, a2, a3, a4 = TestProbe()

      es.subscribe(a1.ref, classOf[AT]) should equal(true)
      es.subscribe(a2.ref, classOf[BT]) should equal(true)
      es.subscribe(a3.ref, classOf[CC]) should equal(true)
      es.subscribe(a4.ref, classOf[CCATBT]) should equal(true)
      es.unsubscribe(a3.ref, classOf[CC]) should equal(true)
      es.publish(tm1)
      es.publish(tm2)
      a1.expectMsgType[AT] should equal(tm2)
      a2.expectMsgType[BT] should equal(tm2)
      a3.expectNoMsg(1 second)
      a4.expectMsgType[CCATBT] should equal(tm2)
      es.unsubscribe(a1.ref, classOf[AT]) should equal(true)
      es.unsubscribe(a2.ref, classOf[BT]) should equal(true)
      es.unsubscribe(a4.ref, classOf[CCATBT]) should equal(true)
    }

    "manage sub-channels using classes and traits (update on unsubscribe all)" in {
      val es = new EventStream(false)
      val tm1 = new CC
      val tm2 = new CCATBT
      val a1, a2, a3, a4 = TestProbe()

      es.subscribe(a1.ref, classOf[AT]) should equal(true)
      es.subscribe(a2.ref, classOf[BT]) should equal(true)
      es.subscribe(a3.ref, classOf[CC]) should equal(true)
      es.subscribe(a4.ref, classOf[CCATBT]) should equal(true)
      es.unsubscribe(a3.ref)
      es.publish(tm1)
      es.publish(tm2)
      a1.expectMsgType[AT] should equal(tm2)
      a2.expectMsgType[BT] should equal(tm2)
      a3.expectNoMsg(1 second)
      a4.expectMsgType[CCATBT] should equal(tm2)
      es.unsubscribe(a1.ref, classOf[AT]) should equal(true)
      es.unsubscribe(a2.ref, classOf[BT]) should equal(true)
      es.unsubscribe(a4.ref, classOf[CCATBT]) should equal(true)
    }

    "manage sub-channels using classes and traits (update on publish)" in {
      val es = new EventStream(false)
      val tm1 = new CC
      val tm2 = new CCATBT
      val a1, a2 = TestProbe()

      es.subscribe(a1.ref, classOf[AT]) should equal(true)
      es.subscribe(a2.ref, classOf[BT]) should equal(true)
      es.publish(tm1)
      es.publish(tm2)
      a1.expectMsgType[AT] should equal(tm2)
      a2.expectMsgType[BT] should equal(tm2)
      es.unsubscribe(a1.ref, classOf[AT]) should equal(true)
      es.unsubscribe(a2.ref, classOf[BT]) should equal(true)
    }

    "manage sub-channels using classes and traits (unsubscribe classes used with trait)" in {
      val es = new EventStream(false)
      val tm1 = new CC
      val tm2 = new CCATBT
      val a1, a2, a3 = TestProbe()

      es.subscribe(a1.ref, classOf[AT]) should equal(true)
      es.subscribe(a2.ref, classOf[BT]) should equal(true)
      es.subscribe(a2.ref, classOf[CC]) should equal(true)
      es.subscribe(a3.ref, classOf[CC]) should equal(true)
      es.unsubscribe(a2.ref, classOf[CC]) should equal(true)
      es.unsubscribe(a3.ref, classOf[CCATBT]) should equal(true)
      es.publish(tm1)
      es.publish(tm2)
      a1.expectMsgType[AT] should equal(tm2)
      a2.expectMsgType[BT] should equal(tm2)
      a3.expectMsgType[CC] should equal(tm1)
      es.unsubscribe(a1.ref, classOf[AT]) should equal(true)
      es.unsubscribe(a2.ref, classOf[BT]) should equal(true)
      es.unsubscribe(a3.ref, classOf[CC]) should equal(true)
    }

    "manage sub-channels using classes and traits (subscribe after publish)" in {
      val es = new EventStream(false)
      val tm1 = new CCATBT
      val a1, a2 = TestProbe()

      es.subscribe(a1.ref, classOf[AT]) should equal(true)
      es.publish(tm1)
      a1.expectMsgType[AT] should equal(tm1)
      a2.expectNoMsg(1 second)
      es.subscribe(a2.ref, classOf[BTT]) should equal(true)
      es.publish(tm1)
      a1.expectMsgType[AT] should equal(tm1)
      a2.expectMsgType[BTT] should equal(tm1)
      es.unsubscribe(a1.ref, classOf[AT]) should equal(true)
      es.unsubscribe(a2.ref, classOf[BTT]) should equal(true)
    }
  }

  private def verifyLevel(bus: LoggingBus, level: Logging.LogLevel) {
    import Logging._
    val allmsg = Seq(Debug("", null, "debug"), Info("", null, "info"), Warning("", null, "warning"), Error("", null, "error"))
    val msg = allmsg filter (_.level <= level)
    allmsg foreach bus.publish
    msg foreach (expectMsg(_))
  }

}