ProblemHandlerTest.scala
9bea78dd1ff61.22.2.x

ProblemHandlerTest.scala
View Options

			package net.shrine.problem

			import net.shrine.log.Loggable
			import net.shrine.util.ShouldMatchersForJUnit
			import org.apache.log4j.Level

			import scala.collection.mutable.{Map => MMap}

			/**
			*
			* @author dwalend
			* @since 1.20
			*/
			final class ProblemHandlerTest extends ShouldMatchersForJUnit {

			def testProblemHandler() = {

			val loggable = new MockLoggable

			case class TestProblem(nodeName:String,summary:String,exception:Exception) extends AbstractProblem(ProblemSources.Hub) {

			val message = s"TestProblem involving $nodeName"

			val description = summary + message

			override def throwable = Some(exception)
			}

			val fakeException = new RuntimeException("test exception")
			fakeException.fillInStackTrace()

			val problem = TestProblem("testProblem","Problem created for testing",fakeException)

			val problemHandler = LoggingProblemHandler

			problemHandler.handleProblem(problem)

			loggable.loggedAt(Level.ERROR) should be (right = true)

			}

			private final class MockLoggable extends Loggable {
			val loggedAt: MMap[Level, Boolean] = MMap.empty

			override def debug(s: => Any) {
			loggedAt(Level.DEBUG) = true

			super.debug(s)
			}

			override def info(s: => Any) {
			loggedAt(Level.INFO) = true

			super.info(s)
			}

			override def warn(s: => Any) {
			loggedAt(Level.WARN) = true

			super.warn(s)
			}

			override def error(s: => Any) {
			loggedAt(Level.ERROR) = true

			super.error(s)
			}
			}
			}